CN117765341A - 一种数据处理方法及相关装置 - Google Patents
一种数据处理方法及相关装置 Download PDFInfo
- Publication number
- CN117765341A CN117765341A CN202211168795.8A CN202211168795A CN117765341A CN 117765341 A CN117765341 A CN 117765341A CN 202211168795 A CN202211168795 A CN 202211168795A CN 117765341 A CN117765341 A CN 117765341A
- Authority
- CN
- China
- Prior art keywords
- representation
- feature
- feature representation
- image
- fusion
- 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.)
- Pending
Links
- 238000003672 processing method Methods 0.000 title abstract description 18
- 230000003993 interaction Effects 0.000 claims abstract description 93
- 238000000034 method Methods 0.000 claims abstract description 83
- 230000004927 fusion Effects 0.000 claims abstract description 62
- 230000007246 mechanism Effects 0.000 claims abstract description 25
- 239000013598 vector Substances 0.000 claims description 160
- 238000012545 processing Methods 0.000 claims description 152
- 230000015654 memory Effects 0.000 claims description 70
- 238000001514 detection method Methods 0.000 claims description 64
- 230000009466 transformation Effects 0.000 claims description 26
- 238000005070 sampling Methods 0.000 claims description 15
- 238000003709 image segmentation Methods 0.000 claims description 13
- 238000004590 computer program Methods 0.000 claims description 7
- 238000013473 artificial intelligence Methods 0.000 abstract description 12
- 239000011159 matrix material Substances 0.000 description 59
- 230000006870 function Effects 0.000 description 58
- 238000012549 training Methods 0.000 description 47
- 238000013528 artificial neural network Methods 0.000 description 36
- 230000008569 process Effects 0.000 description 36
- 238000011176 pooling Methods 0.000 description 32
- 238000013527 convolutional neural network Methods 0.000 description 29
- 238000010586 diagram Methods 0.000 description 26
- 238000004891 communication Methods 0.000 description 15
- 230000008447 perception Effects 0.000 description 15
- 238000004364 calculation method Methods 0.000 description 11
- 230000001537 neural effect Effects 0.000 description 11
- 210000002569 neuron Anatomy 0.000 description 11
- 238000013500 data storage Methods 0.000 description 9
- 230000004913 activation Effects 0.000 description 8
- 230000018109 developmental process Effects 0.000 description 8
- 238000004422 calculation algorithm Methods 0.000 description 7
- 238000010606 normalization Methods 0.000 description 7
- 238000011161 development Methods 0.000 description 6
- 230000000694 effects Effects 0.000 description 6
- 230000001788 irregular Effects 0.000 description 6
- 238000013135 deep learning Methods 0.000 description 5
- 238000003058 natural language processing Methods 0.000 description 5
- 238000003062 neural network model Methods 0.000 description 5
- 230000001131 transforming effect Effects 0.000 description 5
- MHABMANUFPZXEB-UHFFFAOYSA-N O-demethyl-aloesaponarin I Natural products O=C1C2=CC=CC(O)=C2C(=O)C2=C1C=C(O)C(C(O)=O)=C2C MHABMANUFPZXEB-UHFFFAOYSA-N 0.000 description 4
- 230000008901 benefit Effects 0.000 description 3
- 238000013461 design Methods 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- 238000000605 extraction Methods 0.000 description 3
- 238000010801 machine learning Methods 0.000 description 3
- 238000007726 management method Methods 0.000 description 3
- 238000011022 operating instruction Methods 0.000 description 3
- 239000007787 solid Substances 0.000 description 3
- 238000012546 transfer Methods 0.000 description 3
- 230000009471 action Effects 0.000 description 2
- 230000003190 augmentative effect Effects 0.000 description 2
- 239000000872 buffer Substances 0.000 description 2
- 238000010276 construction Methods 0.000 description 2
- 238000013480 data collection Methods 0.000 description 2
- 238000002474 experimental method Methods 0.000 description 2
- 230000007774 longterm Effects 0.000 description 2
- 238000010295 mobile communication Methods 0.000 description 2
- 238000012544 monitoring process Methods 0.000 description 2
- 230000003287 optical effect Effects 0.000 description 2
- 230000002093 peripheral effect Effects 0.000 description 2
- 230000003068 static effect Effects 0.000 description 2
- 238000000844 transformation Methods 0.000 description 2
- 238000013519 translation Methods 0.000 description 2
- 230000001133 acceleration Effects 0.000 description 1
- 238000004458 analytical method Methods 0.000 description 1
- 230000006399 behavior Effects 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 230000015572 biosynthetic process Effects 0.000 description 1
- 210000004027 cell Anatomy 0.000 description 1
- 230000000052 comparative effect Effects 0.000 description 1
- 239000000470 constituent Substances 0.000 description 1
- 238000011217 control strategy Methods 0.000 description 1
- 238000006073 displacement reaction Methods 0.000 description 1
- 238000005538 encapsulation Methods 0.000 description 1
- 238000000802 evaporation-induced self-assembly Methods 0.000 description 1
- 239000012014 frustrated Lewis pair Substances 0.000 description 1
- 239000011521 glass Substances 0.000 description 1
- 230000008676 import Effects 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 230000010354 integration Effects 0.000 description 1
- 230000002452 interceptive effect Effects 0.000 description 1
- 208000013409 limited attention Diseases 0.000 description 1
- 238000012886 linear function Methods 0.000 description 1
- 239000007788 liquid Substances 0.000 description 1
- 239000013307 optical fiber Substances 0.000 description 1
- 230000000737 periodic effect Effects 0.000 description 1
- 230000002085 persistent effect Effects 0.000 description 1
- 238000007781 pre-processing Methods 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 238000012216 screening Methods 0.000 description 1
- 239000004065 semiconductor Substances 0.000 description 1
- 230000035807 sensation Effects 0.000 description 1
- 238000012163 sequencing technique Methods 0.000 description 1
- 238000010897 surface acoustic wave method Methods 0.000 description 1
- 230000001960 triggered effect Effects 0.000 description 1
- 230000000007 visual effect Effects 0.000 description 1
Landscapes
- Image Analysis (AREA)
Abstract
一种数据处理方法,可以应用于人工智能领域,方法包括:获取第一特征表示和第二特征表示,所述第一特征表示为根据注意力网络通过对图像中的多个图像块中不同图像块之间的信息交互得到;所述第二特征表示为基于注意力机制通过对所述图像块中的多个子图像块中不同子图像块之间的信息交互得到;将所述第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示;根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务。本申请通过提取图像数据的全局特征和局部细粒度特征,提高了模型整体的精度。
Description
技术领域
本申请涉及人工智能领域,尤其涉及一种数据处理方法及相关装置。
背景技术
人工智能(artificial intelligence,AI)是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能,感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。换句话说,人工智能是计算机科学的一个分支,它企图了解智能的实质,并生产出一种新的能以人类智能相似的方式作出反应的智能机器。人工智能也就是研究各种智能机器的设计原理与实现方法,使机器具有感知、推理与决策的功能。
随着自注意力结构Transformer从自然语言处理领域迁移到了计算机视觉中,Transformer由于其不受感受野的限制以及更加强大的特征提取能力,在图像处理任务,例如分类,目标检测中取得了更好的效果。
在现有的实现中,对于基于Tranformer实现的图像处理网络,如ViDT或者YOLOS,典型的策略是将图像分成规则的图像块Patches,然后对这些Patches做相互的全局注意力。与受小感受野尺度限制的卷积神经网络相比,基于Transformer的模型的优点是能够感知所有Patches之间的长距离依赖关系。然而,对于现有的基于Tranformer的检测器,仅仅考虑到全局特征表示,限制了它们的多尺度感知,导致模型的精度较差。
发明内容
本申请提供了一种数据处理方法,可以提高模型整体的精度。
第一方面,本申请实施例提供了一种数据处理方法,所述方法包括:获取第一特征表示和第二特征表示,所述第一特征表示为根据注意力网络通过对图像中的多个图像块中不同图像块之间的信息交互得到;所述第二特征表示为基于注意力机制通过对所述图像块中的多个子图像块中不同子图像块之间的信息交互得到;将所述第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示;根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务。
本申请通过提取图像数据的全局特征和局部细粒度特征,考虑到通用的图像处理任务,存在各种大小的目标,细粒度特征有助于识别多尺度、不规则的物体,进而可以提高了模型整体的精度。
在一种可能的实现中,所述方法还包括:对所述图像进行划分,得到所述多个图像块;对所述多个图像块中的至少一个图像块进行划分,得到所述多个子图像块。
在一种可能的实现中,所述目标任务为目标检测或者图像分割。
在一种可能的实现中,可以将第一特征表示融合至第二特征表示,也就是将第一特征表示融合至第二特征表示,得到第一融合特征表示。为提升细粒度特征的表达能力,本申请实施例将其与感受野更大的全局粗粒度特征交互,这样将全局信息的感知带到了内部的特征里,如果不引入它,每一个Patch内部所对应的细粒度特征只能在一个固定的Patch内部区域内获得自注意力感知,无法与其他外部Patch的有效特征做交互。
在一种可能的实现中,所述第一特征表示和所述第二特征表示的尺寸不同,在将第一特征表示融合至第二特征表示时,可以通过对所述第一特征表示进行尺寸变换(例如通过卷积以及上采样实现),得到第三特征表示,所述第三特征表示和所述第二特征表示的尺寸相同;所述第三特征表示用于和所述第二特征表示进行融合得到所述第一融合特征表示。
在一种可能的实现中,可以将第二特征表示融合至第一特征表示,也就是将第二特征表示融合至第一特征表示,得到第二融合特征表示。为了提升全局粗粒度特征的表达能力,本申请实施例将其更加Patch内部的细粒度的特征与其进行融合,从而获得了更加细粒度的感知能力。
在一种可能的实现中,所述第一特征表示和所述第二特征表示的尺寸不同,在将第二特征表示融合至第一特征表示时,可以通过对所述第二特征表示进行尺寸变换(例如通过卷积实现),得到第四特征表示,所述第四特征表示和所述第一特征表示的尺寸相同;所述第四特征表示用于和所述第一特征表示进行融合得到所述第二融合特征表示。
在一种可能的实现中,所述目标任务为与目标对象相关的目标检测或者图像分割;所述根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务,包括:通过对目标对象对应的检测向量(det token)和所述融合特征表示进行信息交互,得到输出特征;根据所述输出特征,通过下游任务网络,对所述图像执行目标任务,以得到处理结果。
在一种可能的实现中,上述融合特征表示可以包括第一融合特征表示(细粒度特征表示)和第二融合特征表示(粗粒度特征表示),由于现有的基于Transformer的图像处理模型只考虑了全局粗粒度的特征,并没有考虑到细粒度的特征。因此在执行下游任务时,也都是直接将检测向量Det Tokens与粗粒度特征进行交互。而本申请实施例中由于利用了其中的细粒度的特征,因此,可以将检测向量Det Tokens分别与细粒度特征和粗粒度特征进行交互。这样对于不同类型的特征交互方案,可以提升Det Tokens的感受丰富度,提升整体的图像处理效果。
在一种可能的实现中,可以将检测向量Det Tokens先与细粒度特征进行交互,之后再和粗粒度特征进行交互。具体的,可以通过对目标对象对应的检测向量和所述第一融合特征表示进行信息交互,得到交互结果,通过对所述交互结果和所述第二融合特征表示进行信息交互,得到输出特征。
在一种可能的实现中,可以将检测向量Det Tokens先与粗粒度特征进行交互,之后再和细粒度特征进行交互。具体的,可以通过对目标对象对应的检测向量和所述第二融合特征表示进行信息交互,得到交互结果,通过对所述交互结果和所述第一融合特征表示进行信息交互,得到输出特征。
此外,为了提高算法的计算效率,降低计算开销,可以在进行检测向量与粗粒度特征或者细粒度特征的信息交互时,将检测向量与粗粒度特征或者细粒度特征中的部分关键特征(也就是多个采样特征点)进行信息交互。
在一种可能的实现中,可以通过对目标对象对应的检测向量和所述融合特征表示中的多个采样特征点进行信息交互。
第二方面,本申请提供了一种数据处理方法,所述方法包括:
处理模块,用于获取第一特征表示和第二特征表示,所述第一特征表示为根据注意力网络通过对图像中的多个图像块中不同图像块之间的信息交互得到;所述第二特征表示为基于注意力机制通过对所述图像块中的多个子图像块中不同子图像块之间的信息交互得到;
将所述第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示;
任务执行模块,用于根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务。
本申请通过提取图像数据的全局特征和局部细粒度特征,考虑到通用的图像处理任务,存在各种大小的目标,细粒度特征有助于识别多尺度、不规则的物体,进而可以提高了模型整体的精度。
在一种可能的实现中,所述处理模块,还用于:
对所述图像进行划分,得到所述多个图像块;
对所述多个图像块中的至少一个图像块进行划分,得到所述多个子图像块。
在一种可能的实现中,所述目标任务为目标检测或者图像分割。
在一种可能的实现中,所述融合特征表示包括第一融合特征表示和第二融合特征表示;
所述处理模块,具体用于:
通过对所述第一特征表示进行尺寸变换,得到第三特征表示,所述第三特征表示和所述第二特征表示的尺寸相同;所述第三特征表示用于和所述第二特征表示进行融合得到所述第一融合特征表示。为提升细粒度特征的表达能力,本申请实施例将其与感受野更大的全局粗粒度特征交互,这样将全局信息的感知带到了内部的特征里,如果不引入它,每一个Patch内部所对应的细粒度特征只能在一个固定的Patch内部区域内获得自注意力感知,无法与其他外部Patch的有效特征做交互。
在一种可能的实现中,所述融合特征表示包括第一融合特征表示和第二融合特征表示;
所述处理模块,具体用于:
通过对所述第二特征表示进行尺寸变换,得到第四特征表示,所述第四特征表示和所述第一特征表示的尺寸相同;所述第四特征表示用于和所述第一特征表示进行融合得到所述第二融合特征表示。
为了提升全局粗粒度特征的表达能力,本申请实施例将其更加Patch内部的细粒度的特征与其进行融合,从而获得了更加细粒度的感知能力。
在一种可能的实现中,所述目标任务为与目标对象相关的目标检测或者图像分割;
所述任务执行模块,具体用于:
通过对目标对象对应的检测向量(det token)和所述融合特征表示进行信息交互,得到输出特征;
根据所述输出特征,通过下游任务网络,对所述图像执行目标任务,以得到处理结果。
在一种可能的实现中,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述第一融合特征表示进行信息交互,得到交互结果;
通过对所述交互结果和所述第二融合特征表示进行信息交互,得到输出特征。
在一种可能的实现中,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述第二融合特征表示进行信息交互,得到交互结果;
通过对所述交互结果和所述第一融合特征表示进行信息交互,得到输出特征。
在一种可能的实现中,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述融合特征表示中的多个采样特征点进行信息交互。
第三方面,本申请实施例提供了一种数据处理装置,可以包括存储器、处理器以及总线系统,其中,存储器用于存储程序,处理器用于执行存储器中的程序,以执行如上述第一方面任一可选的方法。
第四方面,本申请实施例提供了一种计算机可读存储介质,计算机可读存储介质中存储有计算机程序,当其在计算机上运行时,使得计算机执行上述第一方面及任一可选的方法。
第五方面,本申请实施例提供了一种计算机程序产品,包括代码,当代码被执行时,用于实现上述第一方面及任一可选的方法。
第六方面,本申请提供了一种芯片系统,该芯片系统包括处理器,用于支持数据处理装置实现上述方面中所涉及的功能,例如,发送或处理上述方法中所涉及的数据;或,信息。在一种可能的设计中,芯片系统还包括存储器,所述存储器,用于保存执行设备或训练设备必要的程序指令和数据。该芯片系统,可以由芯片构成,也可以包括芯片和其他分立器件。
附图说明
图1A为人工智能主体框架的一种结构示意图;
图1B和图2为本发明的应用系统框架示意;
图3为终端的一种可选的硬件结构示意图;
图4为一种服务器的结构示意图;
图5为本申请的一种系统架构示意;
图6为一种云服务的流程;
图7为本申请实施例中的一种神经网络模型的结构示意;
图8为本申请实施例中的一种神经网络模型的结构示意;
图9为一种transformer层的架构示意;
图10为一种数据处理方法的流程示意;
图11为一种数据处理方法的示意;
图12为一种transformer层的结构示意;
图13为一种transformer层的结构示意;
图14为一个注意力头head的操作示意图;
图15为一个处理流程示意图;
图16为一个效果示意图;
图17为本申请实施例提供的一种数据处理装置的结构示意图;
图18为本申请实施例提供的一种执行设备的示意图;
图19为本申请实施例提供的一种训练设备的示意图;
图20为本申请实施例提供的一种芯片的示意图。
具体实施方式
下面结合本发明实施例中的附图对本发明实施例进行描述。本发明的实施方式部分使用的术语仅用于对本发明的具体实施例进行解释,而非旨在限定本发明。
下面结合附图,对本申请的实施例进行描述。本领域普通技术人员可知,随着技术的发展和新场景的出现,本申请实施例提供的技术方案对于类似的技术问题,同样适用。
本申请的说明书和权利要求书及上述附图中的术语“第一”、“第二”等是用于区别类似的对象,而不必用于描述特定的顺序或先后次序。应该理解这样使用的术语在适当情况下可以互换,这仅仅是描述本申请的实施例中对相同属性的对象在描述时所采用的区分方式。此外,术语“包括”和“具有”以及他们的任何变形,意图在于覆盖不排他的包含,以便包含一系列单元的过程、方法、系统、产品或设备不必限于那些单元,而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它单元。
首先对人工智能系统总体工作流程进行描述,请参见图1A,图1A示出的为人工智能主体框架的一种结构示意图,下面从“智能信息链”(水平轴)和“IT价值链”(垂直轴)两个维度对上述人工智能主题框架进行阐述。其中,“智能信息链”反映从数据的获取到处理的一列过程。举例来说,可以是智能信息感知、智能信息表示与形成、智能推理、智能决策、智能执行与输出的一般过程。在这个过程中,数据经历了“数据—信息—知识—智慧”的凝练过程。“IT价值链”从人智能的底层基础设施、信息(提供和处理技术实现)到系统的产业生态过程,反映人工智能为信息技术产业带来的价值。
(1)基础设施
基础设施为人工智能系统提供计算能力支持,实现与外部世界的沟通,并通过基础平台实现支撑。通过传感器与外部沟通;计算能力由智能芯片(CPU、NPU、GPU、ASIC、FPGA等硬件加速芯片)提供;基础平台包括分布式计算框架及网络等相关的平台保障和支持,可以包括云存储和计算、互联互通网络等。举例来说,传感器和外部沟通获取数据,这些数据提供给基础平台提供的分布式计算系统中的智能芯片进行计算。
(2)数据
基础设施的上一层的数据用于表示人工智能领域的数据来源。数据涉及到图形、图像、语音、文本,还涉及到传统设备的物联网数据,包括已有系统的业务数据以及力、位移、液位、温度、湿度等感知数据。
(3)数据处理
数据处理通常包括数据训练,机器学习,深度学习,搜索,推理,决策等方式。
其中,机器学习和深度学习可以对数据进行符号化和形式化的智能信息建模、抽取、预处理、训练等。
推理是指在计算机或智能系统中,模拟人类的智能推理方式,依据推理控制策略,利用形式化的信息进行机器思维和求解问题的过程,典型的功能是搜索与匹配。
决策是指智能信息经过推理后进行决策的过程,通常提供分类、排序、预测等功能。
(4)通用能力
对数据经过上面提到的数据处理后,进一步基于数据处理的结果可以形成一些通用的能力,比如可以是算法或者一个通用系统,例如,翻译,文本的分析,计算机视觉的处理,语音识别,图像的识别等等。
(5)智能产品及行业应用
智能产品及行业应用指人工智能系统在各领域的产品和应用,是对人工智能整体解决方案的封装,将智能信息决策产品化、实现落地应用,其应用领域主要包括:智能终端、智能交通、智能医疗、自动驾驶、智慧城市等。
本申请实施例可以应用在图像处理的相关任务,例如目标检测、图像分割等领域中。
接下来首先介绍本申请的应用场景,本申请可以但不限于应用在具备图像处理功能的应用程序或者云侧服务器提供的云服务等,接下来分别进行介绍:
一、图像处理类应用程序
本申请实施例的产品形态可以为图像处理类应用程序。图像处理类应用程序可以运行在终端设备或者云侧的服务器上。
在一种可能的实现中,图像处理类应用程序可以实现图像处理任务或者基于图像处理结果进行的任务等。
在一种可能的实现中,用户可以打开终端设备上安装的具备图像处理类功能的应用程序,应用程序可以获取到相机采集的图像数据或者是用户指定的图像数据,图像处理类应用程序可以通过本申请实施例提供的方法基于输入数据得到处理结果,并将图像处理结果或者是基于图像处理结果实现的下游任务结果呈现给用户(呈现方式可以但不限于是显示、保存、上传到云侧等)。
在一种可能的实现中,用户可以打开终端设备上安装的图像处理类应用程序,应用程序可以获取到相机采集的图像数据或者是用户指定的图像数据,图像处理类应用程序可以将数据(或者对该数据进行一定处理后得到的结果)发送至云侧的服务器,云侧的服务器通过本申请实施例提供的方法基于图像生成图像处理结果,并将图像处理结果或者是基于图像处理结果实现的下游任务结果回传至终端设备,终端设备可以将图像处理结果或者是基于图像处理结果实现的下游任务结果呈现给用户(呈现方式可以但不限于是显示、保存、上传到云侧等)。
接下来分别从功能架构以及实现功能的产品架构介绍本申请实施例中的图像处理类应用程序。
参照图1B,图1B为本申请实施例中图像处理类应用程序的功能架构示意:
在一种可能的实现中,如图1B所示,图像处理类应用程序102可接收输入的数据101(例如图像以及事件数据)且产生处理结果103。图像处理类应用程序102可在(举例来说)至少一个计算机系统上执行,且包括计算机代码,所述计算机代码在由一或多个计算机执行时致使所述计算机执行用于执行本文中所描述的图像处理方法。
参照图2,图2为本申请实施例中运行图像处理类应用程序的实体架构示意:
参见图2,图2示出了一种系统架构示意图。该系统可以包括终端100、以及服务器200。其中,服务器200可以包括一个或者多个服务器(图2中以包括一个服务器作为示例进行说明),服务器200可以为一个或者多个终端提供图像处理服务或者基于图像处理结果进行下游任务。
其中,终端100上可以安装有图像处理类应用程序,或者打开与图像处理或者基于图像处理结果进行下游任务相关的网页,上述应用程序和网页可以提供一个界面,终端100可以接收用户在图像处理或者基于图像处理结果进行下游任务界面上输入的相关参数,并将上述参数发送至服务器200,服务器200可以基于接收到的参数,得到处理结果,并将处理结果返回至至终端100。
应理解,在一些可选的实现中,终端100也可以由自身完成基于接收到的参数,得到数据处理结果,而不需要服务器配合实现,本申请实施例并不限定。
接下来描述图2中终端100的产品形态;
本申请实施例中的终端100可以为手机、平板电脑、可穿戴设备、车载设备、增强现实(augmented reality,AR)/虚拟现实(virtual reality,VR)设备、笔记本电脑、超级移动个人计算机(ultra-mobile personal computer,UMPC)、上网本、个人数字助理(personaldigital assistant,PDA)等,本申请实施例对此不作任何限制。
图3示出了终端100的一种可选的硬件结构示意图。
参考图3所示,终端100可以包括射频单元110、存储器120、输入单元130、显示单元140、摄像头150(可选的)、音频电路160(可选的)、扬声器161(可选的)、麦克风162(可选的)、处理器170、外部接口180、电源190等部件。本领域技术人员可以理解,图3仅仅是终端或多功能设备的举例,并不构成对终端或多功能设备的限定,可以包括比图示更多或更少的部件,或者组合某些部件,或者不同的部件。
输入单元130可用于接收输入的数字或字符信息,以及产生与该便携式多功能装置的用户设置以及功能控制有关的键信号输入。具体地,输入单元130可包括触摸屏131(可选的)和/或其他输入设备132。该触摸屏131可收集用户在其上或附近的触摸操作(比如用户使用手指、关节、触笔等任何适合的物体在触摸屏上或在触摸屏附近的操作),并根据预先设定的程序驱动相应的连接装置。触摸屏可以检测用户对触摸屏的触摸动作,将该触摸动作转换为触摸信号发送给该处理器170,并能接收该处理器170发来的命令并加以执行;该触摸信号至少包括触点坐标信息。该触摸屏131可以提供该终端100和用户之间的输入界面和输出界面。此外,可以采用电阻式、电容式、红外线以及表面声波等多种类型实现触摸屏。除了触摸屏131,输入单元130还可以包括其他输入设备。具体地,其他输入设备132可以包括但不限于物理键盘、功能键(比如音量控制按键、开关按键等)、轨迹球、鼠标、操作杆等中的一种或多种。
其中,其他输入设备132可以获取相机采集的图像数据或者用户指定的图像数据等等。
该显示单元140可用于显示由用户输入的信息或提供给用户的信息、终端100的各种菜单、交互界面、文件显示和/或任意一种多媒体文件的播放。在本申请实施例中,显示单元140可用于显示和图像处理相关的应用程序的界面等。
该存储器120可用于存储指令和数据,存储器120可主要包括存储指令区和存储数据区,存储数据区可存储各种数据,如多媒体文件、文本等;存储指令区可存储操作系统、应用、至少一个功能所需的指令等软件单元,或者他们的子集、扩展集。还可以包括非易失性随机存储器;向处理器170提供包括管理计算处理设备中的硬件、软件以及数据资源,支持控制软件和应用。还用于多媒体文件的存储,以及运行程序和应用的存储。
处理器170是终端100的控制中心,利用各种接口和线路连接整个终端100的各个部分,通过运行或执行存储在存储器120内的指令以及调用存储在存储器120内的数据,执行终端100的各种功能和处理数据,从而对终端设备进行整体控制。可选的,处理器170可包括一个或多个处理单元;优选的,处理器170可集成应用处理器和调制解调处理器,其中,应用处理器主要处理操作系统、用户界面和应用程序等,调制解调处理器主要处理无线通信。可以理解的是,上述调制解调处理器也可以不集成到处理器170中。在一些实施例中,处理器、存储器、可以在单一芯片上实现,在一些实施例中,他们也可以在独立的芯片上分别实现。处理器170还可以用于产生相应的操作控制信号,发给计算处理设备相应的部件,读取以及处理软件中的数据,尤其是读取和处理存储器120中的数据和程序,以使其中的各个功能模块执行相应的功能,从而控制相应的部件按指令的要求进行动作。
其中,存储器120可以用于存储图像处理方法相关的软件代码,处理器170可以执行芯片的图像处理方法的步骤,也可以调度其他单元(例如上述输入单元130以及显示单元140)以实现相应的功能。
该射频单元110(可选的)可用于收发信息或通话过程中信号的接收和发送,例如,将基站的下行信息接收后,给处理器170处理;另外,将设计上行的数据发送给基站。通常,RF电路包括但不限于天线、至少一个放大器、收发信机、耦合器、低噪声放大器(Low NoiseAmplifier,LNA)、双工器等。此外,射频单元110还可以通过无线通信与网络设备和其他设备通信。该无线通信可以使用任一通信标准或协议,包括但不限于全球移动通讯系统(Global System of Mobile communication,GSM)、通用分组无线服务(General PacketRadio Service,GPRS)、码分多址(Code Division Multiple Access,CDMA)、宽带码分多址(Wideband Code Division Multiple Access,WCDMA)、长期演进(Long Term Evolution,LTE)、电子邮件、短消息服务(Short Messaging Service,SMS)等。
其中,在本申请实施例中,该射频单元110可以将图像数据发送至服务器200,并接收到服务器200发送的处理结果的信息。
应理解,该射频单元110为可选的,其可以被替换为其他通信接口,例如可以是网口。
终端100还包括给各个部件供电的电源190(比如电池),优选的,电源可以通过电源管理系统与处理器170逻辑相连,从而通过电源管理系统实现管理充电、放电、以及功耗管理等功能。
终端100还包括外部接口180,该外部接口可以是标准的Micro USB接口,也可以使多针连接器,可以用于连接终端100与其他装置进行通信,也可以用于连接充电器为终端100充电。
尽管未示出,终端100还可以包括闪光灯、无线保真(wireless fidelity,WiFi)模块、蓝牙模块、不同功能的传感器等,在此不再赘述。下文中描述的部分或全部方法均可以应用在如图3所示的终端100中。
接下来描述图2中服务器200的产品形态;
图4提供了一种服务器200的结构示意图,如图4所示,服务器200包括总线201、处理器202、通信接口203和存储器204。处理器202、存储器204和通信接口203之间通过总线201通信。
总线201可以是外设部件互连标准(peripheral component interconnect,PCI)总线或扩展工业标准结构(extended industry standard architecture,EISA)总线等。总线可以分为地址总线、数据总线、控制总线等。为便于表示,图4中仅用一条粗线表示,但并不表示仅有一根总线或一种类型的总线。
处理器202可以为中央处理器(central processing unit,CPU)、图形处理器(graphics processing unit,GPU)、微处理器(micro processor,MP)或者数字信号处理器(digital signal processor,DSP)等处理器中的任意一种或多种。
存储器204可以包括易失性存储器(volatile memory),例如随机存取存储器(random access memory,RAM)。存储器204还可以包括非易失性存储器(non-volatilememory),例如只读存储器(read-only memory,ROM),快闪存储器,机械硬盘(hard drivedrive,HDD)或固态硬盘(solid state drive,SSD)。
其中,存储器204可以用于存储图像处理方法相关的软件代码,处理器202可以执行芯片的图像处理方法的步骤,也可以调度其他单元以实现相应的功能。
应理解,上述终端100和服务器200可以为集中式或者是分布式的设备,上述终端100和服务器200中的处理器(例如处理器170以及处理器202)可以为硬件电路(如专用集成电路(application specific integrated circuit,ASIC)、现场可编程门阵列(field-programmable gate array,FPGA)、通用处理器、数字信号处理器(digital signalprocessing,DSP)、微处理器或微控制器等等)、或这些硬件电路的组合,例如,处理器可以为具有执行指令功能的硬件系统,如CPU、DSP等,或者为不具有执行指令功能的硬件系统,如ASIC、FPGA等,或者为上述不具有执行指令功能的硬件系统以及具有执行指令功能的硬件系统的组合。
应理解,本申请实施例中的和模型推理过程相关的步骤涉及AI相关的运算,在执行AI运算时,终端设备和服务器的指令执行架构不仅仅局限在上述介绍的处理器结合存储器的架构。下面结合图5对本申请实施例提供的系统架构进行详细的介绍。
请参阅图5,图5为本申请实施例提供的系统的一种系统架构图,在图5中,任务处理系统500包括执行设备510、训练设备520、数据库530、客户设备540、数据存储系统550和数据采集设备560,执行设备510中包括计算模块511。其中,数据采集设备560用于获取用户需要的开源的大规模数据集(即训练集),并将训练集存入数据库530中,训练设备520基于数据库530中的维护的训练集对目标模型/规则501进行训练,训练得到的训练后的神经网络再在执行设备510上进行运用。执行设备510可以调用数据存储系统550中的数据、代码等,也可以将数据、指令等存入数据存储系统550中。数据存储系统550可以置于执行设备510中,也可以为数据存储系统550相对执行设备510是外部存储器。
经由训练设备520训练的目标模型/规则501后得到的训练后的神经网络可以应用于不同的系统或设备(即执行设备510)中,具体可以是边缘设备或端侧设备,例如,手机、平板、笔记本电脑、监控系统(如,摄像头)、安防系统等等。在图5中,执行设备510配置有I/O接口512,与外部设备进行数据交互,“用户”可以通过客户设备540向I/O接口512输入数据。如,客户设备540可以是监控系统的摄像设备,通过该摄像设备拍摄的图像以及事件数据作为输入数据输入至执行设备510的计算模块511,由计算模块511对输入的该目标图像进行处理后得到处理结果,再将该处理结果输出至摄像设备或直接在执行设备510的显示界面(若有)进行显示;此外,在本申请的一些实施方式中,客户设备540也可以集成在执行设备510中,如,当执行设备510为手机时,则可以直接通过该手机获取到目标任务(如,可以通过该手机的摄像头拍摄到图像以及事件数据等)或者接收其他设备(如,另一个手机)发送的目标任务,再由该手机内的计算模块511对该目标任务进行检测后得出检测结果,并直接将该检测结果呈现在手机的显示界面。此处对执行设备510与客户设备540的产品形态不做限定。
值得注意的是,图5仅是本申请实施例提供的一种系统架构的示意图,图中所示设备、器件、模块等之间的位置关系不构成任何限制,例如,在图5中,数据存储系统550相对执行设备510是外部存储器,在其它情况下,也可以将数据存储系统550置于执行设备510中。应理解,上述执行设备510可以部署于客户设备540中。
从模型的推理侧来说:
本申请实施例中,上述执行设备520的计算模块511可以获取到数据存储系统550中存储的代码来实现本申请实施例中的和模型推理过程相关的步骤。
本申请实施例中,执行设备520的计算模块511可以包括硬件电路(如专用集成电路(application specific integrated circuit,ASIC)、现场可编程门阵列(field-programmable gate array,FPGA)、通用处理器、数字信号处理器(digital signalprocessing,DSP)、微处理器或微控制器等等)、或这些硬件电路的组合,例如,训练设备520可以为具有执行指令功能的硬件系统,如CPU、DSP等,或者为不具有执行指令功能的硬件系统,如ASIC、FPGA等,或者为上述不具有执行指令功能的硬件系统以及具有执行指令功能的硬件系统的组合。
具体的,执行设备520的计算模块511可以为具有执行指令功能的硬件系统,本申请实施例提供的和模型推理过程相关的步骤可以为存储在存储器中的软件代码,执行设备520的计算模块511可以从存储器中获取到软件代码,并执行获取到的软件代码来实现本申请实施例提供的和模型推理过程相关的步骤。
应理解,执行设备520的计算模块511可以为不具有执行指令功能的硬件系统以及具有执行指令功能的硬件系统的组合,本申请实施例提供的和模型推理过程相关的步骤的部分步骤还可以通过执行设备520的计算模块511中不具有执行指令功能的硬件系统来实现,这里并不限定。
从模型的训练侧来说:
本申请实施例中,上述训练设备520可以获取到存储器(图5中未示出,可以集成于训练设备520或者与训练设备520分离部署)中存储的代码来实现本申请实施例中和模型训练相关的步骤。
本申请实施例中,训练设备520可以包括硬件电路(如专用集成电路(applicationspecific integrated circuit,ASIC)、现场可编程门阵列(field-programmable gatearray,FPGA)、通用处理器、数字信号处理器(digital signal processing,DSP)、微处理器或微控制器等等)、或这些硬件电路的组合,例如,训练设备520可以为具有执行指令功能的硬件系统,如CPU、DSP等,或者为不具有执行指令功能的硬件系统,如ASIC、FPGA等,或者为上述不具有执行指令功能的硬件系统以及具有执行指令功能的硬件系统的组合。
应理解,训练设备520可以为不具有执行指令功能的硬件系统以及具有执行指令功能的硬件系统的组合,本申请实施例提供的中和模型训练相关的部分步骤还可以通过训练设备520中不具有执行指令功能的硬件系统来实现,这里并不限定。
二、服务器提供的图像处理类云服务:
在一种可能的实现中,服务器可以通过应用程序编程接口(applicationprogramming interface,API)为端侧提供图像处理或者基于图像处理结果进行下游任务的服务。
其中,终端设备可以通过云端提供的API,将相关参数(例如图像数据)发送至服务器,服务器可以基于接收到的参数,得到处理结果,并将处理结果返回至至终端。
关于终端以及服务器的描述可以上述实施例的描述,这里不再赘述。
如图6示出了使用一项云平台提供的图像处理类云服务的流程。
1.开通并购买内容审核服务。
2.用户可以下载内容审核服务对应的软件开发工具包(software developmentkit,SDK),通常云平台提供多个开发版本的SDK,供用户根据开发环境的需求选择,例如JAVA版本的SDK、python版本的SDK、PHP版本的SDK、Android版本的SDK等。
3.用户根据需求下载对应版本的SDK到本地后,将SDK工程导入至本地开发环境,在本地开发环境中进行配置和调试,本地开发环境还可以进行其他功能的开发,使得形成一个集合了图像处理类能力的应用。
4.图像处理类应用在被使用的过程中,当需要进行图像处理或者基于图像处理结果进行下游任务时,可以触发图像处理或者基于图像处理结果进行下游任务的API调用。当应用触发图像处理或者基于图像处理结果进行下游任务功能时,发起API请求至云环境中的图像处理类服务的运行实例,其中,API请求中携带图像,由云环境中的运行实例对图像进行处理,获得处理结果。
5.云环境将处理结果返回至应用,由此完成一次的图像处理或者基于图像处理结果进行下游任务服务调用。
由于本申请实施例涉及大量神经网络的应用,为了便于理解,下面先对本申请实施例涉及的相关术语及神经网络等相关概念进行介绍。
(1)神经网络
神经网络可以是由神经单元组成的,神经单元可以是指以xs(即输入数据)和截距1为输入的运算单元,该运算单元的输出可以为:
其中,s=1、2、……n,n为大于1的自然数,Ws为xs的权重,b为神经单元的偏置。f为神经单元的激活函数(activation functions),用于将非线性特性引入神经网络中,来将神经单元中的输入信号转换为输出信号。该激活函数的输出信号可以作为下一层卷积层的输入,激活函数可以是sigmoid函数。神经网络是将多个上述单一的神经单元联结在一起形成的网络,即一个神经单元的输出可以是另一个神经单元的输入。每个神经单元的输入可以与前一层的局部接受域相连,来提取局部接受域的特征,局部接受域可以是由若干个神经单元组成的区域。
(2)卷积神经网络(convolutional neuron network,CNN)是一种带有卷积结构的深度神经网络。卷积神经网络包含了一个由卷积层和子采样层构成的特征抽取器,该特征抽取器可以看作是滤波器。卷积层是指卷积神经网络中对输入信号进行卷积处理的神经元层。在卷积神经网络的卷积层中,一个神经元可以只与部分邻层神经元连接。一个卷积层中,通常包含若干个特征平面,每个特征平面可以由一些矩形排列的神经单元组成。同一特征平面的神经单元共享权重,这里共享的权重就是卷积核。共享权重可以理解为提取特征的方式与位置无关。卷积核可以以随机大小的矩阵的形式化,在卷积神经网络的训练过程中卷积核可以通过学习得到合理的权重。另外,共享权重带来的直接好处是减少卷积神经网络各层之间的连接,同时又降低了过拟合的风险。
CNN是一种非常常见的神经网络,下面结合图7重点对CNN的结构进行详细的介绍。如前文的基础概念介绍所述,卷积神经网络是一种带有卷积结构的深度神经网络,是一种深度学习(deep learning)架构,深度学习架构是指通过机器学习的算法,在不同的抽象层级上进行多个层次的学习。作为一种深度学习架构,CNN是一种前馈(feed-forward)人工神经网络,该前馈人工神经网络中的各个神经元可以对输入其中的图像作出响应。
如图7所示,卷积神经网络(CNN)200可以包括输入层210,卷积层/池化层220(其中池化层为可选的),以及全连接层(fully connected layer)230。
卷积层/池化层220:
卷积层:
如图7所示卷积层/池化层220可以包括如示例221-226层,举例来说:在一种实现中,221层为卷积层,222层为池化层,223层为卷积层,224层为池化层,225为卷积层,226为池化层;在另一种实现方式中,221、222为卷积层,223为池化层,224、225为卷积层,226为池化层。即卷积层的输出可以作为随后的池化层的输入,也可以作为另一个卷积层的输入以继续进行卷积操作。
下面将以卷积层221为例,介绍一层卷积层的内部工作原理。
卷积层221可以包括很多个卷积算子,卷积算子也称为核,其在图像处理中的作用相当于一个从输入图像矩阵中提取特定信息的过滤器,卷积算子本质上可以是一个权重矩阵,这个权重矩阵通常被预先定义,在对图像进行卷积操作的过程中,权重矩阵通常在输入图像上沿着水平方向一个像素接着一个像素(或两个像素接着两个像素……这取决于步长stride的取值)的进行处理,从而完成从图像中提取特定特征的工作。该权重矩阵的大小应该与图像的大小相关,需要注意的是,权重矩阵的纵深维度(depth dimension)和输入图像的纵深维度是相同的,在进行卷积运算的过程中,权重矩阵会延伸到输入图像的整个深度。因此,和一个单一的权重矩阵进行卷积会产生一个单一纵深维度的卷积化输出,但是大多数情况下不使用单一权重矩阵,而是应用多个尺寸(行×列)相同的权重矩阵,即多个同型矩阵。每个权重矩阵的输出被堆叠起来形成卷积图像的纵深维度,这里的维度可以理解为由上面所述的“多个”来决定。不同的权重矩阵可以用来提取图像中不同的特征,例如一个权重矩阵用来提取图像边缘信息,另一个权重矩阵用来提取图像的特定颜色,又一个权重矩阵用来对图像中不需要的噪点进行模糊化等。该多个权重矩阵尺寸(行×列)相同,经过该多个尺寸相同的权重矩阵提取后的特征图的尺寸也相同,再将提取到的多个尺寸相同的特征图合并形成卷积运算的输出。
这些权重矩阵中的权重值在实际应用中需要经过大量的训练得到,通过训练得到的权重值形成的各个权重矩阵可以用来从输入图像中提取信息,从而使得卷积神经网络200进行正确的预测。
当卷积神经网络200有多个卷积层的时候,初始的卷积层(例如221)往往提取较多的一般特征,该一般特征也可以称之为低级别的特征;随着卷积神经网络200深度的加深,越往后的卷积层(例如226)提取到的特征越来越复杂,比如高级别的语义之类的特征,语义越高的特征越适用于待解决的问题。
池化层:
由于常常需要减少训练参数的数量,因此卷积层之后常常需要周期性的引入池化层,在如图7中220所示例的221-226各层,可以是一层卷积层后面跟一层池化层,也可以是多层卷积层后面接一层或多层池化层。在图像处理过程中,池化层的唯一目的就是减少图像的空间大小。池化层可以包括平均池化算子和/或最大池化算子,以用于对输入图像进行采样得到较小尺寸的图像。平均池化算子可以在特定范围内对图像中的像素值进行计算产生平均值作为平均池化的结果。最大池化算子可以在特定范围内取该范围内值最大的像素作为最大池化的结果。另外,就像卷积层中用权重矩阵的大小应该与图像尺寸相关一样,池化层中的运算符也应该与图像的大小相关。通过池化层处理后输出的图像尺寸可以小于输入池化层的图像的尺寸,池化层输出的图像中每个像素点表示输入池化层的图像的对应子区域的平均值或最大值。
全连接层230:
在经过卷积层/池化层220的处理后,卷积神经网络200还不足以输出所需要的输出信息。因为如前所述,卷积层/池化层220只会提取特征,并减少输入图像带来的参数。然而为了生成最终的输出信息(所需要的类信息或其他相关信息),卷积神经网络200需要利用全连接层230来生成一个或者一组所需要的类的数量的输出。因此,在全连接层230中可以包括多层隐含层(如图7所示的231、232至23n),该多层隐含层中所包含的参数可以根据具体的任务类型的相关训练数据进行预先训练得到,例如该任务类型可以包括图像识别,图像分类,图像超分辨率重建等等……
在全连接层230中的多层隐含层之后,也就是整个卷积神经网络200的最后层为输出层240,该输出层240具有类似分类交叉熵的损失函数,具体用于计算预测误差,一旦整个卷积神经网络200的前向传播(如图7由210至240方向的传播为前向传播)完成,反向传播(如图7由240至210方向的传播为反向传播)就会开始更新前面提到的各层的权重值以及偏差,以减少卷积神经网络200的损失,及卷积神经网络200通过输出层输出的结果和理想结果之间的误差。
需要说明的是,如图7所示的卷积神经网络200仅作为一种卷积神经网络的示例,在具体的应用中,卷积神经网络还可以以其他网络模型的形式存在,例如,仅包括图7中所示的网络结构的一部分,比如,本申请实施例中所采用的卷积神经网络可以仅包括输入层210、卷积层/池化层220和输出层240。
需要说明的是,如图7所示的卷积神经网络200仅作为一种卷积神经网络的示例,在具体的应用中,卷积神经网络还可以以其他网络模型的形式存在,例如,如图8所示的多个卷积层/池化层并行,将分别提取的特征均输入给全连接层230进行处理。
(3)深度神经网络
深度神经网络(Deep Neural Network,DNN),也称多层神经网络,可以理解为具有很多层隐含层的神经网络,这里的“很多”并没有特别的度量标准。从DNN按不同层的位置划分,DNN内部的神经网络可以分为三类:输入层,隐含层,输出层。一般来说第一层是输入层,最后一层是输出层,中间的层数都是隐含层。层与层之间是全连接的,也就是说,第i层的任意一个神经元一定与第i+1层的任意一个神经元相连。虽然DNN看起来很复杂,但是就每一层的工作来说,其实并不复杂,简单来说就是如下线性关系表达式:其中,是输入向量,/>是输出向量,/>是偏移向量,W是权重矩阵(也称系数),α()是激活函数。每一层仅仅是对输入向量/>经过如此简单的操作得到输出向量/>由于DNN层数多,则系数W和偏移向量/>的数量也就很多了。这些参数在DNN中的定义如下所述:以系数W为例:假设在一个三层的DNN中,第二层的第4个神经元到第三层的第2个神经元的线性系数定义为/>上标3代表系数W所在的层数,而下标对应的是输出的第三层索引2和输入的第二层索引4。总结就是:第L-1层的第k个神经元到第L层的第j个神经元的系数定义为/>需要注意的是,输入层是没有W参数的。在深度神经网络中,更多的隐含层让网络更能够刻画现实世界中的复杂情形。理论上而言,参数越多的模型复杂度越高,“容量”也就越大,也就意味着它能完成更复杂的学习任务。训练深度神经网络的也就是学习权重矩阵的过程,其最终目的是得到训练好的深度神经网络的所有层的权重矩阵(由很多层的向量W形成的权重矩阵)。
(4)transformer层
参照图9,图9为一种transformer层的架构示意,如图9所示,神经网络包括嵌入层和至少一个transformer层,至少一个transformer层可以为N个transformer层(N大于0的整数),其中,每个transformer层包括依次相邻的注意力层、加和与归一化(add&norm)层、前馈(feed forward)层和加和与归一化层。在嵌入层,对当前输入进行嵌入处理,得到多个特征向量;在所述注意力层,从所述transformer层的上一层获取P个输入向量,以P个输入向量中的任意的第一输入向量为中心,基于预设的注意力窗口范围内的各个输入向量与该第一输入向量之间的关联度,得到该第一输入向量对应的中间向量,如此确定出P个输入向量对应的P个中间向量;在所述池化层,将所述P个中间向量合并为Q个输出向量,其中至少一个transformer层中最后一个transformer层得到的多个输出向量用作所述当前输入的特征表示。
接下来,结合具体例子对上述各步骤进行具体介绍。
首先,在所述嵌入层,对当前输入进行嵌入处理,得到多个特征向量。
嵌入层可以称为输入嵌入(input embedding)层。当前输入可以为文本输入,例如可以为一段文本,也可以为一个句子。文本可以为中文文本,也可以为英文文本,还可以为其他语言文本。嵌入层在获取当前输入后,可以对该当前输入中各个词进行嵌入处理,可得到各个词的特征向量。在一些实施例中,如图9所示,所述嵌入层包括输入嵌入层和位置编码(positional encoding)层。在输入嵌入层,可以对当前输入中的各个词进行词嵌入处理,从而得到各个词的词嵌入向量。在位置编码层,可以获取各个词在该当前输入中的位置,进而对各个词的位置生成位置向量。在一些示例中,各个词的位置可以为各个词在该当前输入中的绝对位置。以当前输入为“几号应还花呗”为例,其中的“几”的位置可以表示为第一位,“号”的位置可以表示为第二位,……。在一些示例中,各个词的位置可以为各个词之间的相对位置。仍以当前输入为“几号应还款”为例,其中的“几”的位置可以表示为“号”之前,“号”的位置可以表示为“几”之后、“应”之前,……。当得到当前输入中各个词的词嵌入向量和位置向量时,可以将各个词的位置向量和对应的词嵌入向量进行组合,得到各个词特征向量,即得到该当前输入对应的多个特征向量。多个特征向量可以表示为具有预设维度的嵌入矩阵。可以设定该多个特征向量中的特征向量个数为M,预设维度为H维,则该多个特征向量可以表示为M×H的嵌入矩阵。
其次,从所述第一transformer层的上一层获取P个输入向量,以P个输入向量中的任意的第一输入向量为中心,基于预设的注意力窗口范围内的各个输入向量与该第一输入向量之间的关联度,得到该第一输入向量对应的中间向量,如此确定出P个输入向量对应的P个中间向量。注意力层也可以称为多头注意力(multi-head attention)层。在一个例子中,注意力层可以为固定窗口多头注意力(fixed window multi-head attention)层。
在一些实施例中,第一transformer层可以为上述嵌入层的下一层,P个输入向量为从嵌入层得到的所述多个特征向量。在一些实施例中,本说明书实施例提供的神经网络中的至少一个transformer层还包括第二transformer层。该第二transformer层为第一自注意力的上一层,则P个输入向量为第二transformer层输出的P个输出向量。在该神经网络中的最后一个transformer层,通过上述步骤的多个输出向量可用作当前输入的特征表示。该特征表示为为当前输入的一种适合计算机处理的特征表示。
(5)注意力机制(attention mechanism)
注意力机制模仿了生物观察行为的内部过程,即一种将内部经验和外部感觉对齐从而增加部分区域的观察精细度的机制,能够利用有限的注意力资源从大量信息中快速筛选出高价值信息。注意力机制可以快速提取稀疏数据的重要特征,因而被广泛用于自然语言处理任务,特别是机器翻译。而自注意力机制(self-attention mechanism)是注意力机制的改进,其减少了对外部信息的依赖,更擅长捕捉数据或特征的内部相关性。注意力机制的本质思想可以改写为如下公式:
其中,Lx=||Source||代表Source的长度,公式含义即将Source中的构成元素想象成是由一系列的数据对构成,此时给定目标Target中的某个元素Query,通过计算Query和各个Key的相似性或者相关性,得到每个Key对应Value的权重系数,然后对Value进行加权求和,即得到了最终的Attention数值。所以本质上Attention机制是对Source中元素的Value值进行加权求和,而Query和Key用来计算对应Value的权重系数。从概念上理解,把Attention可以理解为从大量信息中有选择地筛选出少量重要信息并聚焦到这些重要信息上,忽略大多不重要的信息。聚焦的过程体现在权重系数的计算上,权重越大越聚焦于其对应的Value值上,即权重代表了信息的重要性,而Value是其对应的信息。自注意力机制可以理解为内部Attention(intra attention),Attention机制发生在Target的元素Query和Source中的所有元素之间,自注意力机制指的是在Source内部元素之间或者Target内部元素之间发生的Attention机制,也可以理解为Target=Source这种特殊情况下的注意力计算机制,其具体计算过程是一样的,只是计算对象发生了变化而已。
(6)损失函数
在训练深度神经网络的过程中,因为希望深度神经网络的输出尽可能的接近真正想要预测的值,所以可以通过比较当前网络的预测值和真正想要的目标值,再根据两者之间的差异情况来更新每一层神经网络的权重向量(当然,在第一次更新之前通常会有初始化的过程,即为深度神经网络中的各层预先配置参数),比如,如果网络的预测值高了,就调整权重向量让它预测低一些,不断的调整,直到深度神经网络能够预测出真正想要的目标值或与真正想要的目标值非常接近的值。因此,就需要预先定义“如何比较预测值和目标值之间的差异”,这便是损失函数(loss function)或目标函数(objective function),它们是用于衡量预测值和目标值的差异的重要方程。其中,以损失函数举例,损失函数的输出值(loss)越高表示差异越大,那么深度神经网络的训练就变成了尽可能缩小这个loss的过程。
(7)反向传播算法
可以采用误差反向传播(back propagation,BP)算法在训练过程中修正初始模型中参数的大小,使得模型的误差损失越来越小。具体地,前向传递输入信号直至输出会产生误差损失,通过反向传播误差损失信息来更新初始模型中的参数,从而使误差损失收敛。反向传播算法是以误差损失为主导的反向传播运动,旨在得到最优的模型参数,例如权重矩阵。
随着自注意力结构Transformer从自然语言处理领域迁移到了计算机视觉中,Transformer由于其不受感受野的限制以及更加强大的特征提取能力,在图像处理任务,例如分类,目标检测中取得了更好的效果。
在现有的实现中,对于基于Tranformer实现的图像处理网络,如ViDT或者YOLOS,典型的策略是将图像分成规则的图像块Patches,然后对这些Patches做相互的全局注意力。与受小感受野尺度限制的卷积神经网络相比,基于Transformer的模型的优点是能够感知所有Patches之间的长距离依赖关系。然而,这些模型忽略了每个Patch内的空间局部信息。考虑到通用目标检测任务,存在各种大小的目标。这些细粒度特征有助于识别多尺度、不规则的物体。对于现有的基于Tranformer的检测器,仅仅考虑到全局特征表示,限制了它们的多尺度感知,导致模型的精度较差。
为了解决上述问题,本申请提供了一种图像处理方法,该图像处理方法可以为模型训练的前馈过程,也可以为推理过程。
参照图10,图10为本申请实施例提供的一种图像处理方法,如图10所示,本申请提供的图像处理方法,包括:
1001、获取第一特征表示和第二特征表示,所述第一特征表示为根据注意力网络通过对图像中的多个图像块中不同图像块之间的信息交互得到;所述第二特征表示为基于注意力机制通过对所述图像块中的多个子图像块中不同子图像块之间的信息交互得到。
本申请实施例中,步骤1001的执行主体可以为终端设备,终端设备可以为便携式移动设备,例如但不限于移动或便携式计算设备(如智能手机)、个人计算机、服务器计算机、手持式设备(例如平板)或膝上型设备、多处理器系统、游戏控制台或控制器、基于微处理器的系统、机顶盒、可编程消费电子产品、移动电话、具有可穿戴或配件形状因子(例如,手表、眼镜、头戴式耳机或耳塞)的移动计算和/或通信设备、网络PC、小型计算机、大型计算机、包括上面的系统或设备中的任何一种的分布式计算环境等等。
本申请实施例中,步骤1001的执行主体可以为云侧的服务器,服务器可以接收来自终端设备发送的图像数据,进而服务器可以获取到图像数据。
在一种可能的实现中,可以通过注意力网络对图像数据进行处理,以得到第一特征表示和第二特征表示。
在一种可能的实现中,第一特征表示也可以称之为粗粒度特征,具体是通过将图像拆分成规则的图像块Patches,并做不同图像块Patches之间的注意力得到的。例如对于TNT模型来说,粗粒度特征就是一张图像中的每一个Patch与其他Patches做自注意力之后的特征。
在一种可能的实现中,第二特征表示也可以称之为细粒度特征,由于每一个Patch内部也是一个图像块,细粒度特征则是Patch内部包含的图像空间信息。细粒度特征就是各个Patch内部的每一个子Patch之间做自注意力后的特征。对于其他基于Tranfromer的模型,其通常来说只有粗粒度的特征,通过在其上引入细粒度的Patch内部特征对于提升整体精度是必要的。
在一种可能的实现中,可以对所述图像进行划分,得到所述多个图像块;对所述多个图像块中的至少一个图像块进行划分,得到所述多个子图像块。
接下来介绍一个注意力网络的示例:
在一种可能的实现中,参照图12,图12为本申请实施例中的一种神经网络模型的结构示意,如图12中示出的那样,基于transformer层的神经网络模型可以包括依次连接的嵌入层以及多个transformer层。如本领域技术人员所了解,transformer模型多用于执行自然语言处理NLP任务。需要理解,图12的结构仅仅是一个示例,transformer层的数目可以根据需要而设置。神经网络模型基于各transformer层得到的N个输出向量,确定当前节点对应的特征向量。
在嵌入层,对当前输入进行嵌入处理,得到多个特征向量。transformer模型的核心特点在于其采用的独特的注意力机制。参照图13,图13为一种transformer层的结构示意,本申请实施例中的各个神经网络的transformer层都可以参照图13中示出的结构,如图13中示出的那样,transformer层包括依次相邻的多头注意力层、加和与归一化(add&norm)层、前馈(feed forward)层、加和与归一化层。
其中,多头注意力层从其上一层获取N个输入向量Xl,又可以表示为矩阵X,采用自注意力机制,基于向量间的关联度对各个向量进行变换(或者可以称之为交互),得到N个输出向量,又可以表示为矩阵Y。可以理解,当该多头注意力层是与嵌入层直接相连的层,例如图13中与嵌入层直连的transformer层,其获取的输入向量即为嵌入层输出的嵌入向量;当该多头注意力层是后续的transformer层包括的多头注意力层,例如图13中与上一级transformer层直连的transformer层包括的多头注意力层,其获取的输入向量即为前一级transformer层的输出向量。在多头注意力层,基于多头注意力(multi-head attention,MHA)的MHA层包括多个注意力头head(如图14中示出的Head 1、Head 2、…、Head N)。
图14为一个注意力头head的操作示意图,该示意图示出注意力头head如何将输入矩阵X变换为输出矩阵Y。如图14所示,分别采用第一变换矩阵Q,第二变换矩阵K和第三变换矩阵V对N个输入向量<X1,X2,…,XN>中各个输入向量Xi进行变换,得到各个输入向量对应的第一中间向量(q向量),第二中间向量(k向量)和第三中间向量(v向量)。操作上,可以分别用第一变换矩阵Q,第二变换矩阵K和第三变换矩阵V,对N个输入向量构成的输入矩阵X进行线性变换,分别得到输入矩阵的Q矩阵,K矩阵和V矩阵,再分别对矩阵进行拆分,即可得到各个输入向量对应的q向量,k向量和v向量。对于N个输入向量中任意的第i输入向量Xi,基于该第i输入向量对应的第一中间向量(q向量,qi)与各个输入向量Xj对应的各个第二中间向量(k向量,kj)的点乘操作,确定该第i输入向量Xi与各个输入向量Xj的各个关联度。尽管也可以直接将qi与kj的点乘结果确定为关联度,但是更经典地,先将点乘结果除以一常数,然后进行softmax运算,将运算结果作为输入向量Xi与Xj的关联度(也就是Q向量和K向量之间的关联度),即:
于是,可以以该第i输入向量Xi与各个输入向量Xj的各个关联度αi,j作为权重因子,对各个输入向量Xj对应的第三中间向量(v向量,vj)进行加权组合,得到该第i输入向量Xi对应的第i组合向量Ci:
于是,可以得到N个输入向量对应的N个组合向量的向量序列<C1,C2,…,CN>,或矩阵C。基于该组合向量序列,可以得到N个输出向量。具体地,在一个实施例中,可以直接将N个组合向量的向量序列作为N个输出向量,即Yi=Ci。此时,输出矩阵Y即为组合向量矩阵C,又可以写成:
以上为一个注意力头head的处理过程描述,在MHA架构中,MHA层维护m套变换矩阵,每套变换矩阵包括前述第一变换矩阵Q、第二变换矩阵K和第三变换矩阵V,从而可以并行地进行上述操作,得到m个组合向量序列(即m个矩阵C),每个向量序列包括基于一套变换矩阵得到的N个组合向量。在这样的情况下,MHA层将得到的m个组合向量序列进行拼接,得到拼接矩阵;再通过第四变换矩阵W对该拼接矩阵进行变换,得到最终的输出矩阵Y。将该输出矩阵Y拆分即对应于N个输出向量<Y1,Y2,…,YN>。通过以上的操作过程,MHA层基于N个输入向量之间的关联度进行变换操作,得到N个输出向量。
在一种可能的实现中,第一特征表示和第二特征表示可以为多分辨率的特征图(例如金字塔形的特征),注意力网络可以是多个串联的处理单元,每个处理单元可以得到一个分辨率的特征图,处理单元之间可以连接上采样或者下采样模块,进而不同的处理单元可以得到不同分辨率的特征图,上述多个串联的处理单元可以具体为基于Transformer实现的主干网络Backbone,在Backbone中可以包括两个分支,一个分支用于提取图像中的粗粒度特征,一个分支用于提取图像中的细粒度特征,不但充分保持了Transformer对全局特征的感知性能,还挖掘了局部细粒度特征的提取能力,提高了Backbone的特征提取能力,进而提升了整体的精度。
本申请实施例中,通过提取图像数据的全局特征和局部细粒度特征,考虑到通用的图像处理任务,存在各种大小的目标,细粒度特征有助于识别多尺度、不规则的物体,进而可以提高了模型整体的精度。
1002、将所述第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示。
在一种可能的实现中,可以将第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示,融合特征表示可以即携带有图像数据的全局特征也携带有图像数据的局部细粒度特征。
在一种可能的实现中,可以将第一特征表示融合至第二特征表示,也就是将第一特征表示融合至第二特征表示,得到第一融合特征表示。为提升细粒度特征的表达能力,本申请实施例将其与感受野更大的全局粗粒度特征交互,这样将全局信息的感知带到了内部的特征里,如果不引入它,每一个Patch内部所对应的细粒度特征只能在一个固定的Patch内部区域内获得自注意力感知,无法与其他外部Patch的有效特征做交互。
在一种可能的实现中,所述第一特征表示和所述第二特征表示的尺寸不同,在将第一特征表示融合至第二特征表示时,可以通过对所述第一特征表示进行尺寸变换(例如通过卷积以及上采样实现),得到第三特征表示,所述第三特征表示和所述第二特征表示的尺寸相同;所述第三特征表示用于和所述第二特征表示进行融合得到所述第一融合特征表示。
在一种可能的实现中,可以将第二特征表示融合至第一特征表示,也就是将第二特征表示融合至第一特征表示,得到第二融合特征表示。为了提升全局粗粒度特征的表达能力,本申请实施例将其更加Patch内部的细粒度的特征与其进行融合,从而获得了更加细粒度的感知能力。
在一种可能的实现中,所述第一特征表示和所述第二特征表示的尺寸不同,在将第二特征表示融合至第一特征表示时,可以通过对所述第二特征表示进行尺寸变换(例如通过卷积实现),得到第四特征表示,所述第四特征表示和所述第一特征表示的尺寸相同;所述第四特征表示用于和所述第一特征表示进行融合得到所述第二融合特征表示。
两种融合方式的细节可以如图11所示,以PyramidTNT为例,将用于全局粗粒度特征与局部细粒度特征同时交互的模组称之为IOCF,在每一个Stage的最后都嵌入了IOCF(当然也可以对部分Stage的最后都嵌入IOCF)。对于IOCF的输入,包括粗粒度特征(Coarsepatches)以及细粒度特征(Fine Patches),这些是在模型刚开始通过一个Conv Stem生成的,并在PyramidTNT Blocks里分别做自注意力,提取各自模式下的特征。在基础模型中生成这些特征之后,可以将粗粒度特征和细粒度特征分别通过IOCF,完成全局粗粒度特征以及局部细粒度特征的相互交互。在以PyramidTNT为例中,由“粗粒度”融合到“细粒度”的过程是通过1x1的点卷积以及上采样完成的,由“细粒度”融合到“粗粒度”的过程是直接通过4x4的卷积完成。对于以其他Transformer模型作为基础,IOCF中的粗细粒度变换所用到的卷积可以适当的变换。经过变换相加后,又通过一个3x3卷积、LayerNorm归一以及GeLU激活的联合,将所相加的特征做进一步的融合,获得融合过后的粗粒度(Crossed CoarsePatches)以及细粒度特征(Crossed Fine Patches)。
1003、根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务。
在一种可能的实现中,所述目标任务为目标检测或者图像分割。
在一种可能的实现中,所述目标任务为与目标对象相关的目标检测或者图像分割;可以通过对目标对象对应的检测向量(det token)和所述融合特征表示进行信息交互,得到输出特征,并根据所述输出特征,通过下游任务网络,对所述图像执行目标任务,以得到处理结果。
在一种可能的实现中,上述融合特征表示可以包括第一融合特征表示(细粒度特征表示)和第二融合特征表示(粗粒度特征表示),由于现有的基于Transformer的图像处理模型只考虑了全局粗粒度的特征,并没有考虑到细粒度的特征。因此在执行下游任务时,也都是直接将检测向量Det Tokens与粗粒度特征进行交互。而本申请实施例中由于利用了其中的细粒度的特征,因此,可以将检测向量Det Tokens分别与细粒度特征和粗粒度特征进行交互。这样对于不同类型的特征交互方案,可以提升Det Tokens的感受丰富度,提升整体的图像处理效果。
在一种可能的实现中,可以将检测向量Det Tokens先与细粒度特征进行交互,之后再和粗粒度特征进行交互。具体的,可以通过对目标对象对应的检测向量和所述第一融合特征表示进行信息交互,得到交互结果,通过对所述交互结果和所述第二融合特征表示进行信息交互,得到输出特征。
在一种可能的实现中,可以将检测向量Det Tokens先与粗粒度特征进行交互,之后再和细粒度特征进行交互。具体的,可以通过对目标对象对应的检测向量和所述第二融合特征表示进行信息交互,得到交互结果,通过对所述交互结果和所述第一融合特征表示进行信息交互,得到输出特征。
此外,为了提高算法的计算效率,降低计算开销,可以在进行检测向量与粗粒度特征或者细粒度特征的信息交互时,将检测向量与粗粒度特征或者细粒度特征中的部分关键特征(也就是多个采样特征点)进行信息交互。
在一种可能的实现中,可以通过对目标对象对应的检测向量和所述融合特征表示中的多个采样特征点进行信息交互。
上述进行的检测向量Det Tokens与粗粒度特征和细粒度特征进行的交互可以由网络的额Neck实现,对于Neck部分,用于粗细粒度信息同时交叉感知的Neck。由于之前的基于Transformer的目标检测模型只考虑了全局粗粒度的特征,并没有考虑到细粒度的特征。因此在Neck部分,也都是直接将Det Tokens与粗粒度特征进行交互。而我们充分利用了其中的细粒度的特征,在Neck部分先将Det Tokens与细粒度特征交互,再将其与粗粒度的特征完成交互。这样对于不同类型的特征交互方案,可以提升Det Tokens的感受丰富度,提升整体的检测效果。如图15所示,在neck中主要是Det token与所提取的特征进行交互。而我们由于在backbone中考虑到了Transformer中的粗细粒度特征,因此我们让det token与二者共同交互。我们利用Multi-Scale Deformable Attention(MSDA)完成det token与二者的交互。其中的虚线所包围的区域表示在每一个stage的粗或者细粒度特征图采样点的位置,再通过aggragated sampled values操作,完成与采样点的交互。整个过程,先于细粒度特征做MSDA,再与粗粒度特征做MSDA。
本申请实施例中,通过粗细粒度信息交叉感知方案。让Det Tokens与粗细粒度的同时交叉感知,提升了Det Tokens的交叉特征源的丰富度,进一步提升了整体的检测效果。
本申请实施例提供了一种数据处理方法,所述方法包括:获取第一特征表示和第二特征表示,所述第一特征表示为根据注意力网络通过对图像中的多个图像块中不同图像块之间的信息交互得到;所述第二特征表示为基于注意力机制通过对所述图像块中的多个子图像块中不同子图像块之间的信息交互得到;将所述第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示;根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务。本申请通过提取图像数据的全局特征和局部细粒度特征,考虑到通用的图像处理任务,存在各种大小的目标,细粒度特征有助于识别多尺度、不规则的物体,进而可以提高了模型整体的精度。
接下来结合实验介绍本申请实施例的有益效果:
将本申请实施例在通用大规模的公开目标检测数据集COCO上进行了实验,实验结果可以如表1所示。
表1.COCO上的检测结果
由表1可以看到,相比于其他的基于Transformer的目标检测模型,本申请提供的方法(名称为FCDT)在COCO上取得了更好的效果,也就是更高的精度以及更少的计算量。不管在那个FLOPs区间内,我们的方法相比于最新的对比模型,可以取得更高的精度。图16更直观地对比了本申请提供的方法与其他模型的对比。
接下来从装置的角度介绍本申请实施例提供的一种数据处理装置,参照图17,图17为本申请实施例提供的一种数据处理装置的结构示意,如图17所示,本申请实施例提供的一种数据处理装置1700包括:
处理模块1701,用于获取第一特征表示和第二特征表示,所述第一特征表示为根据注意力网络通过对图像中的多个图像块中不同图像块之间的信息交互得到;所述第二特征表示为基于注意力机制通过对所述图像块中的多个子图像块中不同子图像块之间的信息交互得到;
将所述第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示;
其中,关于处理模块1701的具体描述可以参照上述实施例中步骤1001和步骤1002的描述,这里不再赘述。
任务执行模块1702,用于根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务。
其中,关于任务执行模块1702的具体描述可以参照上述实施例中步骤1003的描述,这里不再赘述。
本申请通过提取图像数据的全局特征和局部细粒度特征,考虑到通用的图像处理任务,存在各种大小的目标,细粒度特征有助于识别多尺度、不规则的物体,进而可以提高了模型整体的精度。
在一种可能的实现中,所述处理模块,还用于:
对所述图像进行划分,得到所述多个图像块;
对所述多个图像块中的至少一个图像块进行划分,得到所述多个子图像块。
在一种可能的实现中,所述目标任务为目标检测或者图像分割。
在一种可能的实现中,所述融合特征表示包括第一融合特征表示和第二融合特征表示;
所述处理模块,具体用于:
通过对所述第一特征表示进行尺寸变换,得到第三特征表示,所述第三特征表示和所述第二特征表示的尺寸相同;所述第三特征表示用于和所述第二特征表示进行融合得到所述第一融合特征表示。为提升细粒度特征的表达能力,本申请实施例将其与感受野更大的全局粗粒度特征交互,这样将全局信息的感知带到了内部的特征里,如果不引入它,每一个Patch内部所对应的细粒度特征只能在一个固定的Patch内部区域内获得自注意力感知,无法与其他外部Patch的有效特征做交互。
在一种可能的实现中,所述融合特征表示包括第一融合特征表示和第二融合特征表示;
所述处理模块,具体用于:
通过对所述第二特征表示进行尺寸变换,得到第四特征表示,所述第四特征表示和所述第一特征表示的尺寸相同;所述第四特征表示用于和所述第一特征表示进行融合得到所述第二融合特征表示。
为了提升全局粗粒度特征的表达能力,本申请实施例将其更加Patch内部的细粒度的特征与其进行融合,从而获得了更加细粒度的感知能力。
在一种可能的实现中,所述目标任务为与目标对象相关的目标检测或者图像分割;
所述任务执行模块,具体用于:
通过对目标对象对应的检测向量(det token)和所述融合特征表示进行信息交互,得到输出特征;
根据所述输出特征,通过下游任务网络,对所述图像执行目标任务,以得到处理结果。
在一种可能的实现中,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述第一融合特征表示进行信息交互,得到交互结果;
通过对所述交互结果和所述第二融合特征表示进行信息交互,得到输出特征。
在一种可能的实现中,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述第二融合特征表示进行信息交互,得到交互结果;
通过对所述交互结果和所述第一融合特征表示进行信息交互,得到输出特征。
在一种可能的实现中,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述融合特征表示中的多个采样特征点进行信息交互。
接下来介绍本申请实施例提供的一种执行设备,请参阅图18,图18为本申请实施例提供的执行设备的一种结构示意图,执行设备1800具体可以表现为手机、平板、笔记本电脑、智能穿戴设备、服务器等,此处不做限定。其中,执行设备1800实现图10对应实施例中图像处理方法的功能。具体的,执行设备1800包括:接收器1801、发射器1802、处理器1803和存储器1804(其中执行设备1800中的处理器1803的数量可以一个或多个),其中,处理器1803可以包括应用处理器18031和通信处理器18032。在本申请的一些实施例中,接收器1801、发射器1802、处理器1803和存储器1804可通过总线或其它方式连接。
存储器1804可以包括只读存储器和随机存取存储器,并向处理器1803提供指令和数据。存储器1804的一部分还可以包括非易失性随机存取存储器(non-volatile randomaccess memory,NVRAM)。存储器1804存储有处理器和操作指令、可执行模块或者数据结构,或者它们的子集,或者它们的扩展集,其中,操作指令可包括各种操作指令,用于实现各种操作。
处理器1803控制执行设备的操作。具体的应用中,执行设备的各个组件通过总线系统耦合在一起,其中总线系统除包括数据总线之外,还可以包括电源总线、控制总线和状态信号总线等。但是为了清楚说明起见,在图中将各种总线都称为总线系统。
上述本申请实施例揭示的方法可以应用于处理器1803中,或者由处理器1803实现。处理器1803可以是一种集成电路芯片,具有信号的处理能力。在实现过程中,上述方法的各步骤可以通过处理器1803中的硬件的集成逻辑电路或者软件形式的指令完成。上述的处理器1803可以是通用处理器、数字信号处理器(digital signal processing,DSP)、微处理器或微控制器、以及视觉处理器(vision processing unit,VPU)、张量处理器(tensorprocessing unit,TPU)等适用于AI运算的处理器,还可进一步包括专用集成电路(application specific integrated circuit,ASIC)、现场可编程门阵列(field-programmable gate array,FPGA)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件。该处理器1803可以实现或者执行本申请实施例中的公开的各方法、步骤及逻辑框图。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。结合本申请实施例所公开的方法的步骤可以直接体现为硬件译码处理器执行完成,或者用译码处理器中的硬件及软件模块组合执行完成。软件模块可以位于随机存储器,闪存、只读存储器,可编程只读存储器或者电可擦写可编程存储器、寄存器等本领域成熟的存储介质中。该存储介质位于存储器1804,处理器1803读取存储器1804中的信息,结合其硬件完成上述实施例中步骤1001至步骤1003的步骤。
接收器1801可用于接收输入的数字或字符信息,以及产生与执行设备的相关设置以及功能控制有关的信号输入。发射器1802可用于通过第一接口输出数字或字符信息;发射器1802还可用于通过第一接口向磁盘组发送指令,以修改磁盘组中的数据;发射器1802还可以包括显示屏等显示设备。
本申请实施例还提供了一种训练设备,请参阅图19,图19是本申请实施例提供的训练设备一种结构示意图,具体的,训练设备1900由一个或多个服务器实现,训练设备1900可因配置或性能不同而产生比较大的差异,可以包括一个或一个以上中央处理器(centralprocessing units,CPU)1919(例如,一个或一个以上处理器)和存储器1932,一个或一个以上存储应用程序1942或数据1944的存储介质1930(例如一个或一个以上海量存储设备)。其中,存储器1932和存储介质1930可以是短暂存储或持久存储。存储在存储介质1930的程序可以包括一个或一个以上模块(图示没标出),每个模块可以包括对训练设备中的一系列指令操作。更进一步地,中央处理器1919可以设置为与存储介质1930通信,在训练设备1900上执行存储介质1930中的一系列指令操作。
训练设备1900还可以包括一个或一个以上电源1926,一个或一个以上有线或无线网络接口1950,一个或一个以上输入输出接口1958;或,一个或一个以上操作系统1941,例如Windows ServerTM,Mac OS XTM,UnixTM,LinuxTM,FreeBSDTM等等。
本申请实施例中还提供一种包括计算机可读指令的计算机程序产品,当其在计算机上运行时,使得计算机执行如前述执行设备所执行的步骤,或者,使得计算机执行如前述训练设备所执行的步骤。
本申请实施例中还提供一种计算机可读存储介质,该计算机可读存储介质中存储有用于进行信号处理的程序,当其在计算机上运行时,使得计算机执行如前述执行设备所执行的步骤,或者,使得计算机执行如前述训练设备所执行的步骤。
本申请实施例提供的执行设备、训练设备或终端设备具体可以为芯片,芯片包括:处理单元和通信单元,所述处理单元例如可以是处理器,所述通信单元例如可以是输入/输出接口、管脚或电路等。该处理单元可执行存储单元存储的计算机执行指令,以使执行设备内的芯片执行上述实施例描述的模型训练方法,或者,以使训练设备内的芯片执行上述实施例中与模型训练相关的步骤。可选地,所述存储单元为所述芯片内的存储单元,如寄存器、缓存等,所述存储单元还可以是所述无线接入设备端内的位于所述芯片外部的存储单元,如只读存储器(read-only memory,ROM)或可存储静态信息和指令的其他类型的静态存储设备,随机存取存储器(random access memory,RAM)等。
具体的,请参阅图20,图20为本申请实施例提供的芯片的一种结构示意图,所述芯片可以表现为神经网络处理器NPU 2000,NPU 2000作为协处理器挂载到主CPU(Host CPU)上,由Host CPU分配任务。NPU的核心部分为运算电路2003,通过控制器2004控制运算电路2003提取存储器中的矩阵数据并进行乘法运算。
在一些实现中,运算电路2003内部包括多个处理单元(Process Engine,PE)。在一些实现中,运算电路2003是二维脉动阵列。运算电路2003还可以是一维脉动阵列或者能够执行例如乘法和加法这样的数学运算的其它电子线路。在一些实现中,运算电路2003是通用的矩阵处理器。
举例来说,假设有输入矩阵A,权重矩阵B,输出矩阵C。运算电路从权重存储器2002中取矩阵B相应的数据,并缓存在运算电路中每一个PE上。运算电路从输入存储器2001中取矩阵A数据与矩阵B进行矩阵运算,得到的矩阵的部分结果或最终结果,保存在累加器(accumulator)2008中。
统一存储器2006用于存放输入数据以及输出数据。权重数据直接通过存储单元访问控制器(Direct Memory Access Controller,DMAC)2005,DMAC被搬运到权重存储器2002中。输入数据也通过DMAC被搬运到统一存储器2006中。
BIU为Bus Interface Unit即,总线接口单元2010,用于AXI总线与DMAC和取指存储器(Instruction Fetch Buffer,IFB)2009的交互。
总线接口单元2010(Bus Interface Unit,简称BIU),用于取指存储器2009从外部存储器获取指令,还用于存储单元访问控制器2005从外部存储器获取输入矩阵A或者权重矩阵B的原数据。
DMAC主要用于将外部存储器DDR中的输入数据搬运到统一存储器2006或将权重数据搬运到权重存储器2002中或将输入数据数据搬运到输入存储器2001中。
向量计算单元2007包括多个运算处理单元,在需要的情况下,对运算电路的输出做进一步处理,如向量乘,向量加,指数运算,对数运算,大小比较等等。主要用于神经网络中非卷积/全连接层网络计算,如Batch Normalization(批归一化),像素级求和,对特征平面进行上采样等。
在一些实现中,向量计算单元2007能将经处理的输出的向量存储到统一存储器2006。例如,向量计算单元2007可以将线性函数;或,非线性函数应用到运算电路2003的输出,例如对卷积层提取的特征平面进行线性插值,再例如累加值的向量,用以生成激活值。在一些实现中,向量计算单元2007生成归一化的值、像素级求和的值,或二者均有。在一些实现中,处理过的输出的向量能够用作到运算电路2003的激活输入,例如用于在神经网络中的后续层中的使用。
控制器2004连接的取指存储器(instruction fetch buffer)2009,用于存储控制器2004使用的指令;
统一存储器2006,输入存储器2001,权重存储器2002以及取指存储器2009均为On-Chip存储器。外部存储器私有于该NPU硬件架构。
其中,上述任一处提到的处理器,可以是一个通用中央处理器,微处理器,ASIC,或一个或多个用于控制上述程序执行的集成电路。
另外需说明的是,以上所描述的装置实施例仅仅是示意性的,其中所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。另外,本申请提供的装置实施例附图中,模块之间的连接关系表示它们之间具有通信连接,具体可以实现为一条或多条通信总线或信号线。
通过以上的实施方式的描述,所属领域的技术人员可以清楚地了解到本申请可借助软件加必需的通用硬件的方式来实现,当然也可以通过专用硬件包括专用集成电路、专用CPU、专用存储器、专用元器件等来实现。一般情况下,凡由计算机程序完成的功能都可以很容易地用相应的硬件来实现,而且,用来实现同一功能的具体硬件结构也可以是多种多样的,例如模拟电路、数字电路或专用电路等。但是,对本申请而言更多情况下软件程序实现是更佳的实施方式。基于这样的理解,本申请的技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品存储在可读取的存储介质中,如计算机的软盘、U盘、移动硬盘、ROM、RAM、磁碟或者光盘等,包括若干指令用以使得一台计算机设备(可以是个人计算机,训练设备,或者网络设备等)执行本申请各个实施例所述的方法。
在上述实施例中,可以全部或部分地通过软件、硬件、固件或者其任意组合来实现。当使用软件实现时,可以全部或部分地以计算机程序产品的形式实现。
所述计算机程序产品包括一个或多个计算机指令。在计算机上加载和执行所述计算机程序指令时,全部或部分地产生按照本申请实施例所述的流程或功能。所述计算机可以是通用计算机、专用计算机、计算机网络、或者其他可编程装置。所述计算机指令可以存储在计算机可读存储介质中,或者从一个计算机可读存储介质向另一计算机可读存储介质传输,例如,所述计算机指令可以从一个网站站点、计算机、训练设备或数据中心通过有线(例如同轴电缆、光纤、数字用户线(DSL))或无线(例如红外、无线、微波等)方式向另一个网站站点、计算机、训练设备或数据中心进行传输。所述计算机可读存储介质可以是计算机能够存储的任何可用介质或者是包含一个或多个可用介质集成的训练设备、数据中心等数据存储设备。所述可用介质可以是磁性介质,(例如,软盘、硬盘、磁带)、光介质(例如,DVD)、或者半导体介质(例如固态硬盘(Solid State Disk,SSD))等。
Claims (19)
1.一种数据处理方法,其特征在于,所述方法包括:
获取第一特征表示和第二特征表示,所述第一特征表示为根据注意力网络通过对图像中的多个图像块中不同图像块之间的信息交互得到;所述第二特征表示为基于注意力机制通过对所述图像块中的多个子图像块中不同子图像块之间的信息交互得到;
将所述第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示;
根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务。
2.根据权利要求1所述的方法,其特征在于,所述方法还包括:
对所述图像进行划分,得到所述多个图像块;
对所述多个图像块中的至少一个图像块进行划分,得到所述多个子图像块。
3.根据权利要求1或2所述的方法,其特征在于,所述目标任务为目标检测或者图像分割。
4.根据权利要求1至3任一所述的方法,其特征在于,所述融合特征表示包括第一融合特征表示和第二融合特征表示;
所述第一特征表示和所述第二特征表示的尺寸不同,所述将所述第一特征表示和所述第二特征表示进行特征融合,包括:
通过对所述第一特征表示进行尺寸变换,得到第三特征表示,所述第三特征表示和所述第二特征表示的尺寸相同;所述第三特征表示用于和所述第二特征表示进行融合得到所述第一融合特征表示;或者,
通过对所述第二特征表示进行尺寸变换,得到第四特征表示,所述第四特征表示和所述第一特征表示的尺寸相同;所述第四特征表示用于和所述第一特征表示进行融合得到所述第二融合特征表示。
5.根据权利要求1至4任一所述的方法,其特征在于,所述目标任务为与目标对象相关的目标检测或者图像分割;
所述根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务,包括:
通过对目标对象对应的检测向量(det token)和所述融合特征表示进行信息交互,得到输出特征;
根据所述输出特征,通过下游任务网络,对所述图像执行目标任务,以得到处理结果。
6.根据权利要求5所述的方法,其特征在于,所述通过对目标对象对应的检测向量(dettoken)和所述融合特征表示进行信息交互,得到输出特征,包括:
通过对目标对象对应的检测向量和所述第一融合特征表示进行信息交互,得到交互结果;
通过对所述交互结果和所述第二融合特征表示进行信息交互,得到输出特征。
7.根据权利要求5所述的方法,其特征在于,所述通过对目标对象对应的检测向量(dettoken)和所述融合特征表示进行信息交互,得到输出特征,包括:
通过对目标对象对应的检测向量和所述第二融合特征表示进行信息交互,得到交互结果;
通过对所述交互结果和所述第一融合特征表示进行信息交互,得到输出特征。
8.根据权利要求5至7任一所述的方法,其特征在于,所述通过对目标对象对应的检测向量(det token)和所述融合特征表示进行信息交互,包括:
通过对目标对象对应的检测向量和所述融合特征表示中的多个采样特征点进行信息交互。
9.一种数据处理装置,其特征在于,所述装置包括:
处理模块,用于获取第一特征表示和第二特征表示,所述第一特征表示为根据注意力网络通过对图像中的多个图像块中不同图像块之间的信息交互得到;所述第二特征表示为基于注意力机制通过对所述图像块中的多个子图像块中不同子图像块之间的信息交互得到;
将所述第一特征表示和所述第二特征表示进行特征融合,得到融合特征表示;
任务执行模块,用于根据所述融合特征表示,通过下游任务网络,对所述图像执行目标任务。
10.根据权利要求9所述的装置,其特征在于,所述处理模块,还用于:
对所述图像进行划分,得到所述多个图像块;
对所述多个图像块中的至少一个图像块进行划分,得到所述多个子图像块。
11.根据权利要求9或10所述的装置,其特征在于,所述目标任务为目标检测或者图像分割。
12.根据权利要求9至11任一所述的装置,其特征在于,所述融合特征表示包括第一融合特征表示和第二融合特征表示;
所述处理模块,具体用于:
通过对所述第一特征表示进行尺寸变换,得到第三特征表示,所述第三特征表示和所述第二特征表示的尺寸相同;所述第三特征表示用于和所述第二特征表示进行融合得到所述第一融合特征表示;或者,
通过对所述第二特征表示进行尺寸变换,得到第四特征表示,所述第四特征表示和所述第一特征表示的尺寸相同;所述第四特征表示用于和所述第一特征表示进行融合得到所述第二融合特征表示。
13.根据权利要求9至12任一所述的装置,其特征在于,所述目标任务为与目标对象相关的目标检测或者图像分割;
所述任务执行模块,具体用于:
通过对目标对象对应的检测向量(det token)和所述融合特征表示进行信息交互,得到输出特征;
根据所述输出特征,通过下游任务网络,对所述图像执行目标任务,以得到处理结果。
14.根据权利要求13所述的装置,其特征在于,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述第一融合特征表示进行信息交互,得到交互结果;
通过对所述交互结果和所述第二融合特征表示进行信息交互,得到输出特征。
15.根据权利要求13所述的装置,其特征在于,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述第二融合特征表示进行信息交互,得到交互结果;
通过对所述交互结果和所述第一融合特征表示进行信息交互,得到输出特征。
16.根据权利要求13至15任一所述的装置,其特征在于,所述处理模块,具体用于:
通过对目标对象对应的检测向量和所述融合特征表示中的多个采样特征点进行信息交互。
17.一种计算设备,其特征在于,所述计算设备包括存储器和处理器;所述存储器存储有代码,所述处理器被配置为获取所述代码,并执行如权利要求1至8任一所述的方法。
18.一种计算机存储介质,其特征在于,所述计算机存储介质存储有一个或多个指令,所述指令在由一个或多个计算机执行时使得所述一个或多个计算机实施权利要求1至8任一所述的方法。
19.一种计算机程序产品,包括代码,其特征在于,在所述代码被执行时用于实现如权利要求1至8任一所述的方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202211168795.8A CN117765341A (zh) | 2022-09-24 | 2022-09-24 | 一种数据处理方法及相关装置 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202211168795.8A CN117765341A (zh) | 2022-09-24 | 2022-09-24 | 一种数据处理方法及相关装置 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN117765341A true CN117765341A (zh) | 2024-03-26 |
Family
ID=90317102
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202211168795.8A Pending CN117765341A (zh) | 2022-09-24 | 2022-09-24 | 一种数据处理方法及相关装置 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN117765341A (zh) |
-
2022
- 2022-09-24 CN CN202211168795.8A patent/CN117765341A/zh active Pending
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112183718A (zh) | 一种用于计算设备的深度学习训练方法和装置 | |
WO2024041479A1 (zh) | 一种数据处理方法及其装置 | |
CN112529149B (zh) | 一种数据处理方法及相关装置 | |
CN115081588A (zh) | 一种神经网络参数量化方法和装置 | |
CN113505193A (zh) | 一种数据处理方法及相关设备 | |
CN116861850A (zh) | 一种数据处理方法及其装置 | |
CN113656563A (zh) | 一种神经网络搜索方法及相关设备 | |
CN113627163A (zh) | 一种注意力模型、特征提取方法及相关装置 | |
CN116737895A (zh) | 一种数据处理方法及相关设备 | |
CN115879508A (zh) | 一种数据处理方法及相关装置 | |
CN116883715A (zh) | 一种数据处理方法及其装置 | |
CN116052714A (zh) | 一种数据处理方法及其装置 | |
CN115623242A (zh) | 一种视频处理方法及其相关设备 | |
CN117765341A (zh) | 一种数据处理方法及相关装置 | |
CN113065638A (zh) | 一种神经网络压缩方法及其相关设备 | |
CN117669691A (zh) | 一种数据处理方法及其装置 | |
CN117350913A (zh) | 一种数据处理方法及其装置 | |
CN113065636B (zh) | 一种卷积神经网络的剪枝处理方法、数据处理方法及设备 | |
WO2023207665A1 (zh) | 一种数据处理方法及相关设备 | |
CN117173035A (zh) | 一种数据处理方法及其装置 | |
CN117251592A (zh) | 一种数据处理方法及其装置 | |
CN117391138A (zh) | 一种数据处理方法及其装置 | |
CN117273074A (zh) | 一种数据处理方法及其装置 | |
CN116309226A (zh) | 一种图像处理方法及其相关设备 | |
WO2023231796A1 (zh) | 一种视觉任务处理方法及其相关设备 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication |