CN114942984A - 视觉场景文本融合模型的预训练和图文检索方法及装置 - Google Patents
视觉场景文本融合模型的预训练和图文检索方法及装置 Download PDFInfo
- Publication number
- CN114942984A CN114942984A CN202210590151.1A CN202210590151A CN114942984A CN 114942984 A CN114942984 A CN 114942984A CN 202210590151 A CN202210590151 A CN 202210590151A CN 114942984 A CN114942984 A CN 114942984A
- Authority
- CN
- China
- Prior art keywords
- text
- sample
- scene
- coding
- image
- 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.)
- Granted
Links
- 230000004927 fusion Effects 0.000 title claims abstract description 378
- 230000000007 visual effect Effects 0.000 title claims abstract description 311
- 238000012549 training Methods 0.000 title claims abstract description 139
- 238000000034 method Methods 0.000 title claims abstract description 103
- 238000012545 processing Methods 0.000 claims abstract description 26
- 108091026890 Coding region Proteins 0.000 claims description 53
- 239000013598 vector Substances 0.000 claims description 52
- 238000004590 computer program Methods 0.000 claims description 9
- 238000000605 extraction Methods 0.000 claims description 7
- 238000000638 solvent extraction Methods 0.000 claims description 6
- 238000013473 artificial intelligence Methods 0.000 abstract description 8
- 238000013135 deep learning Methods 0.000 abstract description 3
- 238000005516 engineering process Methods 0.000 description 19
- 238000010586 diagram Methods 0.000 description 13
- 230000008569 process Effects 0.000 description 12
- 230000006870 function Effects 0.000 description 9
- 238000004891 communication Methods 0.000 description 8
- 239000011159 matrix material Substances 0.000 description 5
- 238000012015 optical character recognition Methods 0.000 description 5
- 230000009977 dual effect Effects 0.000 description 3
- 238000002474 experimental method Methods 0.000 description 3
- 238000012986 modification Methods 0.000 description 3
- 230000004048 modification Effects 0.000 description 3
- 230000003287 optical effect Effects 0.000 description 3
- 230000008447 perception Effects 0.000 description 3
- 230000000694 effects Effects 0.000 description 2
- 235000015220 hamburgers Nutrition 0.000 description 2
- 230000003993 interaction Effects 0.000 description 2
- 238000010801 machine learning Methods 0.000 description 2
- 238000011160 research Methods 0.000 description 2
- 238000003491 array Methods 0.000 description 1
- 230000006399 behavior Effects 0.000 description 1
- 230000005540 biological transmission Effects 0.000 description 1
- 238000004364 calculation method Methods 0.000 description 1
- 230000015556 catabolic process Effects 0.000 description 1
- 230000001413 cellular effect Effects 0.000 description 1
- 238000010276 construction Methods 0.000 description 1
- 238000006731 degradation reaction Methods 0.000 description 1
- 238000013461 design Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 230000002452 interceptive effect Effects 0.000 description 1
- 239000004973 liquid crystal related substance Substances 0.000 description 1
- 230000003924 mental process Effects 0.000 description 1
- 238000003058 natural language processing Methods 0.000 description 1
- 238000010606 normalization Methods 0.000 description 1
- 239000013307 optical fiber Substances 0.000 description 1
- 239000004065 semiconductor Substances 0.000 description 1
- 230000001953 sensory effect Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
- 238000012795 verification Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
- G06F16/30—Information retrieval; Database structures therefor; File system structures therefor of unstructured textual data
- G06F16/33—Querying
- G06F16/332—Query formulation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
- G06F16/30—Information retrieval; Database structures therefor; File system structures therefor of unstructured textual data
- G06F16/38—Retrieval characterised by using metadata, e.g. metadata not derived from the content or metadata generated manually
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
- G06F16/50—Information retrieval; Database structures therefor; File system structures therefor of still image data
- G06F16/53—Querying
- G06F16/532—Query formulation, e.g. graphical querying
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
- G06F16/50—Information retrieval; Database structures therefor; File system structures therefor of still image data
- G06F16/58—Retrieval characterised by using metadata, e.g. metadata not derived from the content or metadata generated manually
- G06F16/583—Retrieval characterised by using metadata, e.g. metadata not derived from the content or metadata generated manually using metadata automatically derived from the content
- G06F16/5846—Retrieval characterised by using metadata, e.g. metadata not derived from the content or metadata generated manually using metadata automatically derived from the content using extracted text
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
- G06F16/50—Information retrieval; Database structures therefor; File system structures therefor of still image data
- G06F16/58—Retrieval characterised by using metadata, e.g. metadata not derived from the content or metadata generated manually
- G06F16/5866—Retrieval characterised by using metadata, e.g. metadata not derived from the content or metadata generated manually using information manually generated, e.g. tags, keywords, comments, manually generated location and time information
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/25—Fusion techniques
- G06F18/253—Fusion techniques of extracted features
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F40/00—Handling natural language data
- G06F40/30—Semantic analysis
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/40—Extraction of image or video features
- G06V10/42—Global feature extraction by analysis of the whole pattern, e.g. using frequency domain transformations or autocorrelation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/77—Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
- G06V10/774—Generating sets of training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/77—Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
- G06V10/80—Fusion, i.e. combining data from various sources at the sensor level, preprocessing level, feature extraction level or classification level
- G06V10/806—Fusion, i.e. combining data from various sources at the sensor level, preprocessing level, feature extraction level or classification level of extracted features
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/82—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/60—Type of objects
- G06V20/62—Text, e.g. of license plates, overlay texts or captions on TV images
- G06V20/63—Scene text, e.g. street names
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N19/00—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
- H04N19/10—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding
- H04N19/169—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the coding unit, i.e. the structural portion or semantic portion of the video signal being the object or the subject of the adaptive coding
- H04N19/17—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the coding unit, i.e. the structural portion or semantic portion of the video signal being the object or the subject of the adaptive coding the unit being an image region, e.g. an object
- H04N19/176—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the coding unit, i.e. the structural portion or semantic portion of the video signal being the object or the subject of the adaptive coding the unit being an image region, e.g. an object the region being a block, e.g. a macroblock
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02D—CLIMATE CHANGE MITIGATION TECHNOLOGIES IN INFORMATION AND COMMUNICATION TECHNOLOGIES [ICT], I.E. INFORMATION AND COMMUNICATION TECHNOLOGIES AIMING AT THE REDUCTION OF THEIR OWN ENERGY USE
- Y02D10/00—Energy efficient computing, e.g. low power processors, power management or thermal management
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Databases & Information Systems (AREA)
- General Engineering & Computer Science (AREA)
- Multimedia (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Artificial Intelligence (AREA)
- Health & Medical Sciences (AREA)
- General Health & Medical Sciences (AREA)
- Library & Information Science (AREA)
- Computing Systems (AREA)
- Software Systems (AREA)
- Computational Linguistics (AREA)
- Mathematical Physics (AREA)
- Medical Informatics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Signal Processing (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- Molecular Biology (AREA)
- Audiology, Speech & Language Pathology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Evolutionary Biology (AREA)
- Image Analysis (AREA)
- Compression Or Coding Systems Of Tv Signals (AREA)
Abstract
本公开提供了一种视觉场景文本融合模型的预训练和图文检索方法及装置,涉及人工智能技术领域,具体涉及深度学习、图像处理和计算机视觉技术领域。具体实现方案为:获取样本图文对;提取样本图像中的样本场景文本;将样本文本输入文本编码网络,得到样本文本特征;将样本图像和初始的样本融合特征输入视觉编码子网络,以及将初始的样本融合特征和样本场景文本输入场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征;根据样本文本特征、样本图像的全局图像特征和经学习的样本融合特征,对视觉场景文本融合模型进行预训练。通过上述技术方案,能够提高图文跨模态检索性能。
Description
技术领域
本公开涉及人工智能技术领域,具体涉及深度学习、图像处理和计算机视觉技术领域。
背景技术
随着人工智能技术的发展,检索的形式越来越多样化。其中,图文检索是目前已有检索形式中的重要一种。图文检索的任务形式可以是给定一个搜索文本,从图像库中选出与搜索文本最相关的图像,或者还可以是给定一张图像,从文本库中选出与所给定的图像最相关的文本。在图文检索场景下,如何精准的完成检索任务至关重要。
发明内容
本公开提供了一种视觉场景文本融合模型的预训练和图文检索方法及装置。
根据本公开的一方面,提供了一种视觉场景文本融合模型的预训练方法,其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络,所述方法包括:
获取样本图文对;其中,所述样本图文对包括样本图像和样本文本;
提取所述样本图像中的样本场景文本;
将所述样本文本输入所述文本编码网络,得到样本文本特征;
将所述样本图像和初始的样本融合特征输入所述视觉编码子网络,以及将初始的样本融合特征和所述样本场景文本输入所述场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征;
根据所述样本文本特征、所述样本图像的全局图像特征和经学习的样本融合特征,对所述视觉场景文本融合模型进行预训练。
根据本公开的另一方面,提供了一种视觉场景文本融合模型的训练方法,该方法包括:
获取业务方提供的业务图文对;所述业务图文对包括业务图像和业务文本;
将所述业务图像和所述业务文本作为训练数据,对视觉场景文本融合模型进行微调;其中,所述视觉场景文本融合模型基于本公开任一种视觉场景文本融合模型的预训练方法进行预训练得到。
根据本公开的另一方面,提供了一种视觉场景文本融合模型的图文检索方法,其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络,所述方法包括:
获取待检索的目标文本;
提取候选图像中的候选场景文本;
将所述目标文本输入所述文本编码网络,得到目标文本特征;
将所述候选图像和初始的候选融合特征输入所述视觉编码子网络,以及将初始的候选融合特征和所述候选场景文本输入所述场景编码子网络,得到所述候选图像的全局图像特征;
根据所述目标文本特征和所述候选图像的全局图像特征,从所述候选图像中确定目标图像。
根据本公开的又一方面,提供了一种视觉场景文本融合模型的预训练装置,其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络,所述装置包括:
样本图文对获取模块,用于获取样本图文对;其中,所述样本图文对包括样本图像和样本文本;
样本场景文本提取模块,用于提取所述样本图像中的样本场景文本;
样本文本特征确定模块,用于将所述样本文本输入所述文本编码网络,得到样本文本特征;
样本全局特征确定模块,用于将所述样本图像和初始的样本融合特征输入所述视觉编码子网络,以及将初始的样本融合特征和所述样本场景文本输入所述场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征;
模型预训练模块,用于根据所述样本文本特征、所述样本图像的全局图像特征和经学习的样本融合特征,对所述视觉场景文本融合模型进行预训练。
根据本公开的又一方面,提供了一种视觉场景文本融合模型的训练装置,包括:
业务图文对获取模块,用于获取业务方提供的业务图文对;所述业务图文对包括业务图像和业务文本;
微调模块,用于将所述业务图像和所述业务文本作为训练数据,对视觉场景文本融合模型进行微调;其中,所述视觉场景文本融合模型基于本公开任一种视觉场景文本融合模型的预训练装置得到。
根据本公开的又一方面,提供了一种视觉场景文本融合模型的图文检索装置,其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络,所述装置包括:
目标文本获取模块,用于获取待检索的目标文本;
候选场景文本提取模块,用于提取候选图像中的候选场景文本;
目标文本特征确定模块,用于将所述目标文本输入所述文本编码网络,得到目标文本特征;
候选全局特征确定模块,用于将所述候选图像和初始的候选融合特征输入所述视觉编码子网络,以及将初始的候选融合特征和所述候选场景文本输入所述场景编码子网络,得到所述候选图像的全局图像特征;
目标图像确定模块,用于根据所述目标文本特征和所述候选图像的全局图像特征,从所述候选图像中确定目标图像。
根据本公开的又一方面,提供了一种电子设备,该电子设备包括:
至少一个处理器;以及
与所述至少一个处理器通信连接的存储器;其中,
所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行本公开任一实施例所述的视觉场景文本融合模型的预训练方法、训练方法或图文检索方法。
根据本公开的另一方面,提供了一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使计算机执行本公开任一实施例所述的视觉场景文本融合模型的预训练方法、训练方法或图文检索方法。
应当理解,本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征,也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。
附图说明
附图用于更好地理解本方案,不构成对本公开的限定。其中:
图1是根据本公开实施例提供的一种视觉场景文本融合模型的预训练方法的流程图;
图2A是根据本公开实施例提供的另一种视觉场景文本融合模型的预训练方法的流程图;
图2B是根据本公开实施例提供的一种视觉场景文本融合模型的示意图;
图2C是根据本公开实施例提供的一种样本融合特征的确定过程示意图;
图3是根据本公开实施例提供的又一种视觉场景文本融合模型的预训练方法的流程图;
图4是根据本公开实施例提供的一种视觉场景文本融合模型的图文检索方法的流程图;
图5是根据本公开实施例提供的一种视觉场景文本融合模型的预训练装置的结构示意图;
图6是根据本公开实施例提供的一种视觉场景文本融合模型的训练装置的结构示意图;
图7是根据本公开实施例提供的一种视觉场景文本融合模型的图文检索装置的结构示意图;
图8是用来实现本公开实施例的视觉场景文本融合模型的预训练方法、训练方法或图文检索方法的电子设备的框图。
具体实施方式
以下结合附图对本公开的示范性实施例做出说明,其中包括本公开实施例的各种细节以助于理解,应当将它们认为仅仅是示范性的。因此,本领域普通技术人员应当认识到,可以对这里描述的实施例做出各种改变和修改,而不会背离本公开的范围和精神。同样,为了清楚和简明,以下的描述中省略了对公知功能和结构的描述。
需要说明的是,本公开的说明书和权利要求书及上述附图中的术语“第一”、“第二”、“第三”、“目标”等是用于区别类似的对象,而不必用于描述特定的顺序或先后次序。其次,本公开的说明书和权利要求书及上述附图中的术语“样本”、“候选”等是用于区别不同阶段的对象,而不必用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换,以便这里描述的本公开的实施例能够以除了在这里图示或描述的那些以外的顺序实施。此外,术语“包括”和“具有”以及他们的任何变形,意图在于覆盖不排他的包含,例如,包含了一系列步骤或单元的过程、方法、***、产品或设备不必限于清楚地列出的那些步骤或单元,而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它步骤或单元。
在图文跨模态检索过程中,图像中的场景文本对理解图像的视觉语义信息具有积极价值,相关技术中的图文跨模态检索未充分利用场景文本信息。然而,只有部分图像中存在场景文本,还有部分图像中没有场景文本,如果直接将场景文本信息增加到图像跨模态检索过程中,将导致没有场景文本的图文检索性能下降。
基于上述问题,本公开提出了一种效果显著的视觉场景文本融合(Vision andScene Text Aggregation,ViSTA)模型,用于处理存在场景文本和不存在场景文本情况下的跨模态检索任务。ViSTA模型完全采用变形器(transformer)设计,通过视觉编码和场景编码共享融合特征,使得视觉编码过程中引入场景文本。为了解决场景文本缺失问题,我们进一步提出了融合文本对比损失和图像文本对比损失的双重对比监督。与相关技术中的跨模态检索相比,ViSTA能够将场景文本语义与视觉特征融合在一起,并且能够解决没有场景文本情况下的检索性能下降问题。
本公开的贡献至少包括如下几个方面:1)提出了一种完全采用变形器的网络架构,可以有效地对视觉图像和场景文本进行融合,该架构适用于存在场景文本的情况和不存在场景文本的情况;2)通过视觉编码子网络和场景编码子网络通过共享融合特征,对视觉、场景文本的相关信息进行交换,并通过双重对比损失增强视觉特征;3)经研究发现,ViSTA模型的性能远优于相关技术中的图文跨模态检索方法。
图1是根据本公开实施例提供的一种视觉场景文本融合模型的预训练方法的流程图。本实施例适用于如何对视觉场景文本融合模型进行预训练的情况。该方法可以由视觉场景文本融合模型的预训练装置来执行,该装置可以采用软件和/或硬件的方式实现,并可集成于承载视觉场景文本融合模型的预训练功能的电子设备中。可选的,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,视觉场景编码网络包括视觉编码子网络和场景编码子网络。如图1所示,本实施例的视觉场景文本融合模型的预训练方法可以包括:
S101,获取样本图文对。
本实施例中,样本图文对是用于对视觉场景文本融合模型进行预训练的样本数据。可选的,每一个样本图文对中包括一个样本图像和一个样本文本,样本文本用于对样本图像中的图像内容进行描述。以吃汉堡图像为例,样本文本可以为一位年轻人在吃汉堡。
可选的,可以用从互联网等中抓取大量的样本图文对,来对视觉场景文本融合模型进行预训练。考虑到实际场景中的图像存在具有场景文本和不具有场景文本两种情况,为了使视觉场景文本融合模型能够同时适用于具有场景文本的图文检索场景,还适用于不具有场景文本的图文检索场景,本实施例中的样本图像可为具有场景文本的图像,也可为不具有场景文本的图像。
S102,提取样本图像中的样本场景文本。
本实施例中,样本场景文本是指样本图像中存在的文本。
具体的,可以基于光学字符识别(Optical Character Recognition,OCR)技术,对样本图像进行处理,得到样本图像中的样本场景文本;进一步的,还可以得到样本场景文本在样本图像中的图位置编码信息。需要说明的是,样本图像中可以包括一个或多个样本场景文本;对于每一样本场景文本,可以包括一个词语或多个词语。
需要说明的是,针对没有场景文本的样本图像,从该样本图像中提取的样本场景文本为空文本。
S103,将样本文本输入文本编码网络,得到样本文本特征。
本实施例中,样本文本特征是指样本文本的语义特征,可以采用矩阵或向量形式表示。
具体的,可以将样本文本输入至文本编码网络,经过文本编码网络处理,得到样本文本特征,即文本token。
S104,将样本图像和初始的样本融合特征输入视觉编码子网络,以及将初始的样本融合特征和样本场景文本输入场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征。
本实施例中,样本融合特征用于表征图像和场景文本之间的相关性特征,即融合token,可以采用矩阵或向量形式;需要说明的是,初始的样本融合特征可以预先设定,在模型预训练、模型微调过程中,样本融合特征可动态学习更新。所谓样本图像的全局图像特征通过从全局角度,对样本图像进行多模态特征提取得到,可称为图像token。
具体的,可以将样本图像和初始的样本融合特征输入视觉场景编码网络中的视觉编码子网络,以及将初始的样本融合特征和样本场景文本输入视觉编码网络中的场景编码子网络,由视觉编码子网络和场景编码子网络配合处理,得到视觉编码场景网络输出的样本图像的全局图像特征和经学习的样本融合特征。
S105,根据样本文本特征、样本图像的全局图像特征和经学习的样本融合特征,对视觉场景文本融合模型进行预训练。
一种可选方式,可以基于预先设定的损失函数,根据样本文本特征、样本图像的全局图像特征和经学习的样本融合特征,计算训练损失;并根据训练损失对视觉场景文本融合模型进行预训练,直到训练损失稳定在设定范围,或者训练次数达到设定次数,则停止预训练。其中,设定范围和设定次数可以由本领域技术人员根据实际需求设定。
本公开实施例的技术方案,通过将样本图文对中的样本文本输入文本编码网络,得到样本文本特征;以及将样本图文对中的样本图像和初始的样本融合特征输入视觉编码子网络,将初始的样本融合特征和从样本图像中提取的样本场景文本输入场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征,进而根据样本文本特征、样本图像的全局图像特征和经学习的样本融合特征,对视觉场景文本融合模型进行预训练。上述技术方案,引入样本融合特征、样本图像的全局图像特征和样本文本特征对视觉场景文本融合模型进行预训练,使得无论输入的图像是否有场景文本,均可以准确地输出图像的全局图像特征,从而使得视觉场景文本融合模型可以支持在大范围图文集下的检索任务,也就是说,视觉场景文本融合模型不仅适用于有场景文本情况下的跨模态检索,同样适用于没有场景文本情况下的跨模态检索;此外,本方案通过双编码结构即并行的文本编码网络和视觉场景编码网络,对图像和文本进行并行处理,可以提高跨模态检索效率。
在上述实施例的基础上,作为本公开的一种可选方式,将样本文本输入文本编码网络,得到样本文本特征可以是,对样本文本进行词嵌入,得到样本文本词向量;根据样本文本的模态信息、样本文本的位置编码信息和样本文本词向量,确定样本文本的词编码结果;根据初始的样本文本特征和样本文本的词编码结果,构建样本文本的编码序列;将样本文本的编码序列输入文本编码网络,得到经学习的样本文本特征。
其中,所谓样本文本的模态信息即为预先设定的用于表征样本文本这一模态的编码信息,以区别于后续样本图像块的模态信息和样本场景文本的模态信息,例如,样本图像块的模态信息、样本场景文本的模态信息和样本文本的模态信息可依次用m0、m1和m2表示,即各模态的编码信息不同。样本文本的位置编码信息即为样本文本中的词语在样本文本中词序编码信息。
具体的,可以基于词嵌入技术,对样本文本进行处理,得到样本文本中每一词语的向量,即样本文本词向量。
并且,对于样本文本中的每一词语,可以基于一定的规则,将样本文本的模态信息、该词语的位置编码信息和该词语的样本文本词向量进行拼接,得到该词语的词编码结果。
在确定样本文本中各词语的词编码结果后,可以根据初始的样本文本特征和各词语的词编码结果,构建样本文本的编码序列。其中,初始的样本文本特征具体可以由初始的文本编码、样本文本的模态信息和样本文本的位置编码信息构成;可选的,初始的样本文本特征位于样本文本的编码序列的首位。
在确定样本文本的编码序列后,可以将样本文本的编码序列输入文本编码网络,得到经学习的样本文本特征。
可以理解的是,结合样本文本的模态信息、样本文本的位置编码信息和样本文本词向量等,来学习样本文本特征,使得最终所确定的样本文本特征更加准确。
在上述实施例的基础上,作为本公开的一种可选方式,将样本图像输入视觉编码子网络可以是,对样本图像进行分块,得到样本图像块序列;对样本图像块序列中的样本图像块进行线性投影,得到样本图像块的编码结果;根据样本图像块的模态信息和样本图像块的位置编码信息对样本图像块的编码结果进行处理,得到样本图像块经处理的编码结果;根据初始的全局图像特征和样本图像块经处理的编码结果,构建样本图像的编码序列;将样本图像的编码序列输入视觉编码子网络中的输入层。
本实施例中,所谓样本图像块的模态信息即为预先设定的用于表征样本图像这一模态的编码信息。所谓样本图像块的位置编码信息即为样本图像块的标识信息。
在得到样本图像块序列后,针对样本图像块序列中的每一样本图像块,对该样本图像块进行线性投影,得到该样本图像块的编码结果。并且,可以基于一定的规则,将该样本图像块的编码结果、样本图像块的模态信息和样本图像块的位置编码信息进行拼接,得到该样本图像块经处理的编码结果。
在确定各样本图像块经处理的编码结果后,可以根据初始的全局图像特征和各样本图像块经处理的编码结果,构建样本图像的编码序列。其中,初始的全局图像特征具体可以由待学习的图像编码、样本图像块的模态信息和样本图像块的位置编码信息构成。可选的,初始的全局图像特征位于样本图像的编码序列的首位。在确定样本图像的编码序列后,可以将样本图像的编码序列输入视觉编码子网络中的输入层。
可以理解的是,本公开通过使用图像分块线性投影的方式,确定样本图像块的编码结果,相比于采用检测器获取图像中图像块的编码结果而言,减少了计算复杂度,提高了效率;同时,结合样本图像块的模态信息、样本图像块的位置编码信息和初始的全局图像特征等,确定样本图像的编码序列,使得视觉编码子网络能够提取到样本图像中更加丰富的特征。
在上述实施例的基础上,作为本公开的一种可选方式,将样本场景文本输入场景编码子网络可以是,对样本场景文本进行词嵌入,得到样本场景文本向量;根据样本场景文本的图位置编码信息、样本场景文本的模态信息、样本场景文本的字位置编码信息和样本场景文本向量,确定样本图像场景文本的编码结果;根据初始的样本场景文本特征和样本图像场景文本的编码结果,构建样本场景文本的编码序列;将样本场景文本的编码序列输入场景编码子网络中的输入层。
本实施例中,所谓样本场景文本的模态信息即为预先设定的用于表征样本图像中场景文本这一模态的编码信息。所谓样本场景文本的字位置编码信息用于唯一标识样本场景文本中的文字的顺序。所谓样本场景文本的图位置编码信息即为样本场景文本在样本图像中的图像位置信息。
具体的,对于从样本图像中提取到的每一样本场景文本,可以基于词嵌入技术,对该样本场景文本进行处理,得到该样本场景文本的样本场景文本向量,记为Embedding(oword)。
在确定该样本场景文本的样本场景文本向量后,可以基于一定的规则,将该样本场景文本的图位置编码信息(记为Flinear(obbox))、样本场景文本的模态信息(记为Stype)、样本场景文本的字位置编码信息(记为Stoken_id)和样本场景文本向量(记为Embedding(oword))进行拼接,得到该样本场景文本的编码结果(记为S0)。例如可以依次将图位置编码信息、样本场景文本的模态信息、样本场景文本的字位置编码信息和样本场景文本向量进行拼接,得到该样本场景文本的编码结果,例如,可以通过如下公式得到样本场景文本的编码结果:
Sinit=Embedding(oword)+Flinear(obbox)
S0=Sinit+Stype+Stoken_id
其中,S0表示样本场景文本的编码结果,Flinear(obbox)表示图位置编码信息,Embedding(oword)表示样本场景文本向量,Stype表示样本场景文本的模态信息,Stoken_id表示样本场景文本的字位置编码信息。
在确定样本图像中各样本场景文本的编码结果后,可以根据初始的样本场景文本特征和各样本场景文本的编码结果,构建样本场景文本的编码序列。初始的样本场景文本特征具体可以由待学习的场景文本编码、样本场景文本的模态信息、样本场景文本的图位置编码信息和字位置编码信息构成;可选的,初始的样本场景文本特征位于样本场景文本的编码序列的首位。在确定样本场景文本的编码序列后,可以将样本场景文本的编码序列输入至场景编码子网络中的输入层。
可以理解的是,结合图位置编码信息、样本场景文本的模态信息、样本场景文本的字位置编码信息和初始的样本场景文本特征等,确定样本场景文本的编码序列,使得视觉编码子网络能够提取到样本场景文本中更加丰富的特征。
图2A是根据本公开实施例提供的另一种视觉场景文本融合模型的训练方法的流程图;图2B是根据本公开实施例提供的一种视觉场景文本融合模型的示意图。本实施例在上述实施例的基础上,对“将样本图像和初始的样本融合特征输入视觉编码网络,以及将初始的样本融合特征和样本场景文本输入场景编码子网络,得到样本图像的样本图像的全局图像特征和经学习的样本融合特征”进一步优化,提供一种可选方案。如图2A所示,本实施例的视觉场景文本融合模型的训练方法可以包括:
S201,获取样本图文对。
其中,样本图文对包括样本图像和样本文本。
S202,提取样本图像中的样本场景文本。
S203,将样本文本输入文本编码网络,得到样本文本特征。
S204,将样本图像输入视觉编码子网络中的输入层,以及将初始的样本融合特征输入视觉编码子网络中的融合层,得到视觉编码子网络输出的样本图像的全局图像特征和视觉编码子网络输出的视觉融合特征。
S205,将样本场景文本输入场景编码子网络中的输入层,以及将初始的样本融合特征输入场景编码子网络中的融合层,得到场景编码子网络输出的场景融合特征。
S206,对视觉编码子网络输出的视觉融合特征和场景编码子网络输出的场景融合特征进行融合,得到经学习的样本融合特征。
S207,根据样本文本特征、样本图像的全局图像特征和经学习的样本融合特征,对视觉场景文本融合模型进行预训练。
本实施例中,视觉编码子网络可以由若干个Transformer(变形器)层构成,场景编码子网络也可以由若干个Transformer层构成;优选的,视觉编码子网络和场景编码子网络的网络结构即网络层数可以相同。
进一步的,视觉编码子网络可以包括至少一个融合层,场景编码子网络也可以包括至少一个融合层。可选的,视觉编码子网络中融合层的数量与场景编码子网络中融合层的数量相同;在视觉编码子网络和场景编码子网络的网络层数相同的情况下,视觉编码子网络中融合层所处的位置,与场景编码子网络中融合层所处的位置相同。
示例性的,在视觉编码子网络中,融合层之前视觉编码子网络中任意一个Transformer层的学习过程可以通过如下公式表示:
Yl←MHSA(LN(Vl))+Vl
Vl+1←MLP(LN(Yl))+Yl,
其中,Vl表示第l层预测的图像特征(即图像token),MHSA(·)指的是多头自注意力层,MLP(·)指的是多层感知层,LN(·)指的是层归一化,Yl为中间变量;其中,l为正整数。
在视觉编码子网络中,融合层之前场景编码子网络中的任意一个Transformer层的学习过程可以通过如下公式表示:
Xl←MHSA(LN(Sl))+Sl
Sl+1←MLP(LN(Xl))+Xl,
其中,Sl表示第l层的预测场景文本特征(即场景文本token),Xl为中间变量。
在视觉编码子网络中,首个融合层根据初始的样本融合特征和从上一层接收的全局图像特征,输出新的全局图像特征和视觉融合特征;在场景编码子网络中,首个融合层根据初始的样本融合特征和从上一层接收的场景文本特征,输出新的场景文本特征和场景融合特征;对视觉编码子网络中首个融合层输出的视觉融合特征和场景编码子网络中首个融合层输出的场景融合特征进行融合,得到更经学习的样本融合特征。
之后,在视觉编码子网络中,第二个融合层根据首个融合层输出的全局图像特征和更经学习的样本融合特征,输出新的全局图像特征和新的视觉融合特征;在场景编码子网络中,第二个融合层根据首个融合层输出的场景文本特征和更经学习的样本融合特征,输出新的场景文本特征和新的场景融合特征;对视觉编码子网络中第二个融合层输出的视觉融合特征和场景编码子网络中第二个融合层输出的场景融合特征进行融合,得到更经学习的样本融合特征。
依次类推,在视觉编码子网络中,可以得到最后一个融合层(即输出层)输出的全局图像特征和视觉融合特征;在场景编码子网络中,可得到最后一个融合层(即输出层)输出的场景融合特征;对视觉编码子网络中最后一个融合层输出的视觉融合特征和场景编码子网络中最后一个融合层输出的场景融合特征进行融合,得到经学习的样本融合特征,作为样本融合特征的最终学习结果。
示例性的,视觉编码子网络中的任一融合层的处理流程可以是:
Ml←MHSA(LN([Vl;Fl]))+[Vl;Fl]
[Vl+1;VFUS]←MLP(LN(Ml))+Ml,
其中,VFUS表示视觉编码子网络输出的视觉融合特征,Ml为中间变量,Fl为样本融合特征,即样本融合token。
同理,场景编码子网络中的任一融合层的处理流程可以是:
Nl←MHSA(LN([Sl;Fl]))+[Sl;Fl]
[Sl+1;SFUS]←MLP(LN(Nl))+Nl,
其中,Sl表示第l层的场景文本特征,SFUS表示是场景编码子网络输出的场景融合特征,Nl为中间变量。
结合图2B所示,假设视觉编码子网络由K个Transformer层构成,场景编码子网络由K个Transformer层构成;并且,视觉编码子网络中第K-1和第K个Transformer层均作为融合层;场景编码子网络中第K-1和第K个Transformer层均作为融合层。其中,K为大于2的正整数。
样本图像的全局图像特征和经学习的样本融合特征可以通过如下过程确定:在视觉编码子网络中,将样本图像输入视觉编码子网络中的输入层,经过前K-2个Transformer层学习,得到第K-2个Transformer层所预测的全局图像特征。同理,在场景编码子网络中,将样本场景文本输入场景编码子网络中的输入层,经过前K-2个Transformer层学习,得到第K-2个Transformer层所预测的场景文本特征。
如图2C所示,在视觉编码子网络中,将初始的样本融合特征和第K-2个Transformer层预测得到的全局图像特征,输入视觉编码子网络中首个融合层,得到视觉编码子网络中首个融合层输出的全局图像特征和视觉融合特征;在场景编码子网络中,将初始的样本融合特征和第K-2个Transformer层预测得到的场景文本特征,输入场景编码子网络中的首个融合层,得到场景编码子网络中的首个融合层输出的场景文本特征和场景融合特征;对首个融合层所确定的视觉融合特征和场景融合特征进行融合,得到更经学习的样本融合特征。
之后,在视觉编码子网络中,将更经学习的样本融合特征和视觉编码子网络中的首个融合层输出的全局图像特征,输入至视觉编码子网络中的第二个融合层,得到经学习的全局图像特征和新的视觉融合特征;在场景编码子网络中,将更经学习的样本融合特征和场景编码子网络中的首个融合层输出的场景文本特征,输入至场景编码子网络中的第二个融合层,得到场景融合特征;对第二个融合层所确定的视觉融合特征和场景融合特征进行融合,得到更经学习的样本融合特征,作为最终的样本融合特征。
本公开实施例的技术方案,通过提取样本图文对中样本图像中的样本场景文本,并将样本图文对中样本文本输入文本编码网络,得到样本文本特征,之后将样本图像输入视觉编码子网络中的输入层,以及将初始的样本融合特征输入视觉编码子网络中的融合层,得到视觉编码子网络输出的样本图像的全局图像特征和视觉融合特征;并且,将样本场景文本输入场景编码子网络中的输入层,以及将初始的样本融合特征输入场景编码子网络中的融合层,得到场景编码子网络输出的场景融合特征,并对视觉融合特征和场景融合特征进行融合,得到经学习的样本融合特征,进而根据样本文本特征、样本图像的全局图像特征和经学习的样本融合特征,对视觉场景文本融合模型进行预训练。上述技术方案,引入在视觉编码子网络和场景编码子网络中分别引入融合层,使视觉场景编码网络可以充分学习场景文本特征,并采用场景文本特征预测全局图像特征,从而提高全局图像特征的准确性。
图3是根据本公开实施例提供的又一种视觉场景文本融合模型的训练方法的流程图。本实施例在上述实施例的基础上,对“根据样本文本特征、样本图像的全局图像特征和经学习的样本融合特征,对视觉场景文本融合模型进行预训练”进一步优化,提供一种可选方案。如图3所示,本实施例的视觉场景文本融合模型的训练方法可以包括:
S301,获取样本图文对。
其中,样本图文对包括样本图像和样本文本。
S302,提取样本图像中的样本场景文本。
S303,将样本文本输入文本编码网络,得到样本文本特征。
S304,将样本图像和初始的样本融合特征输入视觉编码子网络,以及将初始的样本融合特征和样本场景文本输入场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征。
S305,根据样本文本特征和经学习的样本融合特征,确定融合文本对比损失。
具体的,对于N组样本图文对,可以构成N2个图像-文本特征对,或者N2个融合-文本特征对,则对比损失是最大化其中正确的N对数据间的相似度,最小化N2-N对数据间的相似度,则定义融合文本对比损失为:
其中,
其中,Lf2c表示融合文本对比损失;fi表示第i组样本图文对经学习的样本融合特征;ti表示第i组样本图文对的样本文本特征;σ表示训练变量,初始可以设置为0.07。需要说明的是,N和i为正整数,i小于或等于N。
S306,根据样本图像的全局图像特征和样本文本特征,确定图像文本对比损失。
具体的,对于N组样本图文对,可以构成N2个图像-文本特征对,或者N2个全局-文本特征对,则对比损失是最大化其中正确的N对数据间的相似度,最小化N2-N对数据间的相似度,则定义图像文本对比损失为:
其中,
其中,Litc表示图像文本对比损失;vi表示第i组样本图文对对应的样本图像的全局图像特征;ti表示第i组样本图文对对应的样本文本特征;σ表示训练变量,初始可以设置为0.07。需要说明的是,N和i为正整数,i小于或等于N。
S307,根据融合文本对比损失和图像文本对比损失,确定训练损失。
具体的,可以对融合文本对比损失和图像文本对比损失进行加权求和,得到训练损失。例如可以通过如下公式确定训练损失:
Ltotal=αLitc+(1-α)Lftc
其中,Ltotal表示训练损失,Lftc表示融合文本对比损失,Litc表示图像文本对比损失,α表示一个参数,用于调整融合文本对比损失和图像文本对比损失之间的权重,优选的可以是0.9。
S308,采用训练损失,对视觉场景文本融合模型进行预训练。
本公开实施例的技术方案,通过提取样本图文对中样本图像中的样本场景文本,并将样本图文对中的样本文本输入文本编码网络,得到样本文本特征;以及将样本图文对中样本图像和初始的样本融合特征输入视觉编码子网络,以及将初始的样本融合特征和样本场景文本输入场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征,进而根据样本文本特征和经学习的样本融合特征,确定融合文本对比损失,根据样本图像的全局图像特征和样本文本特征,确定图像文本对比损失,并根据融合文本对比损失和图像文本对比损失,确定训练损失,采用训练损失,对视觉场景文本融合模型进行预训练。上述技术方案,通过计算样本文本特征和经学习的样本融合特征之间的损失,以及样本图像的全局图像特征和样本文本特征之间的损失,即双重对比损失,使得在有场景文本的情况下,提升视觉场景文本融合模型的图文检索精度;进一步的,由于双重对比损失中包含样本图像的全局图像特征和样本文本特征之间的损失,即图像与文本之间的损失,即使在没有场景文本的情况下,也可以保证视觉场景文本融合模型的图文检索精度。
在上述实施例的基础上,作为本公开的一种可选方式,根据融合文本对比损失和图像文本对比损失,确定训练损失可以是,确定样本场景文本是否为空文本;
若所述样本场景文本为空文本,则将所述图像文本对比损失作为所述训练损失;否则,将所述融合文本对比损失与所述图像文本对比损失之和,作为训练损失。
具体的,可以基于OCR技术确定样本场景文本是否为空文本,得到确定结果;若确定结果为空文本,则确定融合文本对比损失无效,将图像文本对比损失作为训练损失;若确定结果不为空文本,则确定融合文本对比损失有效,并根据融合文本对比损失和图像文本对比损失,确定训练损失。通过在样本场景文本为空文本的情况下,仅采用图像文本对比损失进行模型预训练,使得模型训练结果针对没有场景文本的图像仍然能够保持良好的性能。
在上述实施例的基础上,分别采用场景文本感知检索任务和传统图文检索任务进行实验。其中,场景文本感知检索任务,即为图像中具有场景文本场景下的图文检索;传统图文检索任务,即为图像中没有场景文本场景下的图文检索。
示例性的,在场景文本感知检索任务的过程中,采用CTC-1K、CTC-5K数据集进行实验,实验效果验证如表1。相较于现有的SCAN、VSRN、STARNet等跨模态检索技术,本公开的视觉场景文本融合模型(即ViSTA-S和ViSTA-B)在CTC数据集上表现优越,以CTC-1K文本检索任务为例,本公开的视觉场景文本融合模型取得了7.4%的提升。在增大模型体量的情况下,本公开的视觉场景文本融合模型可以获得更高的性能。
表1在CTC数据集上本公开技术方案与其他方法的性能对比
示例性的,在传统图文检索任务的过程中,采用Flickr30K、COCO-5K数据集进行实验,实验验证效果如表2。在相近耗时的情况下,本公开的视觉场景文本融合模型(即ViSTA-B模型)相比于IMRAM、GSMN、SGRAF、Vil-BERT、Unicoder-VL、UNITER-B、ERNIE-Vil-B、Pixel-BERT-X、SOHO、H Xue et al.、Pixel-BERT-R、ViLT-B等跨模态检索技术,具有更优异的性能;并且,在增大模型体量的情况下,本公开的视觉场景文本融合模型(即ViSTA-L模型)可以和耗时较大的单塔模型可比。
表2在Flickr30K、COCO-5K数据集上本公开的技术方案与其他方法的性能对比
综上,场景文本感知数据集、传统图文检索数据集实验验证了本公开的视觉场景文本融合模型的有效性,能够同时在有无场景文本的图文检索任务下取得公开指标最佳水平。
需要说明的是,本公开通过图文对比预训练后得到的视觉场景文本融合模型可直接用于下游检索任务,也可以供任一业务方对预训练后的视觉场景文本融合模型进行微调,以满足实际业务需求。
可选的,本公开实施例还提供了一种视觉场景文本融合模型的训练方法,具体包括:获取业务方提供的业务图文对;将业务图像和业务文本作为训练数据,对视觉场景文本融合模型进行微调;其中,视觉场景文本融合模型基于本公开任一种视觉场景文本融合模型的预训练方法得到。其中,业务图文对包括业务图像和业务文本,业务文本用于对业务图像中的图像内容进行描述。
具体的,获取业务方的业务图文对,并将业务图文对作为训练数据;将训练数据输入视觉场景文本融合模型中,对视觉场景文本融合模型中的模型参数进行调整。
可以理解的是,本实施例通过对预训练后的视觉场景文本融合模型进行微调,使得视觉场景文本融合模型更适用于实际业务场景。
图4是根据本公开实施例提供的一种视觉场景文本融合模型的图文检索方法的流程图。本实施例适用于如何进行图文检索的情况。该方法可以由图文检索装置来执行,该装置可以采用软件和/或硬件的方式实现,并可集成于承载图文检索功能的电子设备中。视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,视觉场景编码网络包括视觉编码子网络和场景编码子网络。如图4所示,本实施例的视觉场景文本融合模型的图文检索方法可以包括:
S401,获取待检索的目标文本。
具体的,可以获取搜索需求方输入的目标文本。例如可以是,搜索需求方通过搜索交互界面,输入目标文本。
S402,提取候选图像中的候选场景文本。
本实施例中,所谓候选图像即为可供选择的图像,例如可以是待检索图像库中的部分或全部图像。所谓候选场景文本是指候选图像中的文本,可以包括一个或多个词语。
具体的,可以基于光学字符识别(Optical Character Recognition,OCR)技术,对每一候选图像进行处理,以得到该候选图像中的候选场景文本,进一步的,还可与得到候选场景文本在该候选图像中的图位置编码信息。需要说明的是,候选图像中可以包括一个或多个候选场景文本;对于每一候选场景文本,可以包括一个或多个词语。
S403,将目标文本输入文本编码网络,得到目标文本特征。
本实施例中,目标文本特征是指目标文本的语义特征,可以采用矩阵或向量形式表示。具体的,可以将目标文本输入至文本编码网络,经过文本编码网络处理,得到目标文本特征。
S404,将候选图像和初始的候选融合特征输入视觉编码子网络,以及将初始的候选融合特征和候选场景文本输入场景编码子网络,得到候选图像的全局图像特征。
本实施例中,候选融合特征用于表征候选图像和文本之间的相关性特征,可以采用矩阵或向量形式;需要说明的是,初始的候选融合特征为预先设定的图像和文本之间的相关性特征。所谓候选图像的全局图像特征是指从全局角度所提取的候选图像的特征,可以采用矩阵或向量形式。
具体的,可以将候选图像输入视觉编码子网络中的输入层,将初始的候选融合特征输入视觉编码子网络中的融合层,以及将候选场景文本输入场景编码子网络中的输入层,将初始的候选融合特征输入场景编码子网络中的融合层,由视觉编码子网络和场景编码子网络配合处理,得到候选图像的全局图像特征。
需要说明的是,在模型的训练阶段,视觉场景编码网络能够充分学习到视觉和场景文本两者之间的相关性,进行全局图像特征的预测;因此,在模型的使用阶段,视觉场景编码网络输出的全局图像特征中融合了场景文本的相关特征。
S405,根据目标文本特征和候选图像的全局图像特征,从候选图像中确定目标图像。
具体的,对于每一候选图像,可以确定目标文本特征和该候选图像的全局图像特征之间的相似度;并且,根据相似度,从候选图像中确定目标图像,例如可以将相似度最大的候选图像,作为目标图像。
需要说明的是,本实施例中的视觉场景文本融合模型可以通过本公开任一种视觉场景文本融合模型的预训练得到,也可以通过本公开任一种视觉场景文本融合模型的训练方法得到。
本公开实施例的技术方案,通过提取候选图像中的候选场景文本,并将所获取的目标文本输入文本编码网络,得到目标文本特征,之后将候选图像和初始的候选融合特征输入视觉编码子网络,以及将初始的候选融合特征和候选场景文本输入场景编码子网络,得到候选图像的全局图像特征,进而根据目标文本特征和候选图像的全局图像特征,从候选图像中确定目标图像。上述技术方案,通过并行的文本编码网络和视觉场景编码网络,这种双编码结构的视觉场景文本融合模型来进行图文检索,提高了图文检索的效率;并且,通过引入融合特征,无论在有无场景文本的情况下,都可以有效学习到图像的全局图像特征,从而提高了图文检索的精度。
在上述实施例的基础上,作为本公开的一种可选方式,将目标文本输入文本编码网络,得到目标文本特征可以是,对目标文本进行词嵌入,得到目标文本词向量;根据目标文本的模态信息、目标文本的位置编码信息和目标文本词向量,确定目标文本的词编码结果;根据初始的目标文本特征和目标文本的词编码结果,构建目标文本的编码序列;将目标文本的编码序列输入文本编码网络,得到经处理的目标文本特征。
本实施例中,目标文本的模态信息即为预先设定的用于表征文本这一模态的标识信息,其取值与样本文本的模态信息相同。所谓目标文本的位置编码信息即为目标文本中词语在目标文本中位置标识信息。
具体的,可以基于词嵌入技术,对目标文本进行处理,得到目标文本中每一词语的向量,即目标文本词向量。
之后,针对目标文本中的每一词语,可以基于一定的规则,将目标文本的模态信息、该词语的位置编码信息和该词语的目标文本词向量进行拼接,得到该词语的编码结果。
在确定目标文本中各词语的编码结果后,可以根据初始的目标文本特征和各词语的编码结果,构建目标文本的编码序列。其中,初始的目标文本特征具体可由待学习的文本特征、目标文本的模态信息和位置编码信息构成;可选的,初始的目标文本特征位于目标文本的编码序列的首位。在确定目标文本的编码序列后,可以将目标文本的编码序列输入文本编码网络,得到目标文本特征。
可以理解的是,结合目标文本的模态信息、位置编码信息和目标文本词向量等,确定目标文本特征,使得最终所确定的目标文本特征更加准确。
在上述实施例的基础上,作为本公开的一种可选方式,将候选图像输入视觉编码子网络可以是,对候选图像进行分块,得到候选图像块序列;对候选图像块序列中的候选图像块进行线性投影,得到候选图像块的编码结果;根据候选图像块的模态信息和候选图像块的位置编码信息对候选图像块的编码结果进行处理,得到候选图像块经处理的编码结果;根据初始的全局图像特征和候选图像块经处理的编码结果,构建候选图像的编码序列;将候选图像的编码序列输入视觉编码子网络中的输入层。
本实施例中,所谓候选图像块的模态信息即为预先设定的用于表征图像这一模态的标识信息,其取值可与样本图像的模态信息相同。所谓候选图像块的位置编码信息即为候选图像块的标识信息。
具体的,可以按照一定的尺寸,将候选图像分割成若干个(比如N块,N为正整数)大小相同的候选图像块,得到候选图像块序列。
在得到候选图像块序列后,对于候选图像块序列中的每一候选图像块,对该候选图像块进行线性投影,得到该候选图像块的编码结果。并且,可以基于一定的规则,将该候选图像块的编码结果、候选图像块的模态信息和候选图像块的位置编码信息进行拼接,得到该候选图像块经处理的编码结果。
在确定各候选图像块经处理的编码结果后,可以根据初始的全局图像特征和各候选图像块经处理的编码结果,构建候选图像的编码序列,可以将候选图像的编码序列输入视觉编码子网络中的输入层。其中,初始的全局图像特征具体可以由待学习的图像编码、候选图像块的模态信息和候选图像块的位置编码信息构成。
可以理解的是,本公开使用图像分块线性投影的方式,确定候选图像块的编码结果,相比于采用检测器来获取图像中图像块的编码结果而言,减少了计算复杂度,提高了效率;同时,结合候选图像块的模态信息、候选块的位置编码信息和初始的全局图像特征等,来确定候选图像的编码序列,使得视觉编码子网络能够提取到候选图像中更加丰富的特征。
在上述实施例的基础上,作为本公开的一种可选方式,将候选场景文本输入场景编码子网络可以是,对候选场景文本进行词嵌入,得到候选场景文本向量;根据候选场景文本的图位置编码信息、候选场景文本的模态信息、候选场景文本的字位置编码信息和候选场景文本向量,确定候选场景文本的编码结果;根据初始的候选场景文本特征和候选场景文本的编码结果,构建候选场景文本的编码序列;将候选场景文本的编码序列输入场景编码子网络中的输入层。
本实施例中,候选场景文本的模态信息即为预先设定的用于表征场景文本这一模态的标识信息,其取值可与样本场景文本的模态信息相同。所谓候选场景文本的字位置编码信息用于唯一性标识候选场景文本中字的顺序。所谓候选场景文本的图位置编码信息用于唯一性标识候选场景文本在候选图像中的图像位置信息。
具体的,针对从候选图像中提取到的每一候选场景文本,可以基于词嵌入技术,对该候选场景文本进行处理,得到该候选场景文本的候选场景文本向量。并且,可以基于一定的规则,将该候选场景文本的图位置编码信息、候选场景文本的模态信息、候选场景文本的字位置编码信息和候选场景文本向量进行拼接,得到该候选场景文本的编码结果。以及,可以根据初始的候选场景文本特征和各候选场景文本的编码结果,构建候选场景文本的编码序列,可以将候选场景文本的编码序列输入场景编码子网络中的输入层。其中,初始的候选场景文本特征具体可以由待学习的场景文本特征、候选场景文本的模态信息、候选场景文本的图位置编码信息和字位置编码信息构成。
可以理解的是,结合候选场景文本的图位置编码信息、候选场景文本的模态信息、候选场景文本的字位置编码信息和初始的候选视觉特征等,来确定候选场景文本的编码序列,使得视觉编码子网络能够提取到候选场景文本中更加丰富的特征。
图5是根据本公开实施例提供的一种视觉场景文本融合模型的预训练装置的结构示意图。本实施例适用于如何对视觉场景文本融合模型进行预训练的情况。该装置可以采用软件和/或硬件的方式实现,并可集成于承载视觉场景文本融合模型的预训练功能的电子设备中。可选的,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,视觉场景编码网络包括视觉编码子网络和场景编码子网络。如图5所示,本实施例的视觉场景文本融合模型的预训练装置500可以包括:
样本图文对获取模块501,用于获取样本图文对;其中,样本图文对包括样本图像和样本文本;
样本场景文本提取模块502,用于提取样本图像中的样本场景文本;
样本文本特征确定模块503,用于将样本文本输入文本编码网络,得到样本文本特征;
样本全局特征确定模块504,用于将所述样本图像和初始的样本融合特征输入所述视觉编码子网络,以及将初始的样本融合特征和所述样本场景文本输入所述场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征;
模型预训练模块505,用于根据所述样本文本特征、所述样本图像的全局图像特征和经学习的样本融合特征,对所述视觉场景文本融合模型进行预训练。
本公开实施例的技术方案,通过将样本图文对中的样本文本输入文本编码网络,得到样本文本特征;以及将样本图文对中的样本图像和初始的样本融合特征输入视觉编码子网络,将初始的样本融合特征和从样本图像中提取的样本场景文本输入场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征,进而根据样本文本特征、样本图像的全局图像特征和经学习的样本融合特征,对视觉场景文本融合模型进行预训练。上述技术方案,引入样本融合特征、样本图像的全局图像特征和样本文本特征对视觉场景文本融合模型进行预训练,使得无论输入的图像是否有场景文本,均可以准确地输出图像的全局图像特征,从而使得视觉场景文本融合模型可以支持在大范围图文集下的检索任务,也就是说,视觉场景文本融合模型不仅适用于有场景文本情况下的跨模态检索,同样适用于没有场景文本情况下的跨模态检索;此外,本方案通过双编码结构即并行的文本编码网络和视觉场景编码网络,来对图像和文本进行并行处理,可以提高跨模态检索效率。
进一步地,样本全局特征确定模块504具体用于:
将样本图像输入视觉编码子网络中的输入层,以及将初始的样本融合特征输入视觉编码子网络中的融合层,得到视觉编码子网络输出的样本图像的全局图像特征和视觉编码子网络输出的视觉融合特征;
将样本场景文本输入场景编码子网络中的输入层,以及将初始的样本融合特征输入场景编码子网络中的融合层,得到场景编码子网络输出的场景融合特征;
对视觉编码子网络输出的视觉融合特征和场景编码子网络输出的场景融合特征进行融合,得到经学习的样本融合特征。
进一步地,样本文本特征确定模块503具体用于:
对样本文本进行词嵌入,得到样本文本词向量;
根据样本文本的模态信息、样本文本的位置编码信息和样本文本词向量,确定样本文本的词编码结果;
根据初始的样本文本特征和样本文本的词编码结果,构建样本文本的编码序列;
将样本文本的编码序列输入文本编码网络,得到经学习的样本文本特征。
进一步地,样本全局特征确定模块504具体用于:
对样本图像进行分块,得到样本图像块序列;
对样本图像块序列中的样本图像块进行线性投影,得到样本图像块的样本的编码结果;
根据样本图像块的模态信息和样本图像块的位置编码信息述样本图像块的编码结果进行处理,得到样本图像块经处理的编码结果;
根据初始的全局图像特征和样本图像块经处理的编码结果,构建样本图像的编码序列;
将样本图像的编码序列输入视觉编码子网络中的输入层。
进一步地,样本全局特征确定模块504具体用于:
对样本场景文本进行词嵌入,得到样本场景文本向量;
根据样本场景文本的图位置编码信息、样本场景文本的模态信息、样本场景文本的字位置编码信息和样本场景文本向量,确定样本场景文本的编码结果;
根据初始的样本场景文本特征和样本场景文本的编码结果,构建样本场景文本的编码序列;
将样本场景文本的编码序列输入场景编码子网络中的输入层。
进一步地,模型预训练模块505,包括:
融合对比损失单元,用于根据样本文本特征和经学习的样本融合特征,确定融合文本对比损失;
图文对比损失单元,用于根据样本图像的全局图像特征和经学习的样本文本特征,确定图像文本对比损失;
训练损失单元,用于根据融合文本对比损失和图像文本对比损失,确定训练损失;
模型预训练单元,用于采用训练损失,对视觉场景文本融合模型进行预训练。
进一步地,训练损失单元具体用于:
确定样本场景文本是否为空文本;
若样本场景文本为空文本,则将图像文本对比损失作为训练损失;否则,将融合文本对比损失与图像文本对比损失之和,作为训练损失。
图6是根据本公开实施例提供的一种视觉场景文本融合模型的训练装置的结构示意图;该装置可以采用软件和/或硬件的方式实现,并可集成于承载视觉场景文本融合模型的训练功能的电子设备中。如图6所示,本实施例的视觉场景文本融合模型的训练装置600可以包括:
业务图文对获取模块601,用于获取业务方提供的业务图文对;所述业务图文对包括业务图像和业务文本;
微调模块602,用于将所述业务图像和所述业务文本作为训练数据,对视觉场景文本融合模型进行微调;其中,所述视觉场景文本融合模型基于本公开中任一种视觉场景文本融合模型的预训练方法得到。
本实施例的技术方案,通过对预训练后的视觉场景文本融合模型进行微调,使得视觉场景文本融合模型更适用于实际业务场景。
图7是根据本公开实施例提供的一种视觉场景文本融合模型的图文检索装置的结构示意图。本实施例适用于如何进行图文检索的情况。该装置可以采用软件和/或硬件的方式实现,并可集成于承载图文检索功能的电子设备中。其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络。如图7所示,本实施例的视觉场景文本融合模型的图文检索装置700可以包括:
目标文本获取模块701,用于获取待检索的目标文本;
候选场景文本提取模块702,用于提取候选图像中的候选场景文本;
目标文本特征确定模块703,用于将目标文本输入文本编码网络,得到目标文本特征;
候选全局特征确定模块704,用于将候选图像和初始的候选融合特征输入视觉编码子网络,以及将初始的候选融合特征和候选场景文本输入场景编码子网络,得到候选图像的全局图像特征;
目标图像确定模块705,用于根据目标文本特征和候选图像的全局图像特征,从候选图像中确定目标图像。
本公开实施例的技术方案,通过提取候选图像中的候选场景文本,并将所获取的目标文本输入文本编码网络,得到目标文本特征,将候选图像和初始的候选融合特征输入视觉编码子网络,以及将初始的候选融合特征和候选场景文本输入场景编码子网络,得到候选图像的全局图像特征,进而根据目标文本特征和候选图像的全局图像特征,从候选图像中确定目标图像。上述技术方案,通过并行的文本编码网络和视觉场景编码网络,这种双编码结构的视觉场景文本融合模型来进行图文检索,提高了图文检索的效率;并且,通过引入候选融合特征,无论在有无场景文本的情况下,都可以有效学习到候选图像的全局图像特征,从而提高了图文检索的精度。
进一步地,目标文本特征确定模块703具体用于:
对目标文本进行词嵌入,得到目标文本词向量;
根据目标文本的模态信息、目标文本的位置编码信息和目标文本词向量,确定目标文本的词编码结果;
根据初始的目标文本特征和目标文本的词编码结果,构建目标文本的编码序列;
将目标文本的编码序列输入文本编码网络,得到经处理的目标文本特征。
进一步地,候选全局特征确定模块704具体用于:
对候选图像进行分块,得到候选图像块序列;
对候选图像块序列中的候选图像块进行线性投影,得到候选图像块的编码结果;
根据候选图像块的模态信息和候选图像块的位置编码信息对候选图像块的编码结果进行处理,确定候选图像块经处理的编码结果;
根据初始的全局图像特征和候选图像块经处理的编码结果,构建候选图像的编码序列;
将候选图像的编码序列输入视觉编码子网络中的输入层。
进一步地,候选全局特征确定模块704还具体用于:
对候选场景文本进行词嵌入,得到候选场景文本向量;
根据所述候选场景文本的图位置编码信息、候选场景文本的模态信息、候选场景文本的字位置编码信息和候选场景文本向量,确定候选场景文本的编码结果;
根据初始的候选场景文本特征和候选场景文本的编码结果,构建候选场景文本的编码序列;
将候选场景文本的编码序列输入场景编码子网络中的输入层。
进一步地,视觉场景文本融合模型基于本公开任一种视觉场景文本融合模型的训练装置得到。
本公开的技术方案中,所涉及的样本图文对、目标文本、候选图像等的收集、存储、使用、加工、传输、提供和公开等,均符合相关法律法规的规定,且不违背公序良俗。
根据本公开的实施例,本公开还提供了一种电子设备、一种可读存储介质和一种计算机程序产品。
图8是用来实现本公开实施例的视觉场景文本融合模型的训练方法、训练方法或图文检索方法的电子设备的框图。图8示出了可以用来实施本公开的实施例的示例电子设备800的示意性框图。电子设备旨在表示各种形式的数字计算机,诸如,膝上型计算机、台式计算机、工作台、个人数字助理、服务器、刀片式服务器、大型计算机、和其它适合的计算机。电子设备还可以表示各种形式的移动装置,诸如,个人数字处理、蜂窝电话、智能电话、可穿戴设备和其它类似的计算装置。本文所示的部件、它们的连接和关系、以及它们的功能仅仅作为示例,并且不意在限制本文中描述的和/或者要求的本公开的实现。
如图8所示,电子设备800包括计算单元801,其可以根据存储在只读存储器(ROM)802中的计算机程序或者从存储单元808加载到随机访问存储器(RAM)803中的计算机程序,来执行各种适当的动作和处理。在RAM 803中,还可存储电子设备800操作所需的各种程序和数据。计算单元801、ROM 802以及RAM 803通过总线804彼此相连。输入/输出(I/O)接口805也连接至总线804。
电子设备800中的多个部件连接至I/O接口805,包括:输入单元806,例如键盘、鼠标等;输出单元807,例如各种类型的显示器、扬声器等;存储单元808,例如磁盘、光盘等;以及通信单元809,例如网卡、调制解调器、无线通信收发机等。通信单元809允许电子设备800通过诸如因特网的计算机网络和/或各种电信网络与其他设备交换信息/数据。
计算单元801可以是各种具有处理和计算能力的通用和/或专用处理组件。计算单元801的一些示例包括但不限于中央处理单元(CPU)、图形处理单元(GPU)、各种专用的人工智能(AI)计算芯片、各种运行机器学习模型算法的计算单元、数字信号处理器(DSP)、以及任何适当的处理器、控制器、微控制器等。计算单元801执行上文所描述的各个方法和处理,例如视觉场景文本融合模型的预训练方法、训练方法或图文检索方法。例如,在一些实施例中,视觉场景文本融合模型的预训练方法、训练方法或图文检索方法可被实现为计算机软件程序,其被有形地包含于机器可读介质,例如存储单元808。在一些实施例中,计算机程序的部分或者全部可以经由ROM 802和/或通信单元809而被载入和/或安装到电子设备800上。当计算机程序加载到RAM 803并由计算单元801执行时,可以执行上文描述的视觉场景文本融合模型的预训练方法、训练方法或图文检索方法的一个或多个步骤。备选地,在其他实施例中,计算单元801可以通过其他任何适当的方式(例如,借助于固件)而被配置为执行视觉场景文本融合模型的预训练方法、训练方法或图文检索方法。
本文中以上描述的***和技术的各种实施方式可以在数字电子电路***、集成电路***、现场可编程门阵列(FPGA)、专用集成电路(ASIC)、专用标准产品(ASSP)、芯片上***的***(SOC)、复杂可编程逻辑设备(CPLD)、计算机硬件、固件、软件、和/或它们的组合中实现。这些各种实施方式可以包括:实施在一个或者多个计算机程序中,该一个或者多个计算机程序可在包括至少一个可编程处理器的可编程***上执行和/或解释,该可编程处理器可以是专用或者通用可编程处理器,可以从存储***、至少一个输入装置、和至少一个输出装置接收数据和指令,并且将数据和指令传输至该存储***、该至少一个输入装置、和该至少一个输出装置。
用于实施本公开的方法的程序代码可以采用一个或多个编程语言的任何组合来编写。这些程序代码可以提供给通用计算机、专用计算机或其他可编程数据处理装置的处理器或控制器,使得程序代码当由处理器或控制器执行时使流程图和/或框图中所规定的功能/操作被实施。程序代码可以完全在机器上执行、部分地在机器上执行,作为独立软件包部分地在机器上执行且部分地在远程机器上执行或完全在远程机器或服务器上执行。
在本公开的上下文中,机器可读介质可以是有形的介质,其可以包含或存储以供指令执行***、装置或设备使用或与指令执行***、装置或设备结合地使用的程序。机器可读介质可以是机器可读信号介质或机器可读储存介质。机器可读介质可以包括但不限于电子的、磁性的、光学的、电磁的、红外的、或半导体***、装置或设备,或者上述内容的任何合适组合。机器可读存储介质的更具体示例会包括基于一个或多个线的电气连接、便携式计算机盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦除可编程只读存储器(EPROM或快闪存储器)、光纤、便捷式紧凑盘只读存储器(CD-ROM)、光学储存设备、磁储存设备、或上述内容的任何合适组合。
为了提供与用户的交互,可以在计算机上实施此处描述的***和技术,该计算机具有:用于向用户显示信息的显示装置(例如,CRT(阴极射线管)或者LCD(液晶显示器)监视器);以及键盘和指向装置(例如,鼠标或者轨迹球),用户可以通过该键盘和该指向装置来将输入提供给计算机。其它种类的装置还可以用于提供与用户的交互;例如,提供给用户的反馈可以是任何形式的传感反馈(例如,视觉反馈、听觉反馈、或者触觉反馈);并且可以用任何形式(包括声输入、语音输入或者、触觉输入)来接收来自用户的输入。
可以将此处描述的***和技术实施在包括后台部件的计算***(例如,作为数据服务器)、或者包括中间件部件的计算***(例如,应用服务器)、或者包括前端部件的计算***(例如,具有图形用户界面或者网络浏览器的用户计算机,用户可以通过该图形用户界面或者该网络浏览器来与此处描述的***和技术的实施方式交互)、或者包括这种后台部件、中间件部件、或者前端部件的任何组合的计算***中。可以通过任何形式或者介质的数字数据通信(例如,通信网络)来将***的部件相互连接。通信网络的示例包括:局域网(LAN)、广域网(WAN)和互联网。
计算机***可以包括客户端和服务器。客户端和服务器一般远离彼此并且通常通过通信网络进行交互。通过在相应的计算机上运行并且彼此具有客户端-服务器关系的计算机程序来产生客户端和服务器的关系。服务器可以是云服务器,也可以为分布式***的服务器,或者是结合了区块链的服务器。
人工智能是研究使计算机来模拟人的某些思维过程和智能行为(如学习、推理、思考、规划等)的学科,既有硬件层面的技术也有软件层面的技术。人工智能硬件技术一般包括如传感器、专用人工智能芯片、云计算、分布式存储、大数据处理等技术;人工智能软件技术主要包括计算机视觉技术、语音识别技术、自然语言处理技术及机器学习/深度学习技术、大数据处理技术、知识图谱技术等几大方向。
云计算(cloud computing),指的是通过网络接入弹性可扩展的共享物理或虚拟资源池,资源可以包括服务器、操作***、网络、软件、应用和存储设备等,并可以按需、自服务的方式对资源进行部署和管理的技术体系。通过云计算技术,可以为人工智能、区块链等技术应用、模型训练提供高效强大的数据处理能力。
应该理解,可以使用上面所示的各种形式的流程,重新排序、增加或删除步骤。例如,本发公开中记载的各步骤可以并行地执行也可以顺序地执行也可以不同的次序执行,只要能够实现本公开公开的技术方案所期望的结果,本文在此不进行限制。
上述具体实施方式,并不构成对本公开保护范围的限制。本领域技术人员应该明白的是,根据设计要求和其他因素,可以进行各种修改、组合、子组合和替代。任何在本公开的精神和原则之内所作的修改、等同替换和改进等,均应包含在本公开保护范围之内。
Claims (29)
1.一种视觉场景文本融合模型的预训练方法,其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络,所述方法包括:
获取样本图文对;其中,所述样本图文对包括样本图像和样本文本;
提取所述样本图像中的样本场景文本;
将所述样本文本输入所述文本编码网络,得到样本文本特征;
将所述样本图像和初始的样本融合特征输入所述视觉编码子网络,以及将初始的样本融合特征和所述样本场景文本输入所述场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征;
根据所述样本文本特征、所述样本图像的全局图像特征和经学习的样本融合特征,对所述视觉场景文本融合模型进行预训练。
2.根据权利要求1所述的方法,其中,所述将所述样本图像和初始的样本融合特征输入所述视觉编码网络,以及将初始的样本融合特征和所述样本场景文本输入所述场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征,包括:
将所述样本图像输入所述视觉编码子网络中的输入层,以及将初始的样本融合特征输入所述视觉编码子网络中的融合层,得到视觉编码子网络输出的样本图像的全局图像特征和视觉编码子网络输出的视觉融合特征;
将所述样本场景文本输入所述场景编码子网络中的输入层,以及将初始的样本融合特征输入所述场景编码子网络中的融合层,得到所述场景编码子网络输出的场景融合特征;
对所述视觉编码子网络输出的视觉融合特征和所述场景编码子网络输出的场景融合特征进行融合,得到经学习的样本融合特征。
3.根据权利要求1所述的方法,其中,所述将所述样本文本输入所述文本编码网络,得到样本文本特征,包括:
对所述样本文本进行词嵌入,得到样本文本词向量;
根据样本文本的模态信息、样本文本的位置编码信息和所述样本文本词向量,确定所述样本文本的词编码结果;
根据初始的样本文本特征和所述样本文本的词编码结果,构建样本文本的编码序列;
将所述样本文本的编码序列输入所述文本编码网络,得到经学习的样本文本特征。
4.根据权利要求1所述的方法,其中,将所述样本图像输入所述视觉编码子网络,包括:
对所述样本图像进行分块,得到样本图像块序列;
对所述样本图像块序列中的样本图像块进行线性投影,得到样本图像块的编码结果;
根据样本图像块的模态信息和样本图像块的位置编码信息对所述样本图像块的编码结果进行处理,得到样本图像块经处理的编码结果;
根据初始的全局图像特征和样本图像块经处理的编码结果,构建样本图像的编码序列;
将所述样本图像的编码序列输入所述视觉编码子网络中的输入层。
5.根据权利要求1所述的方法,其中,将所述样本场景文本输入所述场景编码子网络,包括:
对所述样本场景文本进行词嵌入,得到样本场景文本向量;
根据样本场景文本的图位置编码信息、样本场景文本的模态信息、样本场景文本的字位置编码信息和所述样本场景文本向量,确定所述样本场景文本的编码结果;
根据初始的样本场景文本特征和所述样本场景文本的编码结果,构建样本场景文本的编码序列;
将所述样本场景文本的编码序列输入所述场景编码子网络中的输入层。
6.根据权利要求1所述的方法,其中,所述根据所述样本文本特征、所述样本图像的全局图像特征和经学习的样本融合特征,对所述视觉场景文本融合模型进行预训练,包括:
根据所述样本文本特征和经学习的样本融合特征,确定融合文本对比损失;
根据所述样本图像的全局图像特征和所述样本文本特征,确定图像文本对比损失;
根据所述融合文本对比损失和所述图像文本对比损失,确定训练损失;
采用所述训练损失,对所述视觉场景文本融合模型进行预训练。
7.根据权利要求6所述的方法,其中,所述根据融合文本对比损失和所述图像文本对比损失,确定训练损失,包括:
确定所述样本场景文本是否为空文本;
若所述样本场景文本为空文本,则将所述图像文本对比损失作为所述训练损失;否则,将所述融合文本对比损失与所述图像文本对比损失之和,作为训练损失。
8.一种视觉场景文本融合模型的训练方法,包括:
获取业务方提供的业务图文对;所述业务图文对包括业务图像和业务文本;
将所述业务图像和所述业务文本作为训练数据,对视觉场景文本融合模型进行微调;其中,所述视觉场景文本融合模型基于权利要求1-7中任一项所述的一种视觉场景文本融合模型的预训练方法得到。
9.一种视觉场景文本融合模型的图文检索方法,其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络,所述方法包括:
获取待检索的目标文本;
提取候选图像中的候选场景文本;
将所述目标文本输入所述文本编码网络,得到目标文本特征;
将所述候选图像和初始的候选融合特征输入所述视觉编码子网络,以及将初始的候选融合特征和所述候选场景文本输入所述场景编码子网络,得到所述候选图像的全局图像特征;
根据所述目标文本特征和所述候选图像的全局图像特征,从所述候选图像中确定目标图像。
10.根据权利要求9所述的方法,其中,所述将所述目标文本输入所述文本编码网络,得到目标文本特征,包括:
对所述目标文本进行词嵌入,得到目标文本词向量;
根据所述目标文本的模态信息、目标文本的位置编码信息和所述目标文本词向量,确定所述目标文本的词编码结果;
根据初始的目标文本特征和所述目标文本的词编码结果,构建目标文本的编码序列;
将所述目标文本的编码序列输入所述文本编码网络,得到经处理的目标文本特征。
11.根据权利要求9所述的方法,其中,将所述候选图像输入所述视觉编码子网络,包括:
对所述候选图像进行分块,得到候选图像块序列;
对所述候选图像块序列中的候选图像块进行线性投影,得到候选图像块的编码结果;
根据候选图像块的模态信息和候选图像块的位置编码信息对所述候选图像块的编码结果进行处理,确定所述候选图像块经处理的编码结果;
根据初始的全局图像特征和所述候选图像块经处理的编码结果,构建候选图像的编码序列;
将所述候选图像的编码序列输入所述视觉编码子网络中的输入层。
12.根据权利要求9所述的方法,其中,将所述候选场景文本输入所述场景编码子网络,包括:
对所述候选场景文本进行词嵌入,得到候选场景文本向量;
根据所述候选场景文本的图位置编码信息、候选场景文本的模态信息、候选场景文本的字位置编码信息和所述候选场景文本向量,确定所述候选场景文本的编码结果;
根据初始的候选场景文本特征和所述候选场景文本的编码结果,构建候选场景文本的编码序列;
将所述候选场景文本的编码序列输入所述场景编码子网络中的输入层。
13.根据权利要求9所述的方法,其中,所述视觉场景文本融合模型基于权利要求8所述的一种视觉场景文本融合模型的训练方法得到。
14.一种视觉场景文本融合模型的预训练装置,其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络,所述装置包括:
样本图文对获取模块,用于获取样本图文对;其中,所述样本图文对包括样本图像和样本文本;
样本场景文本提取模块,用于提取所述样本图像中的样本场景文本;
样本文本特征确定模块,用于将所述样本文本输入所述文本编码网络,得到样本文本特征;
样本全局特征确定模块,用于将所述样本图像和初始的样本融合特征输入所述视觉编码子网络,以及将初始的样本融合特征和所述样本场景文本输入所述场景编码子网络,得到样本图像的全局图像特征和经学习的样本融合特征;
模型预训练模块,用于根据所述样本文本特征、所述样本图像的全局图像特征和经学习的样本融合特征,对所述视觉场景文本融合模型进行预训练。
15.根据权利要求14所述的装置,其中,所述样本全局特征确定模块具体用于:
将所述样本图像输入所述视觉编码子网络中的输入层,以及将初始的样本融合特征输入所述视觉编码子网络中的融合层,得到视觉编码子网络输出的样本图像的全局图像特征和视觉编码子网络输出的视觉融合特征;
将所述样本场景文本输入所述场景编码子网络中的输入层,以及将初始的样本融合特征输入所述场景编码子网络中的融合层,得到所述场景编码子网络输出的场景融合特征;
对所述视觉编码子网络输出的视觉融合特征和所述场景编码子网络输出的场景融合特征进行融合,得到经学习的样本融合特征。
16.根据权利要求14所述的装置,其中,所述样本文本特征确定模块具体用于:
对所述样本文本进行词嵌入,得到样本文本词向量;
根据样本文本的模态信息、样本文本的位置编码信息和所述样本文本词向量,确定所述样本文本的词编码结果;
根据初始的样本文本特征和所述样本文本的词编码结果,构建样本文本的编码序列;
将所述样本文本的编码序列输入所述文本编码网络,得到经学习的样本文本特征。
17.根据权利要求14所述的装置,其中,所述样本全局特征确定模块具体用于:
对所述样本图像进行分块,得到样本图像块序列;
对所述样本图像块序列中的样本图像块进行线性投影,得到所述样本图像块的编码结果;
根据样本图像块的模态信息和样本图像块的位置编码信息对所述样本图像块的编码结果进行处理,得到样本图像块经处理的编码结果;
根据初始的全局图像特征和样本图像块经处理的编码结果,构建样本图像的编码序列;
将所述样本图像的编码序列输入所述视觉编码子网络中的输入层。
18.根据权利要求14所述的装置,其中,所述样本全局特征确定模块具体用于:
对所述样本场景文本进行词嵌入,得到样本场景文本向量;
根据样本场景文本的图位置编码信息、样本场景文本的模态信息、样本场景文本的字位置编码信息和所述样本场景文本向量,确定所述样本场景文本的编码结果;
根据初始的样本场景文本特征和所述样本场景文本的编码结果,构建样本场景文本的编码序列;
将所述样本场景文本的编码序列输入所述场景编码子网络中的输入层。
19.根据权利要求14所述的装置,其中,所述模型预训练模块,包括:
融合对比损失单元,用于根据所述样本文本特征和经学习的样本融合特征,确定融合文本对比损失;
图文对比损失单元,用于根据所述样本图像的全局图像特征和经学习的样本文本特征,确定图像文本对比损失;
训练损失单元,用于根据所述融合文本对比损失和所述图像文本对比损失,确定训练损失;
模型预训练单元,用于采用所述训练损失,对所述视觉场景文本融合模型进行预训练。
20.根据权利要求19所述的装置,其中,所述训练损失单元具体用于:
确定所述样本场景文本是否为空文本;
若所述样本场景文本为空文本,则将所述图像文本对比损失作为所述训练损失;否则,将所述融合文本对比损失与所述图像文本对比损失之和,作为所述训练损失。
21.一种视觉场景文本融合模型的训练装置,包括:
业务图文对获取模块,用于获取业务方提供的业务图文对;所述业务图文对包括业务图像和业务文本;
微调模块,用于将所述业务图像和所述业务文本作为训练数据,对视觉场景文本融合模型进行微调;其中,所述视觉场景文本融合模型基于权利要求14-20中任一项所述的一种视觉场景文本融合模型的预训练方法得到。
22.一种视觉场景文本融合模型的图文检索装置,其中,视觉场景文本融合模型包括文本编码网络和视觉场景编码网络,所述视觉场景编码网络包括视觉编码子网络和场景编码子网络,所述装置包括:
目标文本获取模块,用于获取待检索的目标文本;
候选场景文本提取模块,用于提取候选图像中的候选场景文本;
目标文本特征确定模块,用于将所述目标文本输入所述文本编码网络,得到目标文本特征;
候选全局特征确定模块,用于将所述候选图像和初始的候选融合特征输入所述视觉编码子网络,以及将初始的候选融合特征和所述候选场景文本输入所述场景编码子网络,得到所述候选图像的全局图像特征;
目标图像确定模块,用于根据所述目标文本特征和所述候选图像的全局图像特征,从所述候选图像中确定目标图像。
23.根据权利要求22所述的装置,其中,所述目标文本特征确定模块具体用于:
对所述目标文本进行词嵌入,得到目标文本词向量;
根据所述目标文本的模态信息、目标文本的位置编码信息和所述目标文本词向量,确定所述目标文本的词编码结果;
根据初始的目标文本特征和所述目标文本的词编码结果,构建目标文本的编码序列;
将所述目标文本的编码序列输入所述文本编码网络,得到经处理的目标文本特征。
24.根据权利要求22所述的装置,其中,所述候选全局特征确定模块具体用于:
对所述候选图像进行分块,得到候选图像块序列;
对所述候选图像块序列中的候选图像块进行线性投影,得到所述候选图像块的编码结果;
根据候选图像块的模态信息和候选图像块的位置编码信息对所述候选图像块的编码结果进行处理,确定所述候选图像块经处理的编码结果;
根据初始的全局图像特征和所述候选图像块经处理的编码结果,构建候选图像的编码序列;
将所述候选图像的编码序列输入所述视觉编码子网络中的输入层。
25.根据权利要求22所述的装置,其中,所述候选全局特征确定模块还具体用于:
对所述候选场景文本进行词嵌入,得到候选场景文本向量;
根据所述候选场景文本的图位置编码信息、候选场景文本的模态信息、候选场景文本的字位置编码信息和所述候选场景文本向量,确定所述候选场景文本的编码结果;
根据初始的候选场景文本特征和所述候选场景文本的编码结果,构建候选场景文本的编码序列;
将所述候选场景文本的编码序列输入所述场景编码子网络中的输入层。
26.根据权利要求22所述的装置,其中,所述视觉场景文本融合模型基于权利要求21所述的一种视觉场景文本融合模型的训练装置得到。
27.一种电子设备,包括:
至少一个处理器;以及
与所述至少一个处理器通信连接的存储器;其中,
所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行权利要求1-13中任一项所述的方法。
28.一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使计算机执行根据权利要求1-13中任一项所述的方法。
29.一种计算机程序产品,包括计算机程序,所述计算机程序在被处理器执行时实现根据权利要求1-13中任一项所述的方法。
Priority Applications (2)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210590151.1A CN114942984B (zh) | 2022-05-26 | 2022-05-26 | 视觉场景文本融合模型的预训练和图文检索方法及装置 |
US18/192,393 US20230386168A1 (en) | 2022-05-26 | 2023-03-29 | Pre-training method, image and text retrieval method for a vision and scene text aggregation model, electronic device, and storage medium |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210590151.1A CN114942984B (zh) | 2022-05-26 | 2022-05-26 | 视觉场景文本融合模型的预训练和图文检索方法及装置 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN114942984A true CN114942984A (zh) | 2022-08-26 |
CN114942984B CN114942984B (zh) | 2023-11-21 |
Family
ID=82908258
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210590151.1A Active CN114942984B (zh) | 2022-05-26 | 2022-05-26 | 视觉场景文本融合模型的预训练和图文检索方法及装置 |
Country Status (2)
Country | Link |
---|---|
US (1) | US20230386168A1 (zh) |
CN (1) | CN114942984B (zh) |
Cited By (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN115982403A (zh) * | 2023-01-12 | 2023-04-18 | 之江实验室 | 一种多模态哈希检索方法及装置 |
CN116030048A (zh) * | 2023-03-27 | 2023-04-28 | 山东鹰眼机械科技有限公司 | 灯检机及其方法 |
CN116311320A (zh) * | 2023-05-22 | 2023-06-23 | 建信金融科技有限责任公司 | 文本图像融合层的训练方法、文本图像识别方法及装置 |
CN117077814A (zh) * | 2023-09-25 | 2023-11-17 | 北京百度网讯科技有限公司 | 图片检索模型的训练方法、图片检索方法及装置 |
CN117573839A (zh) * | 2024-01-12 | 2024-02-20 | 阿里云计算有限公司 | 文档检索方法、人机交互方法、电子设备及存储介质 |
WO2024108472A1 (zh) * | 2022-11-24 | 2024-05-30 | 北京京东方技术开发有限公司 | 模型训练方法及装置、文本图像处理方法、设备、介质 |
Families Citing this family (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN117688936B (zh) * | 2024-02-04 | 2024-04-19 | 江西农业大学 | 一种图文融合的低秩多模态融合情感分析方法 |
Citations (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109299216A (zh) * | 2018-10-29 | 2019-02-01 | 山东师范大学 | 一种融合监督信息的跨模态哈希检索方法和*** |
US20200142994A1 (en) * | 2018-11-07 | 2020-05-07 | Adobe Inc. | Guided content discovery in visual search |
CN112784092A (zh) * | 2021-01-28 | 2021-05-11 | 电子科技大学 | 一种混合融合模型的跨模态图像文本检索方法 |
CN113360700A (zh) * | 2021-06-30 | 2021-09-07 | 北京百度网讯科技有限公司 | 图文检索模型的训练和图文检索方法、装置、设备和介质 |
CN114283430A (zh) * | 2021-12-03 | 2022-04-05 | 苏州大创科技有限公司 | 跨模态图文匹配训练方法及装置、存储介质、电子设备 |
WO2022068195A1 (zh) * | 2020-09-30 | 2022-04-07 | 三维通信股份有限公司 | 跨模态的数据处理方法、装置、存储介质以及电子装置 |
CN114445831A (zh) * | 2022-01-14 | 2022-05-06 | 北京百度网讯科技有限公司 | 一种图文预训练方法、装置、设备以及存储介质 |
-
2022
- 2022-05-26 CN CN202210590151.1A patent/CN114942984B/zh active Active
-
2023
- 2023-03-29 US US18/192,393 patent/US20230386168A1/en active Pending
Patent Citations (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109299216A (zh) * | 2018-10-29 | 2019-02-01 | 山东师范大学 | 一种融合监督信息的跨模态哈希检索方法和*** |
US20200142994A1 (en) * | 2018-11-07 | 2020-05-07 | Adobe Inc. | Guided content discovery in visual search |
WO2022068195A1 (zh) * | 2020-09-30 | 2022-04-07 | 三维通信股份有限公司 | 跨模态的数据处理方法、装置、存储介质以及电子装置 |
CN112784092A (zh) * | 2021-01-28 | 2021-05-11 | 电子科技大学 | 一种混合融合模型的跨模态图像文本检索方法 |
CN113360700A (zh) * | 2021-06-30 | 2021-09-07 | 北京百度网讯科技有限公司 | 图文检索模型的训练和图文检索方法、装置、设备和介质 |
CN114283430A (zh) * | 2021-12-03 | 2022-04-05 | 苏州大创科技有限公司 | 跨模态图文匹配训练方法及装置、存储介质、电子设备 |
CN114445831A (zh) * | 2022-01-14 | 2022-05-06 | 北京百度网讯科技有限公司 | 一种图文预训练方法、装置、设备以及存储介质 |
Cited By (9)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2024108472A1 (zh) * | 2022-11-24 | 2024-05-30 | 北京京东方技术开发有限公司 | 模型训练方法及装置、文本图像处理方法、设备、介质 |
CN115982403A (zh) * | 2023-01-12 | 2023-04-18 | 之江实验室 | 一种多模态哈希检索方法及装置 |
CN115982403B (zh) * | 2023-01-12 | 2024-02-02 | 之江实验室 | 一种多模态哈希检索方法及装置 |
CN116030048A (zh) * | 2023-03-27 | 2023-04-28 | 山东鹰眼机械科技有限公司 | 灯检机及其方法 |
CN116311320A (zh) * | 2023-05-22 | 2023-06-23 | 建信金融科技有限责任公司 | 文本图像融合层的训练方法、文本图像识别方法及装置 |
CN116311320B (zh) * | 2023-05-22 | 2023-08-22 | 建信金融科技有限责任公司 | 文本图像融合层的训练方法、文本图像识别方法及装置 |
CN117077814A (zh) * | 2023-09-25 | 2023-11-17 | 北京百度网讯科技有限公司 | 图片检索模型的训练方法、图片检索方法及装置 |
CN117573839A (zh) * | 2024-01-12 | 2024-02-20 | 阿里云计算有限公司 | 文档检索方法、人机交互方法、电子设备及存储介质 |
CN117573839B (zh) * | 2024-01-12 | 2024-04-19 | 阿里云计算有限公司 | 文档检索方法、人机交互方法、电子设备及存储介质 |
Also Published As
Publication number | Publication date |
---|---|
US20230386168A1 (en) | 2023-11-30 |
CN114942984B (zh) | 2023-11-21 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN114942984B (zh) | 视觉场景文本融合模型的预训练和图文检索方法及装置 | |
US20220335711A1 (en) | Method for generating pre-trained model, electronic device and storage medium | |
CN114399769B (zh) | 文本识别模型的训练方法、文本识别方法及装置 | |
CN114155543A (zh) | 神经网络训练方法、文档图像理解方法、装置和设备 | |
CN113792854A (zh) | 一种模型训练及字库建立方法、装置、设备及存储介质 | |
CN110781413B (zh) | 兴趣点确定方法及装置、存储介质、电子设备 | |
CN113033622A (zh) | 跨模态检索模型的训练方法、装置、设备和存储介质 | |
CN113392253B (zh) | 视觉问答模型训练及视觉问答方法、装置、设备及介质 | |
CN115376211B (zh) | 唇形驱动方法、唇形驱动模型的训练方法、装置及设备 | |
CN116152833B (zh) | 基于图像的表格还原模型的训练方法及表格还原方法 | |
CN113204611A (zh) | 建立阅读理解模型的方法、阅读理解方法及对应装置 | |
CN113361363A (zh) | 人脸图像识别模型的训练方法、装置、设备及存储介质 | |
CN111783457A (zh) | 一种基于多模态图卷积网络的语义视觉定位方法及装置 | |
CN113947147A (zh) | 目标地图模型的训练方法、定位方法及相关装置 | |
CN115640520B (zh) | 跨语言跨模态模型的预训练方法、设备和存储介质 | |
EP4191544A1 (en) | Method and apparatus for recognizing token, electronic device and storage medium | |
CN113407850A (zh) | 一种虚拟形象的确定和获取方法、装置以及电子设备 | |
CN115862040A (zh) | 文本纠错方法、装置、计算机设备及可读存储介质 | |
CN113052090A (zh) | 用于生成字幕器以及输出字幕的方法和装置 | |
CN115761839A (zh) | 人脸活体检测模型的训练方法、人脸活体检测方法及装置 | |
CN114863182A (zh) | 图像分类方法、图像分类模型的训练方法及装置 | |
CN112507705B (zh) | 一种位置编码的生成方法、装置及电子设备 | |
CN113360683A (zh) | 训练跨模态检索模型的方法以及跨模态检索方法和装置 | |
CN115577106B (zh) | 基于人工智能的文本分类方法、装置、设备和介质 | |
CN116308738B (zh) | 一种模型训练的方法、业务风控的方法及装置 |
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 |