CN110147806A - 图像描述模型的训练方法、装置及存储介质 - Google Patents
图像描述模型的训练方法、装置及存储介质 Download PDFInfo
- Publication number
- CN110147806A CN110147806A CN201811167476.9A CN201811167476A CN110147806A CN 110147806 A CN110147806 A CN 110147806A CN 201811167476 A CN201811167476 A CN 201811167476A CN 110147806 A CN110147806 A CN 110147806A
- Authority
- CN
- China
- Prior art keywords
- sentence
- neural network
- image
- decoding
- recurrence
- 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
Classifications
-
- 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
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/088—Non-supervised learning, e.g. competitive learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- 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/044—Recurrent networks, e.g. Hopfield 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/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/04—Architecture, e.g. interconnection topology
- G06N3/047—Probabilistic or stochastic networks
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Physics & Mathematics (AREA)
- General Engineering & Computer Science (AREA)
- Computing Systems (AREA)
- Software Systems (AREA)
- Molecular Biology (AREA)
- Computational Linguistics (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- Mathematical Physics (AREA)
- General Health & Medical Sciences (AREA)
- Health & Medical Sciences (AREA)
- Probability & Statistics with Applications (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Evolutionary Biology (AREA)
- Image Analysis (AREA)
Abstract
本申请实施例公开了一种图像描述模型的训练方法,所述图像描述模型包括卷积编码神经网络和递归解码神经网络;所述方法包括:通过所述卷积编码神经网络,获取图像样本的图像特征向量;通过所述递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句;确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整;确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
Description
技术领域
本申请涉及人工智能技术领域,特别涉及一种图像描述模型的训练方法、装置及存储介质。
背景技术
图像描述(Image Caption),是指根据图像自动生成一段描述性文字,即看图说话。为了生成图像对应的描述性文字,首先需要检测出图像中的物体,理解物体之间的相互关系,然后再用合理的语言表达出来。
图像描述技术,可以用于图像检索服务,帮助视觉障碍者理解图像,也可以用于图像场景分类,以及用户相册中的图像自动总结归类。图像描述技术还可以用于婴幼儿的教学,帮助婴幼儿学习说话和辨认图像中的物体和行为。
在一些技术中,可以采用人工标注的图像-句子对来训练图像描述模型。另外,也可以使用半监督学习的技术,在模型的训练过程中使用没有对应关系的图像和句子。没有对应关系的句子数据可以用来训练一个语言模型,一个单独的图像集也可以用来训练一个物体识别模型。也可以使用域适应的方法,把一个数据域上成对的图像和句子数据,迁移到另外一个数据域上。在目标域上,只使用没有对应关系的图像和句子。
目前,半监督以及域适应的方法,都是在监督学习的方法上,增加没有对应关系的图像和句子,来达到提升结果的目的。这些方法仍然需要成对的图像和句子数据来参与模型的训练。给图像标注对应的句子描述是一个非常费时费力的过程。
技术内容
本申请一些实施例提供了一种图像描述模型的训练方法、装置及储存介质,以避免对成对的图像样本和语句样本的依赖,提高图像描述的准确性。
本申请实施例提供了一种图像描述模型的训练方法,所述图像描述模型包括卷积编码神经网络和递归解码神经网络;所述方法包括:
通过所述卷积编码神经网络,获取图像样本的图像特征向量;
通过所述递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句;
确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整;
确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
本申请实施例提供了一种图像描述模型的训练装置,所述图像描述模型包括卷积编码神经网络和递归解码神经网络;所述装置包括:
编码模块,用于通过所述卷积编码神经网络,获取图像样本的图像特征向量;
解码模块,通过所述递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句;
调整模块,用于确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整;以及确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
本申请实施例还提供了一种非易失性计算机可读存储介质,其中所述存储介质中存储有机器可读指令,所述机器可读指令可以由处理器执行以完成上述方法。
在本申请实施例提供的技术方案中,根据递归解码神经网络解码得到的语句的通顺度、以及所述解码得到的语句与图像样本之间的匹配度,对所述递归解码神经网络进行训练调整。这样,在图像描述模型的训练过程中,不需要成对的图像样本和语句样本作为训练集合,从而解除对成对的图像样本和语句样本的依赖,扩大了训练集合的范围,提高了图像描述模型的准确性。
附图说明
为了更清楚地说明本申请实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本申请的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。其中:
图1为本申请一些实施例中的操作环境示意图;
图2A和2B为本申请实施例中模型训练装置116的结构示意图。
图3为本申请一些实施例提供的图像描述模型的训练方法的流程图;
图4A和4B为本申请一些实施例中的图像描述模型的训练方法的另一流程图;
图5为本申请一些实施例中递归解码神经网络的结构示意图;
图6为本申请一些实施例中递归判别神经网络的结构示意图;
图7为本申请一些实施例中图像描述模型训练装置的结构示意图;及
图8为本申请一些实施例中图像描述模型训练装置的另一结构示意图。
具体实施方式
为使本申请的技术方案及优点更加清楚明白,以下参照附图并举实施例,对本申请进一步详细说明。
为了描述上的简洁和直观,下文通过描述若干代表性的实施例来对本申请的方案进行阐述。但本文并未示出所有实施方式。实施例中大量的细节仅用于帮助理解本申请的方案,本申请的技术方案实现时可以不局限于这些细节。为了避免不必要地模糊了本申请的方案,一些实施方式没有进行细致地描述,而是仅给出了框架。下文中,“包括”是指“包括但不限于”,“根据……”是指“至少根据……,但不限于仅根据……”。说明书和权利要求书中的“包括”是指某种程度上至少包括,应当解释为除了包括之后提到的特征外,其它特征也可以存在。
目前,由于深度学习在视觉领域的成功应用,研究者也将其引入到图像描述领域中,采用神经机器翻译的方法生成描述句子。
基于深度学习的图像描述模型可以采用“编码-解码”的流程,首先使用卷积神经网络(CNN,Convolutional Neural Network)提取图像特征向量,将整幅图像编码为一条维度固定的特征向量;然后使用递归神经网络(RNN,recurrent neural network)进行解码,按时间顺序逐个生成相关单词。
CNN,是一种直接从图像底层的像素特征开始,逐层对图像进行特征提取的前馈神经网络,是编码器常用的实现模型,负责将图像编码成向量。
RNN,是一种具有固定权值、外部输入和内部状态的神经网络,可以将其看作是以权值和外部输入为参数,关于内部状态的行为动力学。RNN是解码器常用的实现模型,负责将编码器生成的图像向量翻译成图像的文字描述。
在图像描述模型的训练过程中,需要使用成对的图像样本和句子样本作为训练集合。训练集合不仅需要足够大,还需要尽量多样化。但是,给图像标注对应的句子描述是一个非常费时费力的过程。而且,如果训练集合的大小减小,图像描述的准确性也将降低。
为此,本申请实施例提供了一种图像描述模型的训练方法,可以避免对成对的图像样本和语句样本的依赖,扩大训练集合的范围,从而提高图像描述模型的准确性。
本申请实施例提供的图像描述模型训练方法可以由任何具有数据处理能力的计算机设备来执行,例如,终端设备或者服务器等等。当根据本申请实施例提供的方法完成图像描述模型的训练之后,可以将训练好的图像描述模型应用在所述服务器或者终端设备,用于为指定的图像生成对应的描述语句,例如,可以为用户提供图像检索服务、为用户相册中的图像自动分类,等等。
图1为本申请一些实施例中的操作环境100的示意图。如图1所示,本申请实施例的图像描述模型训练方法可以由模型训练装置116来执行。
在本申请一些实施例中,所述模型训练装置116用于对图像描述模型进行训练,得到训练好的图像描述模型,并将所述训练好的图像描述模型提供给服务器112,以便服务器112为终端设备104提供图像描述生成服务,例如为用户提供图像检索服务等等;或者,将训练好的图像描述模型提供给终端设备104,以便终端设备104为用户提供图像描述生成服务,例如为用户相册中的图像自动分类等等。
在一些实施例中,所述模型训练装置116可以在一个或多个独立的数据处理装置或分布式计算机网络上实现,也可以集成在所述服务器112或者终端设备104中。
在一些实施例中,当所述服务器112为终端设备104提供图像描述生成服务时,如图1所示,多个用户通过各自的终端设备(例如终端设备104-a至104-c)上执行的应用108-a至108-c,通过网络106连接至服务器112。服务器112向终端设备104提供图像描述生成服务,例如接收用户提供的图像,根据服务器112中的图像描述生成模型为该图像生成对应的描述语句,并将生成的描述语句返回给终端设备104等等。
在一些实施例中,所述终端设备104是指具有数据计算处理功能的终端设备,包括但不限于(安装有通信模块的)智能手机、掌上电脑、平板电脑、智能电视(Smart TV)等。这些通信终端上都安装有操作系统,包括但不限于:Android操作系统、Symbian操作系统、Windows mobile操作系统、以及苹果iPhone OS操作系统等等。
在一些实施例中,网络106可以包括局域网(LAN)和广域网(WAN)诸如互联网。可以使用任意公知的网络协议来实现网络106,包括各种有线或无线协议。
在一些实施例中,所述服务器112可以在一个或多个独立的数据处理装置或分布式计算机网络上实现。
下面结合图2A和2B对本申请实施例提供的模型训练装置116进行说明。
图2A和2B为本申请一些实施例中模型训练装置116的结构示意图。如图2A所示,所述模型训练装置116包括:卷积编码神经网络201和递归解码神经网络202。
其中,卷积编码神经网络201用于为输入的图像样本211提取图像特征向量214。
在一些实施例中,可以使用CNN作为卷积编码神经网络201,来提取图像样本211的图像特征向量214。例如,可以使用Inception-X系列网络、ResNet系列网络等。
在一些实施例中,所述卷积编码神经网络201可以是预先训练好的编码网络。
递归解码神经网络202用于将所述图像特征向量214解码成一个用于描述所述图像样本的语句213。在一些实施例中,所述递归解码神经网络202可以通过长短时记忆网络(LSTM,Long Short Term Memory Network)实现,或者利用递归门单元(GRU,GateRecurrent Unit)等其他方式实现。
LSTM,是一种时间递归神经网络,用于处理和预测时间序列中间隔或者延迟相对较长时间的重要事件,属于一种特殊的RNN。
由于本申请实施例中使用的图像样本没有对应的语句样本,所以在对递归解码神经网络202进行训练时,通过以下方式实现对递归解码神经网络202的训练:
根据递归解码神经网络202解码得到的语句213与所述图像样本211之间的匹配度,对所述递归解码神经网络202进行训练调整;以及
根据所述解码得到的语句213的通顺度,对所述递归解码神经网络202进行训练调整。
参见图2B,在一些实施例中,为了实现对递归解码神经网络202的训练,所述模型训练装置116还进一步包括:递归判别神经网络203和调整模块204。
其中,递归判别神经网络203用于当递归解码神经网络202输出一个语句之后,判别该语句的通顺度,并将判别结果提供给调整模块204,由调整模块204根据递归判别神经网络203的判别结果,对所述递归解码神经网络202进行调整。
调整模块204用于对递归解码神经网络202进行调整,包括根据解码得到的语句213与所述图像样本211之间的匹配度,对所述递归解码神经网络202进行调整,以及根据所述解码得到的语句213的通顺度,对所述递归解码神经网络202进行调整。此外,调整模块204也对递归判别神经网络203进行训练调整,使得递归判别神经网络203能够更准确的识别出输入的语句的通顺度。
在一些实施例中,调整模块204可以包括:判别奖励单元2041、物体奖励单元2042、以及判别损失单元2043。其中,判别奖励单元2041和物体奖励单元2042用于对递归解码神经网络202进行调整,而判别损失单元2043用于对递归判别神经网络203进行调整。
判别奖励单元2041,用于根据递归判别神经网络203确定的所述解码得到的语句的通顺度,对所述递归解码神经网络202进行调整。
在一些实施例中,所述通顺度可以是对递归解码神经网络解码得到的语句的打分。
物体奖励单元2042,用于根据递归解码神经网络202解码得到的语句中各个单词与图像样本211之间的匹配度,对递归解码神经网络202进行调整。
在一些实施例中,针对图像样本211,可以预先根据物体检测模型检测出该图像包含的物体,以及每个物体对应的打分。然后物体奖励单元2042将解码得到的语句中包含的单词与物体检测模型检测出的物体进行比较,从而得到解码得到的单词与图像样本211之间的匹配度,然后根据该关联度对递归解码神经网络202进行调整优化。
判别损失单元2043,用于根据递归判别神经网络203对语句样本212和递归解码神经网络202解码得到的语句213的判别结果,确定判别损失,并根据所述判别损失对递归判别神经网络203进行调整。
在本申请一些实施例中,为了使得递归判别神经网络203能够尽量准确的进行判别操作,还需要利用人工编写的语句样本212来训练递归判别神经网络203,使得当一个人工编写的非常通顺的语句样本212输入到递归判别神经网络时,递归判别神经网络203输出1,当输入一个递归解码神经网络202生成的语句时,递归判别神经网络输出0至1之间的任意数值。递归判别神经网络输出的数值越高,表示语句的通顺度越高,即语句越通顺。
在一些实施例中,可以轮流将语句样本212和递归解码神经网络202解码得到的语句213输入到递归判别神经网络203,根据递归判别神经网络203对语句样本212和语句213的打分,得到判别损失,并根据得到的判别损失来调整所述递归判别神经网络。
在一些实施例中,除了上述的判别奖励单元2041、物体奖励单元2042和判别损失单元2043之外,为了进一步提高所述图像描述模型的准确度,所述调整模块204还可以进一步包括:图像重构奖励单元2044、语句重构损失单元2045、及图像重构损失单元2046。
所述图像重构奖励单元2044,用于根据解码得到的语句重构图像,并确定重构的图像与图像样本211之间的相似度,根据所述相似度对所述递归解码神经网络202进行优化调整。
例如,可以通过递归判别神经网络203,得到与所述解码得到的语句对应的语句特征向量,将所述语句特征向量映射到图像特征的空间,得到对应的图像特征向量,并与卷积编码神经网络201得到的图像特征向量214进行比较,得到所述相似度。
在一些实施例中,图像重构奖励单元2044可以通过全连接层将所述语句特征向量映射到图像特征的空间,从而得到重构之后的图像特征向量。
所述语句重构损失单元2045,用于获取语句样本212对应的语句特征向量,通过全连接层将所述语句样本212对应的语句特征向量映射到图像特征的空间,从而得到对应的图像特征向量,将所述图像特征向量输入到递归解码神经网络202,从而重构出一个语句,将所述重构得到的语句与语句样本212进行比较,根据比较的结果对所述递归解码神经网络202进行优化调整。
所述图像重构损失单元2046,用于根据解码得到的语句重构图像,确定所述重构的图像与所述图像样本之间的差异度,并根据所述差异度,对所述递归判别神经网络进行优化调整。
在一些实施例中,所述图像重构损失单元2046可以通过所述递归判别神经网络203,得到与所述解码得到的语句对应的语句特征向量;将所述语句特征向量映射到图像特征空间,得到对应的图像特征向量,将所述通过映射得到的图像特征向量与卷积编码神经网络201得到的图像特征向量214进行比较得到所述重构的图像与所述图像样本之间的差异度,即图像重构损失,根据所述图像重构损失对所述递归判别神经网络203进行优化调整。
通过以上各个单元,调整模块204可以对递归解码神经网络202和递归判别神经网络203进行优化调整,从而提高递归解码神经网络202和递归判别神经网络203的准确度。
下面结合图3对本申请实施例提供的图像描述模型的训练方法进行说明。图3为本申请实施例提供的一种图像描述模型的训练方法的流程图。所述图像描述模型包括卷积编码神经网络和递归解码神经网络。如图3所示,所述方法包括以下步骤:
S301,通过所述卷积编码神经网络,获取图像样本的图像特征向量。
在一些实施例中,可以采用预先训练好的卷积编码神经网络来获取图像样本对应的图像特征向量。另外,在获取所述图像样本的图像特征向量之后,可以进一步将所述图像特征向量进行降维处理,得到降维后的图像特征向量。
S302,通过所述递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句。
在一些实施例中,可以将所述降维后的图像特征向量输入到所述递归解码神经网络,所述递归解码神经网络对所述降维后的图像特征向量进行解码,得到所述用于描述所述图像样本的语句。
在一些实施例中,所述语句中包含多个单词;所述通过递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句包括:
将所述图像特征向量输入到所述递归解码神经网络,得到输出的n个概率分布,其中,n表示所述解码得到的语句的长度;
对于每个概率分布,分别在单词表中选择所述概率分布中最大概率值对应的单词,组成用于描述所述图像样本的语句。
S303,确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整。
在一些实施例中,可以根据物体检测模型对所述图像样本的检测结果,确定所述图像样本中包含的各物体以及所述各物体对应的权重;
将解码得到的语句中包含的各个单词,与所述图像样本中包含的所述各物体进行匹配操作,并根据所述匹配结果以及所述各物体对应的权重,确定所述解码得到的语句与所述图像样本之间的匹配度,然后根据所述匹配度对所述递归解码神经网络进行调整。
S304,确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
在一些实施例中,可以通过各种方法来确定所述解码得到的语句的通顺度。例如,可以将所述解码得到的语句输入递归判别神经网络,根据所述递归判别神经网络各个时刻的第一输出,确定所述解码得到的语句的通顺度。
例如,递归解码神经网络解码出来的语句越通顺,则递归判别神经网络越会认为该语句是人工编写的通顺语句,即通顺度越高。
在一些实施例中,为了提高递归判别神经网络的判别准确度,需要根据语句样本对所述递归判别神经网络进行训练。
例如,可以将语句样本输入到所述递归判别神经网络,获取所述递归判别神经网络各个时刻的第二输出;
根据所述递归判别神经网络各个时刻的第一输出和所述第二输出,对所述递归判别神经网络进行调整。
在本申请实施例提供的技术方案中,根据递归解码神经网络解码得到的语句的通顺度、以及所述解码得到的语句与图像样本之间的匹配度,对所述递归解码神经网络进行训练调整。这样,在图像描述模型的训练过程中,不需要成对的图像样本和语句样本作为训练集合,从而解除对成对的图像样本和语句样本的依赖,扩大了训练集合的范围,提高了图像描述模型的准确性。
图4为本申请实施例中另一图像描述模型的训练方法的流程图。下面结合图4对本申请实施例提供的图像描述模型的训练方法进行说明。如图4所示,所述方法包括以下步骤:
S401,通过卷积编码神经网络,获取图像样本的图像特征向量。
在本申请一些实施例中,可以通过图2所示的卷积编码神经网络201来获取图像样本的图像特征向量。
下面,以所述卷积编码神经网络为Inception-V4卷积神经网络为例进行说明。
在一些实施例中,可以取Inception-V4卷积神经网络的池化层(Average Pooling层)输出的向量作为所述图像特征向量。卷积编码神经网络的操作可以由以下公式(1)表示:
fim=CNN(I) (1)
其中,I代表输入所述卷积编码神经网络的图像样本,CNN代表所述卷积编码神经网络,fim是得到的图像特征向量,fim的维数可以是1536。
S402,对所述图像样本的图像特征向量进行降维处理,得到降维后的图像特征向量。
在一些实施例中,在得到图像样本对应的图像特征向量fim之后,考虑到后续操作的计算量,可以对所述图像特征向量fim进行降维处理。
例如,在卷积编码神经网络之后,可以通过一个全连接层实现所述降维处理,如下公式(2)表示:
x-1=FC(fim) (2)
其中,x-1表示降维之后的图像特征向量,FC()表示通过全连接层进行的降维操作。
S403,将降维后的图像特征向量输入递归解码神经网络,所述递归解码神经网络对所述降维后的图像特征向量进行解码,得到用于描述所述图像样本的语句。
图5是本申请一些实施例中递归解码神经网络的结构示意图。在图5所示的实施例中,以所述递归解码神经网络使用LSTM单元实现为例。参见图4和图5,所述步骤S403包括:
S4031,将所述图像特征向量输入到所述递归解码神经网络,得到输出的n个概率分布,其中,n表示所述解码得到的语句的长度。
例如,如图5所示,在t时刻,将xt和输入到所述LSTM单元,得到所述LSTM单元的输出pt+1和
在一些实施例中,所述LSTM单元的上述操作可以如下公式(3)表示:
其中,xt代表第t个单词对应的词向量,和分别表示t时刻和t+1时刻的LSTM单元的隐含状态;pt+1代表概率分布,即单词表中各个单词对应的概率;n代表语句的长度。
另外,由于本申请实施例中在训练递归解码神经网络的时候,并不知道图像样本对应的语句,所以递归解码神经网络各个时刻的输入需要根据之前时刻输出的单词得到,其中,
当t=-1时,xt=x-1,即xt为所述降维之后的图像特征向量;
当t∈(0,...,n-1}时,xt为根据t时刻LSTM单元输出的单词St对应的词向量,如公式(4)所示:
xt=WeSt,t∈{0,...,n-1} (4)
其中,We代表词向量(word embedding)处理。
所述词向量(Word embedding),是一个用于将来自单词表的单词或短语映射到实数的向量。
S4032,对于每个概率分布,分别在单词表中选择所述概率分布中最大概率值对应的单词,组成用于描述所述图像样本的语句。
例如,根据步骤S4031中LSTM单元输出的所述概率分布pt+1,在单词表中选择概率分布pt+1中最大概率值对应的单词作为t+1时刻的输出。
通过上述的操作,递归解码神经网络可以得到长度为n的语句:S1S2...Sn。
S404,确定所述解码得到的语句与所述图像样本之间的匹配度。
在一些实施例中,所述语句中包含多个单词;每次递归解码神经网络解码出一个单词,可以判断该单词与图像样本之间的匹配度,即判断图像样本中是否含有该单词对应的物体。
在一些实施例中,步骤S404可以包括以下操作:
S4041,根据物体检测模型对所述图像样本的检测结果,确定所述图像样本中包含的各物体以及所述各物体对应的权重。
在一些实施例中,可以预先利用物体检测模型对图像样本进行检测,得到图像中包含的物体以及这个物体对应的打分(权重)。
S4042,对解码得到的语句中的各个单词,与所述图像样本中包含的所述各物体进行匹配操作,并根据所述匹配结果以及所述各物体对应的权重,确定所述匹配度。
在一些实施例中,可以根据以下公式(5)来进行匹配操作并确定递归解码神经网络得到的单词与图像样本之间的关联度:
其中,st表示递归解码神经网络解码得到的单词,ci表示物体检测模型从图像样本中检测出的一个物体,vi是这个物体对应的权重,Nc代表物体检测模型从图像样本中检测出的物体的数量;代表单词st与图像样本之间的匹配度;I()代表指示函数。
在得到关联度之后,可以将该关联度作为物体识别奖励,反馈给递归解码神经网络,从而实现对递归解码神经网络的优化。
S405,根据所述匹配度对所述递归解码神经网络进行调整。
在一些实施例中,可以将所述匹配度反馈到所述递归解码神经网络,从而对所述递归解码神经网络中的权重参数和词向量We进行修正。
这样,通过不断的训练,可以不断地对递归解码神经网络中的权重参数和词向量We做出修正,从而提高所述递归解码神经网络的精度。
对于采用LSTM单元实现的递归解码神经网络,LSTM单元具有三个门:遗忘门(forget gate),输入门(input gate)和输出门(output gate)。其中遗忘门决定了上一时刻的单元状态有多少保留到当前时刻。输入门决定了当前时刻网络的输入有多少保存到当前时刻。输出门决定了当前单元状态有多少输出到LSTM的当前输出值。
门实际上就是一层全连接层,输入是一个向量,输出是一个0到1之间的实数向量。当门输出为0时,任何向量与之相乘都会得到0向量,这就相当于什么都不能通过;输出为1时,任何向量与之相乘都不会有任何改变,这就相当于什么都可以通过。
因此,可以将步骤S404计算得到的匹配度,反馈给递归解码神经网络,用于调整各个门的权重矩阵,从而对所述递归解码神经网络进行修正调整。
另外,在每一次的训练过程中,词向量We也根据所述匹配度进行参数更新而发生变化。
S406,将所述解码得到的语句输入递归判别神经网络,根据所述递归判别神经网络各个时刻的第一输出,确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
在一些实施例中,可以将所述解码得到的语句输入递归判别神经网络,根据所述递归判别神经网络各个时刻的输出,确定所述解码得到的语句的通顺度。
例如,将所述解码得到的语句输入所述递归判别神经网络,根据所述递归判别神经网络各个时刻的输出,来确定所述解码得到的语句的通顺度。
图6为本申请实施例提供的递归判别神经网络的示意图。在图6所示的实施例中,仍以使用LSTM单元实现所述递归判别神经网络为例进行说明。需要注意的是,所述递归判别神经网络也可以采用其他方式实现,例如采用GRU单元来实现。
参见图6,所述递归判别神经网络中LSTM单元的操作可以如以下公式(6)所示:
其中,LSTM单元在t时刻的输入包括:xt和输出包括:qt和这里,xt表示第t个单词对应的词向量,代表LSTM单元在t-1时刻的隐含状态;qt代表t时刻对语句的通顺度的打分,代表LSTM单元在t时刻的隐含状态。
在一些实施例中,qt可以是0至1之间的数值。当输入一个人工编写的通顺的语句时,递归判别神经网络输出1,当输入一个递归解码神经网络生成的句子的时候,递归判别神经网络输出0至1之间的数值。递归解码神经网络的目标是欺骗递归判别神经网络,尽量生成能够让递归判别神经网络产生1的句子,从而能够生成看起来通顺的语句。
在一些实施例中,可以根据如下公式(7)来确定所述解码得到的语句的通顺度。
其中,其中,radu代表所述通顺度,qt代表所述递归判别神经网络在t时刻输出的数值,n代表所述解码得到的语句的长度。
在一些实施例中,在得到所述通顺度之后,可以将所述通顺度反馈到所述递归解码神经网络,从而对递归解码神经网络的参数以及词向量We进行调整,从而提高所述递归解码神经网络的精度。
这里,与步骤S405类似,所述通顺度可以用于调整所述递归解码神经网络中各个门的权重矩阵,以及所述词向量We,从而对所述递归解码神经网络进行修正调整。
S407,将语句样本输入到所述递归判别神经网络,获取所述递归判别神经网络各个时刻的第二输出;根据所述递归判别神经网络各个时刻的第一输出和所述第二输出,对所述递归判别神经网络进行调整。
在一些实施例中,步骤S407可以包括以下操作:
S4071,根据所述递归判别神经网络各个时刻的第一输出和所述第二输出,确定判别损失。
例如,所述判别损失可以根据以下公式(8)计算:
其中,Ladu表示所述判别损失;qt代表所述递归判别神经网络在t时刻的第一输出,代表所述递归判别神经网络在t时刻的第二输出;l表示所述语句样本的长度;n代表所述解码得到的语句的长度。
S4072,根据所述判别损失对所述递归判别神经网络进行调整。
这里,所述判别损失用于对递归判别神经网络的参数进行调整。
例如,对于采用LSTM单元实现所述递归判别神经网络的情况,可以将所述判别损失反馈到所述递归判别神经网络,从而调整所述递归判别神经网络中LSTM单元的各个门的权重矩阵,从而对所述递归判别神经网络进行调整修正。
通过步骤S404至S407,一方面针对解码得到的语句的通顺度、以及与图像样本的匹配度,对递归解码神经网络的参数进行了修正。另一方面,用于监督递归解码神经网络的学习的递归判别神经网络也得到了修正调整。
为了进一步提高递归解码神经网络以及递归判别神经网络的精度,在一些实施例中,所述方法还可以进一步包括以下步骤S408至S410中的任意一个或多个:
S408,根据解码得到的语句重构图像,确定所述重构图像与所述图像样本之间的相似度,根据所述相似度对所述递归解码神经网络进行优化。
在一些实施例中,所述步骤S408包括以下操作:
S4081,通过所述递归判别神经网络,得到与所述解码得到的语句对应的语句特征向量。
S4082,将所述语句特征向量映射到图像特征空间,得到对应的图像特征向量;
在一些实施例中,可以在所述递归判别神经网络的最后增加一个全连接层,这个全连接层可以把语句特征向量映射到图像特征的空间,从而得到重构图像对应的图像特征向量。
S4083,将所述通过映射得到图像特征向量,与所述图像样本的图像特征向量进行比较,确定所述重构的图像与所述图像样本之间的相似度。
在一些实施例中,可以根据下面的公式(9)来计算所述重构图像与所述图像样本之间的相似度:
其中,rim代表所述重构图像与图像样本之间的相似度;x-1代表所述图像样本的全局特征表示;x′代表所述重构图像的全局特征表示;代表取范数操作。
S4084,根据所述相似度对所述递归解码神经网络进行修正调整。
这里,对所述递归解码神经网络进行的修正调整,与步骤S405类似。可以将计算得到的相似度rim反馈到所述递归解码神经网络,从而对递归解码神经网络中各个门的权重矩阵以及词向量进行调整,以便提高所述递归解码神经网络的精度。
S409,根据所述解码得到的语句重构图像,确定所述重构图像与所述图像样本之间的差异度,根据所述差异度对所述递归判别神经网络进行修正调整。
在一些实施例中,所述重构图像与图像样本之间的差异度可以根据以下公式(10)确定:
其中,Lim代表所述重构图像与图像样本之间的差异度;x-1代表所述图像样本的全局特征表示;x′代表所述重构图像的全局特征表示;氏表取范数操作。
这里,在计算得到所述差异度之后,可以将所述差异度反馈到所述递归判别神经网络,从而对递归判别神经网络中各个门的权重矩阵进行调整,从而提高所述递归判别神经网络的精度。
S410,通过所述递归解码神经网络,获取与所述语句样本对应的重构语句,将所述语句样本与所述重构语句进行比较,根据比较结果对所述递归解码神经网络进行修正调整。
在一些实施例中,所述步骤S410包括以下操作:
S4101,获取语句样本对应的语句特征向量,将所述语句特征向量映射到图像特征的空间,得到对应的图像特征向量。
S4102,通过所述递归判别神经网络,得到与所述映射得到的图像特征向量对应的重构语句。
S4103,将所述语句样本与所述重构语句进行比较,根据比较结果对所述递归解码神经网络进行修正调整。
在一些实施例中,可以通过下面的公式(11)来实现对所述语句样本和所述重构语句的比较:
其中,Lsen代表语句重构损失,st代表解码得到的语句中的第t个单词,代表语句样本中的第t个单词,代表条件概率。
在得到重构损失Lsen之后,可以将该重构损失反馈给递归解码神经网络,从而调整递归解码神经网络中各个门的权重矩阵以及词向量。
在本步骤中,重构语句的操作与降噪自动编码器类似。为了使得得到的图像描述模型更加鲁棒,可以给输入的语句增加两种噪音,其中包括随机的去掉一些单词,随机的打乱部分单词的顺序,降噪自动编码器的损失可以如上公式(11)所示。
通过本申请实施例提供的技术方案,根据递归解码神经网络解码得到的语句的通顺度、以及所述解码得到的语句与图像样本之间的匹配度,对所述递归解码神经网络进行训练调整。这样,在图像描述模型的训练过程中,不需要成对的图像样本和语句样本作为训练集合,从而解除对成对的图像样本和语句样本的依赖,扩大了训练集合的范围,提高了图像描述模型的准确性。
进一步的,又通过图像重构与语句重构操作,对所述递归解码神经网络和所述递归判别神经网络做了进一步的修正调整,从而进一步提高了图像描述模型的准确性。
在完成对图像描述模型的训练之后,用训练好的图像描述模型,使用束搜索(beamsearch)方法得到对输入图像的语句描述。
束搜索是一种启发式图搜索算法,通常用在图的解空间比较大的情况下,为了减少搜索所占用的空间和时间,在每一步深度扩展的时候,剪掉一些质量比较差的结点,保留下一些质量较高的结点。
以上对本申请实施例提供的图像描述模型训练方法进行了说明。
下面结合附图对本申请实施例提供的图像描述模型训练装置进行说明。
图7为本申请一些实施例提供的图像描述模型训练装置的一种结构示意图。如图7所示,该装置700包括:
编码模块701,用于通过所述卷积编码神经网络,获取图像样本的图像特征向量;
解码模块702,通过所述递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句;
调整模块703,用于确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整;以及确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
在一些实施例中,所述装置700进一步包括:
降维模块704,用于在所述编码模块701获取所述图像样本的图像特征向量之后,将所述图像特征向量进行降维处理,得到降维后的图像特征向量;
所述解码模块702进一步用于,将所述降维后的图像特征向量输入到所述递归解码神经网络,所述递归解码神经网络对所述降维后的图像特征向量进行解码,得到所述用于描述所述图像样本的语句。
在一些实施例中,所述语句中包含多个单词;所述解码模块702进一步用于:
将所述图像特征向量输入到所述递归解码神经网络,得到输出的n个概率分布,其中,n表示所述解码得到的语句的长度;
对于每个概率分布,分别在单词表中选择所述概率分布中最大概率值对应的单词,组成用于描述所述图像样本的语句。
在一些实施例中,所述调整模块703进一步包括:物体奖励单元7031,用于:
根据物体检测模型对所述图像样本的检测结果,确定所述图像样本中包含的各物体以及所述各物体对应的权重;
将解码得到的语句中包含的各个单词,与所述图像样本中包含的所述各物体进行匹配操作,并根据所述匹配结果以及所述各物体对应的权重,确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整。
在一些实施例中,所述调整模块703进一步包括:判别奖励单元7032,用于:
将所述解码得到的语句输入递归判别神经网络,根据所述递归判别神经网络各个时刻的第一输出,确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
在一些实施例中,所述判别奖励单元7032进一步用于:
根据以下公式确定所述通顺度:
其中,radu代表所述通顺度,qt代表所述递归判别神经网络在t时刻的输出,n代表所述解码得到的语句的长度。
在一些实施例中,所述调整模块703进一步包括:判别损失单元7033,用于:
将语句样本输入到所述递归判别神经网络,获取所述递归判别神经网络各个时刻的第二输出;
根据所述递归判别神经网络各个时刻的第一输出和所述第二输出,对所述递归判别神经网络进行调整。
在一些实施例中,所述判别损失单元7033进一步用于:
根据所述递归判别神经网络各个时刻的第一输出和所述第二输出,按照以下公式确定判别损失:
根据所述判别损失对所述递归判别神经网络进行调整;
其中,qt代表所述递归判别神经网络在t时刻的第一输出,代表所述递归判别神经网络在t时刻的第二输出;l表示所述语句样本的长度;Ladu表示所述判别损失;n代表所述解码得到的语句的长度。
在一些实施例中,所述调整模块703进一步包括,图像重构奖励单元7034,用于:
根据所述解码得到的语句重构图像;
确定所述重构的图像与所述图像样本之间的相似度;
根据所述相似度,对所述递归解码神经网络进行调整。
在一些实施例中,所述图像重构奖励单元7034进一步用于:
获取与所述解码得到的语句对应的语句特征向量;
将所述语句特征向量映射到图像特征空间,得到对应的图像特征向量;
将所述通过映射得到图像特征向量,与所述图像样本的图像特征向量进行比较,确定所述重构的图像与所述图像样本之间的相似度。
在一些实施例中,可以根据上述公式(9)计算得到所述重构图像与所述图像样本之间的相似度rim。然后,可以将计算得到的相似度rim反馈到所述递归解码神经网络,从而对递归解码神经网络中各个门的权重矩阵以及词向量进行调整,以便提高所述递归解码神经网络的精度。
在一些实施例中,所述调整模块703进一步包括:图像重构损失单元7035,用于:
根据所述解码得到的语句重构图像;
确定所述重构的图像与所述图像样本之间的差异度;
根据所述差异度,对所述递归判别神经网络进行调整。
在一些实施例中,所述差异度Lim可以利用上述公式(10)计算得到。在计算得到所述差异度Lim之后,可以将所述差异度反馈到所述递归判别神经网络,从而对递归判别神经网络中各个门的权重矩阵进行调整,从而提高所述递归判别神经网络的精度。
在一些实施例中,所述调整模块703进一步包括:语句重构损失单元7036,用于:
通过所述递归解码神经网络,获取与所述语句样本对应的重构语句,将所述语句样本与所述重构语句进行比较,根据比较结果对所述递归解码神经网络进行修正调整。
在一些实施例中,所述语句重构损失单元7036可以获取语句样本对应的语句特征向量,将所述语句特征向量映射到图像特征的空间,得到对应的图像特征向量;获取与所述映射得到的图像特征向量对应的重构语句;将所述语句样本与所述重构语句进行比较,根据比较结果对所述递归解码神经网络进行调整。
在一些实施例中,所述语句重构损失单元7036可以根据上述公式(11)得到语句重构损失Lsen。在得到重构损失Lsen之后,可以将该重构损失反馈给递归解码神经网络,从而调整递归解码神经网络中各个门的权重矩阵以及词向量。
通过本申请实施例提供的技术方案,根据递归解码神经网络解码得到的语句的通顺度、以及所述解码得到的语句与图像样本之间的匹配度,对所述递归解码神经网络进行训练调整。这样,在图像描述模型的训练过程中,不需要成对的图像样本和语句样本作为训练集合,从而解除对成对的图像样本和语句样本的依赖,扩大了训练集合的范围,提高了图像描述模型的准确性。
进一步的,又通过图像重构与语句重构操作,对所述递归解码神经网络和所述递归判别神经网络做了进一步的修正调整,从而进一步提高了图像描述模型的准确性。
在完成对图像描述模型的训练之后,用训练好的图像描述模型,使用束搜索(beamsearch)方法得到对输入图像的语句描述。
图8是本申请一些实施例中图像描述模型训练装置的另一结构示意图。该图像描述模型训练装置800可以是图1示出的模型训练装置116,也可以是集成于模型训练装置116中的一个组件。
如图8所示,图像描述模型训练装置800包括一个或者多个处理器(CPU)802、网络接口804、存储器806、以及用于互联这些组件的通信总线808。
在一些实施例中,所述网络接口804用于实现所述图像描述模型训练装置800与外部设备之间的网络连接。
所述图像描述模型训练装置800还可以进一步包含一个或多个输出设备812(例如一个或多个可视化显示器),和/或包括一个或多个输入设备814(例如键盘,鼠标,或其他输入控件等)。
存储器806可以是高速随机存取存储器,诸如DRAM、SRAM、DDR RAM、或其他随机存取固态存储设备;或者非易失性存储器,诸如一个或多个磁盘存储设备、光盘存储设备、闪存设备,或其他非易失性固态存储设备。
存储器806包括:
操作系统816,包括用于处理各种基本系统服务和用于执行硬件相关任务的程序;
图像描述模型训练系统818,用于通过所述卷积编码神经网络,获取图像样本的图像特征向量;通过所述递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句;确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整;确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
在一些实施例中,所述图像描述模型训练系统818的具体操作和功能可以参见上面的方法实施例,在此不再赘述。
另外,在本申请各个实施例中的各功能模块可以集成在一个处理单元中,也可以是各个模块单独物理存在,也可以两个或两个以上模块集成在一个单元中。上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。所述各实施例的功能模块可以位于一个终端或网络节点,或者也可以分布到多个终端或网络节点上。
另外,本发明的每一个实施例可以通过由数据处理设备如计算机执行的数据处理程序来实现。显然,数据处理程序构成了本发明。此外,通常存储在一个存储介质中的数据处理程序通过直接将程序读取出存储介质或者通过将程序安装或复制到数据处理设备的存储设备(如硬盘和或内存)中执行。因此,这样的存储介质也构成了本发明。存储介质可以使用任何类型的记录方式,例如纸张存储介质(如纸带等)、磁存储介质(如软盘、硬盘、闪存等)、光存储介质(如CD-ROM等)、磁光存储介质(如MO等)等。
因此本发明还提供了一种存储介质,其中存储有数据处理程序,该数据处理程序用于执行本发明上述方法的任何一种实施例。
本领域普通技术人员可以理解上述实施例的各种方法中的全部或部分步骤是可以通过程序来指令相关的硬件来完成,该程序可以存储于一计算机可读存储介质中,存储介质可以包括:只读存储器(ROM,Read Only Memory)、随机存取记忆体(RAM,RandomAccess Memory)、磁盘或光盘等。
以上所述仅为本申请的较佳实施例而已,并非用于限定本申请的保护范围。凡在本申请的精神和原则之内,所作的任何修改、等同替换以及改进等,均应包含在本申请的保护范围之内。
Claims (15)
1.一种图像描述模型的训练方法,其特征在于,所述图像描述模型包括卷积编码神经网络和递归解码神经网络;所述方法包括:
通过所述卷积编码神经网络,获取图像样本的图像特征向量;
通过所述递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句;
确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整;
确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
2.根据权利要求1所述的方法,其特征在于,在获取所述图像样本的图像特征向量之后,进一步包括:
将所述图像特征向量进行降维处理,得到降维后的图像特征向量;
所述通过递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句,包括:
将所述降维后的图像特征向量输入到所述递归解码神经网络,所述递归解码神经网络对所述降维后的图像特征向量进行解码,得到所述用于描述所述图像样本的语句。
3.根据权利要求1所述的方法,其特征在于,所述通过递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句包括:
将所述图像特征向量输入到所述递归解码神经网络,得到输出的n个概率分布,其中,n表示所述解码得到的语句的长度;
对于每个概率分布,分别在单词表中选择所述概率分布中最大概率值对应的单词,组成用于描述所述图像样本的语句。
4.根据权利要求1所述的方法,其特征在于,确定所述解码得到的语句与所述图像样本之间的匹配度包括:
根据物体检测模型对所述图像样本的检测结果,确定所述图像样本中包含的各物体以及所述各物体对应的权重;
将解码得到的语句中包含的各个单词,与所述图像样本中包含的所述各物体进行匹配操作,并根据所述匹配结果以及所述各物体对应的权重,确定所述匹配度。
5.根据权利要求1所述的方法,其特征在于,所述确定所述解码得到的语句的通顺度包括:
将所述解码得到的语句输入递归判别神经网络,根据所述递归判别神经网络各个时刻的第一输出,确定所述解码得到的语句的通顺度。
6.根据权利要求5所述的方法,其特征在于,所述根据所述递归判别神经网络各个时刻的输出,确定所述解码得到的语句的通顺度包括:
根据以下公式确定所述通顺度:
其中,radv代表所述通顺度,qt代表所述递归判别神经网络在t时刻的输出,n代表所述解码得到的语句的长度。
7.根据权利要求5所述的方法,其特征在于,进一步包括:
将语句样本输入到所述递归判别神经网络,获取所述递归判别神经网络各个时刻的第二输出;
根据所述递归判别神经网络各个时刻的第一输出和所述第二输出,对所述递归判别神经网络进行调整。
8.根据权利要求7所述的方法,其特征在于,所述根据所述递归判别神经网络各个时刻的第一输出和所述第二输出,对所述递归判别神经网络进行调整包括:
根据所述递归判别神经网络各个时刻的第一输出和所述第二输出,按照以下公式确定判别损失:
根据所述判别损失对所述递归判别神经网络进行调整;
其中,qt代表所述递归判别神经网络在t时刻的第一输出,代表所述递归判别神经网络在t时刻的第二输出;l表示所述语句样本的长度;Ladv表示所述判别损失;n代表所述解码得到的语句的长度。
9.根据权利要求5所述的方法,其特征在于,进一步包括:
根据所述解码得到的语句重构图像;
确定所述重构的图像与所述图像样本之间的差异度;
根据所述差异度,对所述递归判别神经网络进行调整。
10.根据权利要求1所述的方法,其特征在于,进一步包括:
根据所述解码得到的语句重构图像;
确定所述重构的图像与所述图像样本之间的相似度;
根据所述相似度,对所述递归解码神经网络进行调整。
11.根据权利要求10所述的方法,其中,所述根据所述解码得到的语句重构图像包括:
获取与所述解码得到的语句对应的语句特征向量;
将所述语句特征向量映射到图像特征空间,得到对应的图像特征向量;
所述确定所述重构的图像与所述图像样本之间的相似度包括:
将所述通过映射得到图像特征向量,与所述图像样本的图像特征向量进行比较,确定所述重构的图像与所述图像样本之间的相似度。
12.根据权利要求1所述的方法,其特征在于,进一步包括:
获取语句样本对应的语句特征向量,将所述语句特征向量映射到图像特征的空间,得到对应的图像特征向量;
获取与所述映射得到的图像特征向量对应的重构语句;
将所述语句样本与所述重构语句进行比较,根据比较结果对所述递归解码神经网络进行调整。
13.一种图像描述模型的训练装置,其特征在于,所述图像描述模型包括卷积编码神经网络和递归解码神经网络;所述装置包括:
编码模块,用于通过所述卷积编码神经网络,获取图像样本的图像特征向量;
解码模块,通过所述递归解码神经网络,对所述图像特征向量进行解码,得到用于描述所述图像样本的语句;
调整模块,用于确定所述解码得到的语句与所述图像样本之间的匹配度,根据所述匹配度对所述递归解码神经网络进行调整;以及确定所述解码得到的语句的通顺度,根据所述通顺度对所述递归解码神经网络进行调整。
14.根据权利要求13所述的装置,其特征在于,所述调整模块进一步用于,将所述解码得到的语句输入递归判别神经网络,根据所述递归判别神经网络各个时刻的第一输出,确定所述解码得到的语句的通顺度。
15.一种非易失性计算机可读存储介质,其中所述存储介质中存储有机器可读指令,所述机器可读指令可以由处理器执行以完成权利要求1-12中任一项所述的方法。
Priority Applications (4)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201811167476.9A CN110147806B (zh) | 2018-10-08 | 2018-10-08 | 图像描述模型的训练方法、装置及存储介质 |
PCT/CN2019/094891 WO2020073700A1 (zh) | 2018-10-08 | 2019-07-05 | 图像描述模型的训练方法、装置及存储介质 |
EP19870771.3A EP3866068A4 (en) | 2018-10-08 | 2019-07-05 | METHOD AND DEVICE FOR FORMING IMAGE DESCRIPTION MODEL AND INFORMATION HOLDER |
US17/075,618 US20210034981A1 (en) | 2018-10-08 | 2020-10-20 | Method and apparatus for training image caption model, and storage medium |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201811167476.9A CN110147806B (zh) | 2018-10-08 | 2018-10-08 | 图像描述模型的训练方法、装置及存储介质 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN110147806A true CN110147806A (zh) | 2019-08-20 |
CN110147806B CN110147806B (zh) | 2023-04-07 |
Family
ID=67588352
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201811167476.9A Active CN110147806B (zh) | 2018-10-08 | 2018-10-08 | 图像描述模型的训练方法、装置及存储介质 |
Country Status (4)
Country | Link |
---|---|
US (1) | US20210034981A1 (zh) |
EP (1) | EP3866068A4 (zh) |
CN (1) | CN110147806B (zh) |
WO (1) | WO2020073700A1 (zh) |
Cited By (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111537853A (zh) * | 2020-06-29 | 2020-08-14 | 国网山东省电力公司菏泽供电公司 | 基于多源异构数据分析的开关柜局部放电智能检测方法 |
CN112183525A (zh) * | 2020-09-15 | 2021-01-05 | 中保车服科技服务股份有限公司 | 一种文本识别模型的构建及文本识别方法和装置 |
CN112927136A (zh) * | 2021-03-05 | 2021-06-08 | 江苏实达迪美数据处理有限公司 | 一种基于卷积神经网络域适应的图像缩小方法及系统 |
CN114881242A (zh) * | 2022-04-21 | 2022-08-09 | 西南石油大学 | 一种基于深度学习的图像描述方法及系统、介质和电子设备 |
Families Citing this family (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112528894A (zh) * | 2020-12-17 | 2021-03-19 | 科大讯飞股份有限公司 | 一种差异项判别方法及装置 |
CN113657390B (zh) * | 2021-08-13 | 2022-08-12 | 北京百度网讯科技有限公司 | 文本检测模型的训练方法和检测文本方法、装置和设备 |
US20230206010A1 (en) * | 2021-12-23 | 2023-06-29 | Microsoft Technology Licensing, Llc | Utilizing visual and textual aspects of images with recommendation systems |
CN114359942A (zh) * | 2022-01-11 | 2022-04-15 | 平安科技(深圳)有限公司 | 基于人工智能的字幕提取方法、装置、设备和存储介质 |
Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107480144A (zh) * | 2017-08-03 | 2017-12-15 | 中国人民大学 | 具备跨语言学习能力的图像自然语言描述生成方法和装置 |
CN107657008A (zh) * | 2017-09-25 | 2018-02-02 | 中国科学院计算技术研究所 | 基于深度判别排序学习的跨媒体训练及检索方法 |
CN107832292A (zh) * | 2017-11-02 | 2018-03-23 | 合肥工业大学 | 一种基于神经网络模型的图像到汉语古诗的转换方法 |
CN108288067A (zh) * | 2017-09-12 | 2018-07-17 | 腾讯科技(深圳)有限公司 | 图像文本匹配模型的训练方法、双向搜索方法及相关装置 |
CN108416059A (zh) * | 2018-03-22 | 2018-08-17 | 北京市商汤科技开发有限公司 | 图像描述模型的训练方法和装置、设备、介质、程序 |
Family Cites Families (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN105654135A (zh) * | 2015-12-30 | 2016-06-08 | 成都数联铭品科技有限公司 | 一种基于递归神经网络的图像文字序列识别系统 |
CN106446782A (zh) * | 2016-08-29 | 2017-02-22 | 北京小米移动软件有限公司 | 图像识别方法及装置 |
US10089742B1 (en) * | 2017-03-14 | 2018-10-02 | Adobe Systems Incorporated | Automatically segmenting images based on natural language phrases |
CN108228686B (zh) * | 2017-06-15 | 2021-03-23 | 北京市商汤科技开发有限公司 | 用于实现图文匹配的方法、装置和电子设备 |
-
2018
- 2018-10-08 CN CN201811167476.9A patent/CN110147806B/zh active Active
-
2019
- 2019-07-05 WO PCT/CN2019/094891 patent/WO2020073700A1/zh unknown
- 2019-07-05 EP EP19870771.3A patent/EP3866068A4/en active Pending
-
2020
- 2020-10-20 US US17/075,618 patent/US20210034981A1/en active Pending
Patent Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107480144A (zh) * | 2017-08-03 | 2017-12-15 | 中国人民大学 | 具备跨语言学习能力的图像自然语言描述生成方法和装置 |
CN108288067A (zh) * | 2017-09-12 | 2018-07-17 | 腾讯科技(深圳)有限公司 | 图像文本匹配模型的训练方法、双向搜索方法及相关装置 |
CN107657008A (zh) * | 2017-09-25 | 2018-02-02 | 中国科学院计算技术研究所 | 基于深度判别排序学习的跨媒体训练及检索方法 |
CN107832292A (zh) * | 2017-11-02 | 2018-03-23 | 合肥工业大学 | 一种基于神经网络模型的图像到汉语古诗的转换方法 |
CN108416059A (zh) * | 2018-03-22 | 2018-08-17 | 北京市商汤科技开发有限公司 | 图像描述模型的训练方法和装置、设备、介质、程序 |
Non-Patent Citations (2)
Title |
---|
SHIYANG YAN ET AL: "Image Captioning using Adversarial Networks and Reinforcement Learning", 《2018 24TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR)》 * |
刘畅 等: "图像语义相似性网络的文本描述方法", 《计算机应用与软件》 * |
Cited By (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111537853A (zh) * | 2020-06-29 | 2020-08-14 | 国网山东省电力公司菏泽供电公司 | 基于多源异构数据分析的开关柜局部放电智能检测方法 |
CN112183525A (zh) * | 2020-09-15 | 2021-01-05 | 中保车服科技服务股份有限公司 | 一种文本识别模型的构建及文本识别方法和装置 |
CN112183525B (zh) * | 2020-09-15 | 2023-11-24 | 中保车服科技服务股份有限公司 | 一种文本识别模型的构建及文本识别方法和装置 |
CN112927136A (zh) * | 2021-03-05 | 2021-06-08 | 江苏实达迪美数据处理有限公司 | 一种基于卷积神经网络域适应的图像缩小方法及系统 |
CN114881242A (zh) * | 2022-04-21 | 2022-08-09 | 西南石油大学 | 一种基于深度学习的图像描述方法及系统、介质和电子设备 |
Also Published As
Publication number | Publication date |
---|---|
EP3866068A4 (en) | 2022-04-13 |
WO2020073700A1 (zh) | 2020-04-16 |
EP3866068A1 (en) | 2021-08-18 |
US20210034981A1 (en) | 2021-02-04 |
CN110147806B (zh) | 2023-04-07 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN110147806A (zh) | 图像描述模型的训练方法、装置及存储介质 | |
Perarnau et al. | Invertible conditional gans for image editing | |
Yao et al. | Describing videos by exploiting temporal structure | |
CN111783902B (zh) | 数据增广、业务处理方法、装置、计算机设备和存储介质 | |
CN110210513B (zh) | 数据分类方法、装置及终端设备 | |
CN111414946B (zh) | 基于人工智能的医疗影像的噪声数据识别方法和相关装置 | |
US11381651B2 (en) | Interpretable user modeling from unstructured user data | |
CN110446063A (zh) | 视频封面的生成方法、装置及电子设备 | |
CN112231582B (zh) | 一种基于变分自编码数据融合的网站推荐方法和设备 | |
CN110765286A (zh) | 跨媒体检索方法、装置、计算机设备和存储介质 | |
Mi et al. | Training-free uncertainty estimation for neural networks | |
CN112036564B (zh) | 图片识别方法、装置、设备及存储介质 | |
Nida et al. | Video augmentation technique for human action recognition using genetic algorithm | |
CN112883227B (zh) | 一种基于多尺度时序特征的视频摘要生成方法和装置 | |
CN111445545B (zh) | 一种文本转贴图方法、装置、存储介质及电子设备 | |
CN116028613B (zh) | 常识问答方法、系统、计算机设备和存储介质 | |
CN116959109A (zh) | 人体姿态图像的生成方法、装置、设备和存储介质 | |
CN116975347A (zh) | 图像生成模型训练方法及相关装置 | |
CN113516182B (zh) | 视觉问答模型训练、视觉问答方法和装置 | |
US11244015B1 (en) | Projecting queries into a content item embedding space | |
CN113569867A (zh) | 一种图像处理方法、装置、计算机设备及存储介质 | |
CN112509559A (zh) | 音频识别方法、模型训练方法、装置、设备及存储介质 | |
Guo et al. | Attributes guided facial image completion | |
CN111599363A (zh) | 一种语音识别的方法及其装置 | |
CN113011555B (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 |