CN112989927B - 一种基于自监督预训练的场景图生成方法 - Google Patents
一种基于自监督预训练的场景图生成方法 Download PDFInfo
- Publication number
- CN112989927B CN112989927B CN202110151201.1A CN202110151201A CN112989927B CN 112989927 B CN112989927 B CN 112989927B CN 202110151201 A CN202110151201 A CN 202110151201A CN 112989927 B CN112989927 B CN 112989927B
- Authority
- CN
- China
- Prior art keywords
- self
- training
- scene graph
- image
- supervision
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
- 238000012549 training Methods 0.000 title claims abstract description 89
- 238000000034 method Methods 0.000 title claims abstract description 32
- 230000000007 visual effect Effects 0.000 claims abstract description 35
- 238000001514 detection method Methods 0.000 claims abstract description 22
- 238000007781 pre-processing Methods 0.000 claims abstract description 7
- 230000006870 function Effects 0.000 claims description 30
- 238000013507 mapping Methods 0.000 claims description 9
- 230000007246 mechanism Effects 0.000 claims description 7
- 238000012360 testing method Methods 0.000 claims description 6
- 238000012544 monitoring process Methods 0.000 claims description 4
- 230000004913 activation Effects 0.000 claims description 3
- 230000004927 fusion Effects 0.000 claims description 3
- 238000010606 normalization Methods 0.000 claims description 2
- 230000000694 effects Effects 0.000 abstract description 3
- 230000006872 improvement Effects 0.000 abstract description 2
- 238000010276 construction Methods 0.000 description 4
- 238000000638 solvent extraction Methods 0.000 description 4
- 238000010200 validation analysis Methods 0.000 description 4
- 238000013528 artificial neural network Methods 0.000 description 3
- 238000004364 calculation method Methods 0.000 description 3
- 238000011160 research Methods 0.000 description 3
- 230000003993 interaction Effects 0.000 description 2
- 238000012545 processing Methods 0.000 description 2
- 230000009286 beneficial effect Effects 0.000 description 1
- 238000004040 coloring Methods 0.000 description 1
- 238000013527 convolutional neural network Methods 0.000 description 1
- 125000004122 cyclic group Chemical group 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 238000010586 diagram Methods 0.000 description 1
- 239000003814 drug Substances 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 238000002372 labelling Methods 0.000 description 1
- 102000004196 processed proteins & peptides Human genes 0.000 description 1
- 108090000765 processed proteins & peptides Proteins 0.000 description 1
- 238000013519 translation Methods 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/40—Scenes; Scene-specific elements in video content
- G06V20/41—Higher-level, semantic clustering, classification or understanding of video scenes, e.g. detection, labelling or Markovian modelling of sport events or news items
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/25—Fusion techniques
Abstract
本发明公开了一种基于自监督预训练的场景图生成方法。本发明步骤如下:1、数据预处理及数据集的划分,2、使用训练好的目标检测网络对图像提取特征,3、构建目标的空间特征,4、构建自监督预训练网络模型,5、训练自监督预训练网络模型,6、构建自监督预训练和微调模型的场景图生成模型,7、训练场景图生成模型,8、网络预测值计算。本发明尤其是用于同时建模目标上下文和关系上下文在场景图生成的任务上取得了显著性的提升效果,超越了该任务上的大部分主流方法。并且本发明的基于自监督预训练的场景图生成方法在其他跨模态相关领域中如图像内容问答和视觉关系检测中也具有十分重要的应用价值和巨大的潜力。
Description
技术领域
本发明提出一种基于自监督预训练的场景图生成方法。
背景技术
场景图生成是新兴任务,主要应用在跨媒体领域。场景图生成任务主要是通过建模目标间的上下文信息来生成场景图。具体来讲是输入一张图片,模型深入理解图像内容,生成这个图像抽象出来的场景图,场景图中包含结点和边信息,结点和边分别表示目标和目标间的关系。例如,图像内容为一个带着头盔的人在街上骑车摩托车,其中可能包含有“人戴着头盔”、“人骑车”、“车在街上”、“头盔在人头上”。为了得到更准确的预测,机器需要深入理解图像的内容,先进行准确的目标检测,并在此基础上对目标上下文信息和目标之间的关系上下文信息进行融合以最大程度地挖掘目标之间的关系。和单纯的关系检测任务比起来,场景图生成可以被理解为具有同时建模目标上下文信息和关系上下文信息的任务,此任务的关键是如何同时建模好视觉信息及利用好上下文信息。
自监督预训练是利用原始数据作为监督来源,它已经被广泛应用在很多计算机视觉任务中,如图像着色、拼图解决、图像修复、旋转预测、相对位置预测等任务,此外自监督预训练对自然语言处理任务的研究有很大的推动作用。最近,将自监督预训练方法应用到多模态任务中越来越受到关注,方法是对大规模图像/视频和文本对进行预训练,然后对下游任务进行微调。
随着图像领域研究的推进,场景图相关任务也越来越受到重视。在场景图生成任务中,主流方法主要分为两种,一种是循环神经网络,一种是是基于卷积神经网络的模型,也有部分研究开始注意到基于注意力机制的模型,但是很少有涉及到将自监督预训练方法。自监督预训练主要是基于对大规模数据的预训练,然后对下游任务进行微调。由于自监督预训练的权重富含图片相关信息,有效的建模视觉信息对场景图生成任务尤为重要,所以,将自监督预训练和场景图生成任务相结合无疑有助于加深场景图的理解进而增强最终场景图生成的表达能力。
在落地应用中,场景图生成任务应用广泛,比如智能驾驶中通过判断驾驶员的手和方向盘的关系,判断驾驶员是否不规范驾驶;再者,在医学领域中,可以通过场景图生成的方法,来构造药物分子结构,具有十分良好的研究前景及重要的研究意义。
综上所述,场景图生成是一个非常值得深入研究的课题,本专利拟从该任务中几个关键点切入展开探讨,解决目前方法存在的难点和重点,形成一套完整的基于自监督预训练的场景图生成系统。
场景图标签一般是由不同的标注者完成的,差异性可能比较大;同时,图像载体主题多样,语义信息复杂差异大,这使得场景图生成面临巨大的挑战。具体而言,主要存在如下两方面的难点:
(1)如何对目标上下文信息进行有效建模:场景图数据集由于标注问题及实际情况,差异性比较大,呈现长尾分布,那么如何更加有效的建模视觉信息来避免长尾分布和先验带来的影响及干扰,解决这个问题尤为重要。为了能够对目标的视觉信息更好的进行建模,这里引入自监督预训练权重,这些自监督预训练的权重是在大量图片中进行自监督预训练的,包含大量图片的视觉信息。再者,数据集中的长尾分布,影响场景图生成的效果,这里主要根据目标的视觉信息来生成场景图,来尽量避免数据集中的干扰,使得生成的场景图更加准确。
(2)如何更深层的建模场景图语义信息,以更好获得丰富的上下文信息:大部分的方法是分开建模目标上下文信息和关系上下文信息,而且未对建模上下文信息的网络进行深度堆叠处理,这些方法可能会出现的问题是不能很好的有效的建模目标上下文信息和关系上下文信息,因为目标上下文信息和关系上下文信息是相互影响的,所以我们这里采用的是同时建模目标上下文信息和关系上下文信息,并且通过将网络深度堆叠来获得更丰富的上下文信息。目标和目标之间一般具有较高的差异性,该问题主要是通过注意力机制进行处理,得到有效的目标上下文特征表达。建模有效的富含上下文信息的目标特征,并通过注意力机制特征建模将目标和所有其他的目标之间的信息进行交互,得到可有效表达目标上下文的特征,更好的预测场景图生成。由于上下文信息对场景图的正确生成尤为重要,因此该任务对上下文信息的有效性更加敏感,对上下文信息的建模依赖性更高,所以如何让算法自动学习到目标和其他目标之间的关系,并生成更有效的目标上下文的表达特征,是场景图生成算法中的难点,也是影响结果性能的根本原因。
发明内容
本发明提供了一种基于自监督预训练的场景图生成方法。本发明主要包含两点:
1、提出一种基于自监督预训练的场景图生成方法,将网络权重在大量图片自监督预训练好的权重来初始化,对上下文信息进行充分建模,充分理解场景图的上下文信息和目标相关性,得到包含目标上下文信息和关系上下文信息的有效表达。
2、将目标上下文信息和关系上下文信息的特征同时进行建模,充分发掘上下文的信息,得到更有效的上下文信息。
本发明解决其技术问题所采用的技术方案包括如下步骤:
步骤(1)、数据预处理及数据集的划分
对于场景图生成的数据集Ⅰ和预训练的数据集Ⅱ中的图像进行预处理,并且对预处理后的数据集Ⅰ和数据集Ⅱ均划分训练数据集和测试数据集。
步骤(2):使用训练好的目标检测网络(Faster RCNN)对预处理后的数据集I和数据集II中的图像提取特征;
对于一张图像,使用训练好的目标检测网络计算出图像中包含物体的m个候选框;针对每一个候选框,将该候选框在图像中对应的区域输入到目标检测网络中,并提取目标检测网络中某一层的输出作为该候选框的视觉特征将一张图像中所有候选框的特征拼接成总体视觉特征/> 表示自然实数集,dv代表候选框的特征维度;
步骤(3):构建候选框的空间特征
针对每张图像,根据步骤(2)中得到的m个候选框来计算每个候选框的空间特征将每张图像中所有候选框的空间特征拼接成总空间特征为/>ds代表候选框的特征维度;
步骤(4):构建自监督预训练网络模型
将图像特征中视觉特征V、空间特征S来生成最终的图像特征将该图像特征Z输入到自监督预训练网络模型进行自监督预训练,根据多层自注意力机制将图像特征Z映射到隐藏特征空间,产生特征向量/>最后将特征向量A输入到掩码区域重建模块中,来重建被掩码的图像区域,输出重建后预测向量/>将预测向量与对应的目标向量(已知标签向量)一起输入到相应的损失函数中,输出损失值I。
步骤(5):训练自监督预训练网络模型
根据步骤(4)中的损失函数产生的损失值I,利用反向传播算法对步骤(4)的自监督预训练网络模型的参数进行梯度回传,不断优化,直至整个网络模型收敛。
步骤(6):构建基于自监督预训练的场景图生成模型
在自监督预训练网络模型的基础上,加入场景图生成模块。将自监督预训练网络模型输出的特征向量输入到场景图生成模块,预测m×m对候选框之间的预测关系/>将输出的预测关系与对应的目标关系(已知标签关系)一起输入到相应的损失函数中,输出损失值Ⅱ。
步骤(7):训练场景图生成模型
将步骤(5)中自监督预训练好的权重作为场景图生成模型的初始化权重,来替代随机初始化的权重,根据步骤(6)中的损失函数产生的损失值Ⅱ,利用反向传播算法对步骤(6)的场景图生成模型的参数进行梯度回传,不断优化,直至整个网络模型收敛。
所述的自监督预训练好的权重,是由预训练的数据集Ⅱ中的训练数据集经过自监督预训练网络模型得出的权重。
步骤(8):网络预测值计算
对场景图生成模型输出的预测关系中的关系分数p进行排序,选择分数最高的作为该预测关系的分类。
步骤(1)具体实现如下:
先处理场景图生成任务的数据集,剔除一些低质量的图片数据,并且使用最常见的150个目标类别和50个关系类别;还有数据集的划分,将数据集中70%的数据用于训练,剩下的30%用于测试。
预训练的数据集很庞大,包含大量的数据,其中包含COCO、Visual Genome、Conceptual Captions,SBU Captions,这四个数据集,剔除一些重合的数据,其中对于COCO数据集,95%用于训练集,5%用于验证集;对于Visual Genome数据集,98%用于训练集,2%用于验证集;对于Conceptual Captions数据集,99%用于验证集,1%用于测试集;对于SBU Captions数据集,99%用于训练集,1%用于验证集。
步骤(2)所述使用训练好的目标检测网络对图像提取特征,具体如下:
进一步的,步骤(2)所述的用目标检测网络对图像提取特征,具体如下:
提取目标检测网络中某一层的输出作为该候选框的特征每张图像选取m个候选框,将图像中所有的候选框拼接成总体视觉特征/>具体公式如下:
V=[v1,v2,...,vi,...,vm] (公式1)。
进一步的,步骤(3)所述的构建目标的空间特征,具体如下:
每个候选框的空间位置坐标为(xmin,ymin,xmax,ymax),(xmin,ymin)表示候选框的左上角点的位置坐标,(xmax,ymax)表示候选框的右下角点的位置坐标,每个候选框的空间特征s公式如下:
其中,W、H表示图像的宽和高;w、h表示对应的候选框的宽和高;
将一张图像中的所有候选框的空间特征拼接成总体空间特征具体公式如下:
S=[s1,s2,...,si,...,sm] (公式3)。
步骤(4)所述构建自监督预训练网络模型,具体如下:
4-1.融合视觉特征、空间特征
视觉特征V、空间特征S看作是目标的多种角度的特征,它们自然对齐,故可以方便融合。因为各特征信息差异较大,所以先将各特征先经过全连接层转换到一样的维度dz,并且将这两种特征相加得到图像特征为公式如下:
Z=V+S (公式4)
4-2.构建自监督预训练网络模型的基础网络结构SA:
图像特征Z经过全连接映射转化为特征向量公式如下:
X=Linear(Z) (公式5)
将特征向量X作为自监督预训练网络模型的输入,输出特征向量
A′=LN(X+MHA(X,X,X)) (公式6)
A=LN(A′+FFN(A′)) (公式7)
其中,MHA是多头注意力模块中的一个函数,其输入是X,输出是富含上下文信息的特征向量公式如下:
Q=Linear(X) (公式8)
K=Linear(X) (公式9)
V′=Linear(X) (公式10)
其中,Q、K、V′分别由Z经过全连接层映射得到,其中, MLP是两层感知机,ReLU是激活函数。
其中,FFN结构,输入是特征向量A′,公式如下:
FFN(A′)=FCd(Drop(ReLU(FC4d(A′)))) (公式12)
其中,LN是归一化函数。
4-3.深度堆叠基础网络结构SA
将融合后的图像特征Z作为基础网络结构SA[SA(1),SA(2),...,SA(N)]的输入。将第n层SA(n)的输出特征作为第n+1层SA(n+1)的输入,迭代往复,公式如下:
Z(n)=SA(n)(Z(n-1)) (公式13)
其中,Z(0)=Z。
4-4.掩码区域重建模块
将步骤4-3的输出Z(n)经过全连接层映射得到预测向量dc代表目标的视觉特征的维度,公式如下:
C=Linear(Z(n)) (公式14)
损失函数,具体如下:
计算经过掩码区域重建模块重建后的视觉特征ci与真实标签ci *之间的差距,这里使用SmoothL1Loss损失函数,具体公式如下:
所述的真实标签ci *为已知标签向量的一个元素。
步骤(6)所述的构建自监督预训练和微调模型的场景图生成模型,具体如下:
将步骤4-3的输出Z(n)作为场景图生成模型的输入,经过全连接层输出单头pj,公式如下:
pj=Linear(Z(n))Linear(Z(n))T (公式16)
将步骤4-3的输出Z(n)作为多头注意力关系预测器的输入,预测关系作为其输出,r代表关系分类数目,公式如下:
P=([p1,p2,...,pj,...,ph])Wr (公式17)
其中,h代表注意力头的数目,
损失函数,具体如下:
计算输出的预测关系与对应的目标关系(已知标签关系)之间的差距,即计算预测分数pi(预测分数是预测关系中的元素)与真实值pi *(真实值是目标关系中的元素)之间的差距,这里使用交叉熵(sigmoid cross entropy),具体公式如下:
本发明有益效果如下:
本发明核心方法为将自监督预训练应用到场景图生成任务中,用在大量图片数据中自监督预训练好的权重作为场景图生成模型的初始化权重,并在场景图生成这一深度学习任务中验证该模型的优越性。本方法首次将自监督预训练应用到场景图生成,将在大量图片中自监督预训练后的网络权重应用到场景图生成中,能有效的建模目标上下文信息和目标间关系上下文信息,同时建立了关系间的上下文信息的深层次理解和交互。本发明提出基于自监督预训练的场景图生成方法,同时将自注意力机制应用到场景图生成,使其不仅仅只适用于机器翻译等自然语言任务。实验中将图像的目标特征输入自监督预训练网络结构,在模型中完成了目标和关系上下文信息的深层理解和交互,进而得到了更好的实验结果,证明该方法普适性强,同时可在关系推理等多模态任务中充分发挥价值。
本发明尤其是用于同时建模目标上下文和关系上下文在场景图生成的任务上取得了显著性的提升效果,超越了该任务上的大部分主流方法。并且本发明的基于自监督预训练的场景图生成方法在其他跨模态相关领域中如图像内容问答和视觉关系检测中也具有十分重要的应用价值和巨大的潜力。
附图说明
图1:自注意力模块
图2:自监督预训练与微调模型图
具体实施方式
下面对本发明的详细参数做进一步具体说明。
如图1和图2所示,本发明提供一种基于深度关系自注意力网络的场景图生成方法。
步骤(1)、数据预处理及数据集的划分
对于场景图生成的数据集I和预训练的数据集II中的图像进行预处理,并且对预处理后的数据集I和数据集II均划分训练数据集和测试数据集。
步骤(2):使用训练好的目标检测网络(Faster RCNN)对预处理后的数据集I和数据集II中的图像提取特征;
对于一张图像,使用训练好的目标检测网络计算出图像中包含物体的m个候选框;针对每一个候选框,将该候选框在图像中对应的区域输入到目标检测网络中,并提取目标检测网络中某一层的输出作为该候选框的视觉特征将一张图像中所有候选框的特征拼接成总体视觉特征/> 表示自然实数集,dv代表候选框的特征维度;
步骤(3):构建候选框的空间特征
针对每张图像,根据步骤(2)中得到的m个候选框来计算每个候选框的空间特征将每张图像中所有候选框的空间特征拼接成总空间特征为/>ds代表候选框的特征维度;
步骤(4):构建自监督预训练网络模型
将图像特征中视觉特征V、空间特征S来生成最终的图像特征将该图像特征Z输入到自监督预训练网络模型进行自监督预训练,根据多层自注意力机制将图像特征Z映射到隐藏特征空间,产生特征向量/>最后将特征向量A输入到掩码区域重建模块中,来重建被掩码的图像区域,输出重建后预测向量/>将预测向量与对应的目标向量(已知标签向量)一起输入到相应的损失函数中,输出损失值I。
步骤(5):训练自监督预训练网络模型
根据步骤(4)中的损失函数产生的损失值I,利用反向传播算法对步骤(4)的自监督预训练网络模型的参数进行梯度回传,不断优化,直至整个网络模型收敛。
步骤(6):构建基于自监督预训练的场景图生成模型
在自监督预训练网络模型的基础上,加入场景图生成模块。将自监督预训练网络模型输出的特征向量输入到场景图生成模块,预测m×m对候选框之间的预测关系/>将输出的预测关系与对应的目标关系(已知标签关系)一起输入到相应的损失函数中,输出损失值II。
步骤(7):训练场景图生成模型
将步骤(5)中自监督预训练好的权重作为场景图生成模型的初始化权重,来替代随机初始化的权重,根据步骤(6)中的损失函数产生的损失值II,利用反向传播算法对步骤(6)的场景图生成模型的参数进行梯度回传,不断优化,直至整个网络模型收敛。
所述的自监督预训练好的权重,是由预训练的数据集II中的训练数据集经过自监督预训练网络模型得出的权重。
步骤(8):网络预测值计算
对场景图生成模型输出的预测关系中的关系分数p进行排序,选择分数最高的作为该预测关系的分类。
步骤(1)具体实现如下:
剔除一些低质量的图片数据,并且使用最常见的150个目标类别和50个关系类别;还有数据集的划分,将数据集中70%的数据用于训练,剩下的30%用于测试。
步骤(2)所述使用训练好的目标检测网络对图像提取特征,具体如下:
每个候选框对应图像区域的特征pf,其中每张图片选取64个候选框,将一张图片中所有的候选框拼接成总体特征/>
步骤(3)所述根据目标的空间位置坐标来构建空间特征,具体如下:
每个回归框的空间位置坐标为(xmin,ymin,xmax,ymax),(xmin,ymin)表示目标框的左上角点的位置坐标,(xmax,ymax)表示目标框的右下角点的位置坐标。
将一张图片中的所有目标空间特征拼接成总体空间特
步骤(4)所述构建自监督预训练网络模型,具体如下:
4-1.融合视觉特征、空间位置特征
视觉特征V、空间位置特征S可以看作是目标的多种角度的特征,它们自然对齐,故可以方便融合。因为各特征信息差异较大,所以先将各特征先经过全连接层转换到一样的维度dz为768,并且将这两种特征相加得到特征为
4-2.构建自监督预训练网络模型的基础网络结构SA
特征Z经过全连接映射转化为特征向量将特征向量X作为自监督预训练网络模型的输入,输出/>MHA的输入是X,输出是富含上下文信息的特征向量/>其中Q、K、V′分别由Z经过全连接层映射得到,其中,/>MLP是两层感知机,ReLU是激活函数。
4-3.深度堆叠SA网络
以融合后的特征Z作为输入,深度堆叠SA网络[SA(1),SA(2),...,SA(N)]。将第n层SA(n)的输出特征作为第n+1层SA(n+1)的输入,迭代往复,其中n取12层。
4-4.掩码区域建模模块
将4-3.的输出Z(n)经过全连接层映射到dc代表目标的视觉特征的维度为2048。
4-5.自监督预训练网络模型的损失函数
计算重建的视觉特征ci与真实标签ci *之间的差距,这里使用SmoothL1Loss损失函数.
步骤(5)所述的自监督预训练网络模型的训练,具体如下:
根据步骤(4)中的损失函数产生的损失值利用反向传播算法对步骤(4)中神经网络的模型参数进行梯度回传,不断优化,直至整个网络模型收敛。
步骤(6)所述的构建自监督预训练和微调模型的场景图生成模型,具体如下:
将4-3的输出Z(n)作为输入,经过全连接层输出单头pj。以4-3的输出Z(n)和关系特征F作为多头注意力关系预测器的输入,关系预测结果作为其输出,r代表关系分类数目为51,其中,h代表注意力头的数目为12.损失函数,具体如下:
计算目标关系对的预测分数与真实值之间的差距,这里使用交叉熵(sigmoidcross entropy)。
步骤(7)所述的训练模型,具体如下:
根据步骤(6)中的损失函数产生的损失值利用反向传播算法对步骤(6)中神经网络的模型参数进行梯度回传,不断优化,直至整个网络模型收敛。
步骤(8)所述的网络预测值计算,具体如下:
根据步骤(7)输出的关系预测分数P进行排序,取分数最高的分类作为目标关系对的最终关系分类。
Claims (3)
1.一种基于自监督预训练的场景图生成方法,其特征在于包括如下步骤:
步骤(1)、数据预处理及数据集的划分;
对于场景图生成的数据集I和预训练的数据集II中的图像进行预处理,并且对预处理后的数据集I和数据集II均划分训练数据集和测试数据集;
步骤(2):使用训练好的目标检测网络对预处理后的数据集I和数据集II中的图像提取特征;
对于一张图像,使用训练好的目标检测网络计算出图像中包含物体的m个候选框;针对每一个候选框,将该候选框在图像中对应的区域输入到目标检测网络中,并提取目标检测网络中某一层的输出作为该候选框的视觉特征将一张图像中所有候选框的特征拼接成总体视觉特征/> 表示自然实数集,dv代表候选框的侯选框的视觉特征维度;
步骤(3):构建候选框的空间特征;
针对每张图像,根据步骤(2)中得到的m个候选框来计算每个候选框的空间特征将每张图像中所有候选框的空间特征拼接成总空间特征为/>ds代表候选框的侯选框的空间特征维度;
步骤(4):构建自监督预训练网络模型;
将图像特征中视觉特征V、空间特征S来生成最终的图像特征将该图像特征Z输入到自监督预训练网络模型进行自监督预训练,根据多层自注意力机制将图像特征Z映射到隐藏特征空间,产生特征向量/>最后将特征向量A输入到掩码区域重建模块中,来重建被掩码的图像区域,输出重建后预测向量/>将预测向量与对应的目标向量一起输入到相应的损失函数中,输出损失值I;
步骤(5):训练自监督预训练网络模型;
根据步骤(4)中的损失函数产生的损失值I,利用反向传播算法对步骤(4)的自监督预训练网络模型的参数进行梯度回传,不断优化,直至整个网络模型收敛;
步骤(6):构建基于自监督预训练的场景图生成模型;
在自监督预训练网络模型的基础上,加入场景图生成模块;将自监督预训练网络模型输出的预测向量输入到场景图生成模块,预测m×m对候选框之间的预测关系r表示关系分类数目;将输出的预测关系与对应的目标关系一起输入到相应的损失函数中,输出损失值II;
步骤(7):训练场景图生成模型;
将步骤(5)中自监督预训练好的权重作为场景图生成模型的初始化权重,来替代随机初始化的权重,根据步骤(6)中的损失函数产生的损失值II,利用反向传播算法对步骤(6)的场景图生成模型的参数进行梯度回传,不断优化,直至整个网络模型收敛;
所述的自监督预训练好的权重,是由预训练的数据集II中的训练数据集经过自监督预训练网络模型得出的权重;
步骤(8):网络预测值计算;
对场景图生成模型输出的预测关系中的关系分数p进行排序,选择分数最高的作为该预测关系的分类。
2.根据权利要求1所述的一种基于自监督预训练的场景图生成方法,其特征在于步骤4具体实现如下:
4-1.融合视觉特征、空间特征
将视觉特征V、空间特征S看作是目标的多种角度的特征,先将各特征先经过全连接层转换到一样的维度dz,并且将这两种特征相加得到图像特征为公式如下:
Z=V+S (1)
4-2.构建自监督预训练网络模型的基础网络结构SA:
图像特征Z经过全连接映射转化为特征向量公式如下:
X=Linear(Z) (2)
将特征向量X作为自监督预训练网络模型的输入,输出特征向量
A′=LN(X+MHA(X,X,X)) (3)
A=LN(A′+FFN(A′)) (4)
其中,MHA是多头注意力模块中的一个函数,其输入是X,输出是富含上下文信息的特征向量公式如下:
Q=Linear(X) (5)
K=Linear(X) (6)
V′=Linear(X) (7)
其中,Q、K、V′分别由Z经过全连接层映射得到,其中, MLP是两层感知机,ReLU是激活函数;
其中,FFN结构,输入是特征向量A′,公式如下:
FFN(A′)=FCd(Drop(ReLU(FC4d(A′)))) (9)
其中,LN是归一化函数;
4-3.深度堆叠基础网络结构SA
将融合后的图像特征Z作为基础网络结构SA[SA(1),SA(2),...,SA(N)]的输入;将第n层SA(n)的输出特征作为第n+1层SA(n+1)的输入,迭代往复,公式如下:
Z(n)=SA(n)(Z(n-1)) (10)
其中,Z(0)=Z;
4-4.掩码区域重建模块
将步骤4-3的输出Z(n)经过全连接层映射得到预测向量dc代表目标的视觉特征的维度,公式如下:
C=Linear(Z(n)) (11)
损失函数,具体如下:
计算经过掩码区域重建模块重建后的视觉特征ci与真实标签ci *之间的差距,这里使用SmoothL1Loss损失函数,具体公式如下:
所述的真实标签ci *为已知标签向量的一个元素。
3.根据权利要求1所述的一种基于自监督预训练的场景图生成方法,其特征在于步骤(6)所述的构建自监督预训练和微调模型的场景图生成模型,具体如下:
将步骤4-3的输出Z(n)作为场景图生成模型的输入,经过全连接层输出单头pj,公式如下:
pj=Linear(Z(n))Linear(Z(n))T (13)
将步骤4-3的输出Z(n)作为多头注意力关系预测器的输入,预测关系作为其输出,r代表关系分类数目,公式如下:
P=([p1,p2,...,pj,...,ph])Wr (14)
其中,h代表注意力头的数目,
损失函数,具体如下:
计算输出的预测关系与对应的目标关系之间的差距,即计算预测分数pi与真实值pi *之间的差距,这里使用交叉熵,具体公式如下:
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110151201.1A CN112989927B (zh) | 2021-02-03 | 2021-02-03 | 一种基于自监督预训练的场景图生成方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110151201.1A CN112989927B (zh) | 2021-02-03 | 2021-02-03 | 一种基于自监督预训练的场景图生成方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112989927A CN112989927A (zh) | 2021-06-18 |
CN112989927B true CN112989927B (zh) | 2024-03-05 |
Family
ID=76346566
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110151201.1A Active CN112989927B (zh) | 2021-02-03 | 2021-02-03 | 一种基于自监督预训练的场景图生成方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112989927B (zh) |
Families Citing this family (11)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113255669B (zh) * | 2021-06-28 | 2021-10-01 | 山东大学 | 任意形状自然场景文本检测方法及系统 |
CN113449643A (zh) * | 2021-06-29 | 2021-09-28 | 福信富通科技股份有限公司 | 使用自监督预训练的TimeSformer进行视频目标检测的方法及其应用 |
CN113689514B (zh) * | 2021-06-29 | 2024-04-09 | 杭州电子科技大学 | 一种面向主题的图像场景图生成方法 |
CN113627557B (zh) * | 2021-08-19 | 2023-10-03 | 电子科技大学 | 一种基于上下文图注意力机制的场景图生成方法 |
CN113570695B (zh) * | 2021-09-27 | 2021-12-24 | 清华大学 | 一种图像生成方法、装置及电子设备 |
CN114266863B (zh) * | 2021-12-31 | 2024-02-09 | 西安交通大学 | 一种基于点云的3d场景图生成方法、系统、设备及可读存储介质 |
CN114511779B (zh) * | 2022-01-20 | 2023-07-25 | 电子科技大学 | 场景图生成模型的训练方法、场景图生成方法及装置 |
CN114612685B (zh) * | 2022-03-22 | 2022-12-23 | 中国科学院空天信息创新研究院 | 一种结合深度特征与对比学习的自监督信息提取方法 |
CN114782791B (zh) * | 2022-04-14 | 2024-03-22 | 华南理工大学 | 基于transformer模型和类别关联的场景图生成方法 |
CN114708472B (zh) * | 2022-06-06 | 2022-09-09 | 浙江大学 | 面向ai实训的多模态数据集标注方法、装置及电子设备 |
CN114723756B (zh) * | 2022-06-09 | 2022-08-12 | 北京理工大学 | 基于双监督网络的低分时序遥感目标检测方法及装置 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109165682A (zh) * | 2018-08-10 | 2019-01-08 | 中国地质大学(武汉) | 一种融合深度特征和显著性特征的遥感图像场景分类方法 |
CN110084128A (zh) * | 2019-03-29 | 2019-08-02 | 安徽艾睿思智能科技有限公司 | 基于语义空间约束和注意力机制的场景图生成方法 |
CN111462282A (zh) * | 2020-04-02 | 2020-07-28 | 哈尔滨工程大学 | 一种场景图生成方法 |
-
2021
- 2021-02-03 CN CN202110151201.1A patent/CN112989927B/zh active Active
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109165682A (zh) * | 2018-08-10 | 2019-01-08 | 中国地质大学(武汉) | 一种融合深度特征和显著性特征的遥感图像场景分类方法 |
CN110084128A (zh) * | 2019-03-29 | 2019-08-02 | 安徽艾睿思智能科技有限公司 | 基于语义空间约束和注意力机制的场景图生成方法 |
CN111462282A (zh) * | 2020-04-02 | 2020-07-28 | 哈尔滨工程大学 | 一种场景图生成方法 |
Also Published As
Publication number | Publication date |
---|---|
CN112989927A (zh) | 2021-06-18 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112989927B (zh) | 一种基于自监督预训练的场景图生成方法 | |
Chen et al. | Learning student networks via feature embedding | |
Kollias et al. | Analysing affective behavior in the first abaw 2020 competition | |
Zhong et al. | Self-adaptive neural module transformer for visual question answering | |
CN112464016B (zh) | 一种基于深度关系自注意力网络的场景图生成方法 | |
Zhou et al. | PGDENet: Progressive guided fusion and depth enhancement network for RGB-D indoor scene parsing | |
CN113297370B (zh) | 基于多交互注意力的端到端多模态问答方法及系统 | |
Wang et al. | A multi-task mean teacher for semi-supervised facial affective behavior analysis | |
Ristea et al. | Emotion recognition system from speech and visual information based on convolutional neural networks | |
Zeng et al. | Video question answering with prior knowledge and object-sensitive learning | |
Cherian et al. | Spatio-temporal ranked-attention networks for video captioning | |
Dai et al. | Video scene segmentation using tensor-train faster-RCNN for multimedia IoT systems | |
Fu et al. | Learning semantic-aware spatial-temporal attention for interpretable action recognition | |
Xia et al. | Maximum structural generation discrepancy for unsupervised domain adaptation | |
Li et al. | Multi-scale residual network model combined with Global Average Pooling for action recognition | |
Mazen et al. | Real time face expression recognition along with balanced FER2013 dataset using CycleGAN | |
Thuseethan et al. | EmoSeC: Emotion recognition from scene context | |
Zhenhua et al. | FTCF: Full temporal cross fusion network for violence detection in videos | |
Yin et al. | Msa-gcn: Multiscale adaptive graph convolution network for gait emotion recognition | |
Wani et al. | Deep learning-based video action recognition: a review | |
Ge | Efficient deep learning in network compression and acceleration | |
Mikhaylevskiy et al. | Fast emotion recognition neural network for IoT devices | |
CN115858728A (zh) | 一种基于多模态数据的情感分析方法 | |
Wang et al. | RETRACTED ARTICLE: Human behaviour recognition and monitoring based on deep convolutional neural networks | |
Bagane et al. | Facial Emotion Detection using Convolutional Neural Network |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |