[发明专利]一种关系网络小样本图像分类方法、系统、装置及介质

说明书

本发明公开了一种关系网络小样本图像分类方法、系统、装置及介质，其中方法包括：划分数据集，在每个训练批次中，将数据集划分为训练集和测试集，训练集的样本标注有样本标签；采用嵌入模块提取数据集中图像的图片特征表示；利用自注意力模块学习任务间的上下文信息，将任务无关的图片特征表示转化为任务相关的图片特征表示；根据任务相关的图片特征表示，以基于原学习的关系网络度量方式训练神经网络，采用训练后的神经网络进行图像分类。本发明通过注意力机制，将任务无关特征向量转化为任务相关的特征向量，具有考虑任务上下文信息、缩小类内距离的优点，更有利于进行后面的度量计算相似度得分，可广泛应用于计算机视觉领域。

技术领域

本发明涉及计算机视觉领域，尤其涉及一种关系网络小样本图像分类方法、系统、装置及介质。

背景技术

小样本学习是人工智能中的重要研宄方向之一，人类具有从少数样本中快速学习和快速归纳的能力。以图片分类任务为例，人类可以通过观察少数几张图片样本，从这些样本中学习特征，迅速分辨出与其同一类或不同类的图片。孩子在小时候学习辨认不同的动物时，通常仅需看一两张这种动物的图像就能在不同的场景下认出这个物种，因为他已经学习到了很多经验知识，利用这些经验知识很快就能完成对新目标的辨认。这种从少数样本中快速学习，并且利用已学到的经验学习新的知识，而不是面对每个任务都从头学的能力，称为小样本学习。

通过有限样本完成模型的训练，然后在面对训练过程中没有出现过的新样本同样具有识别能力。随着深度学习的发展，深度学习技术己在视频分类、身份识别、自动驾驶M等视觉领域得到了广泛应用，然而在这些领域中常常面对数据稀缺的问题，同时对一些数据的采集也具有较高的难度，对采集来的数据进行标注也是费时费力的事情。人类能够从单个或少数样本中学习新概念，而且能够从稀疏的数据中获得极其丰富的表征，并且可以通过之前的学习经验，进一步增强未来的学习能力。如何让机器具备这种能力，对于人工智能至关重要。然而目前此领域大多是从特征提取网络直接提取特征向量来进行度量，这类装置的缺点是没有充分考虑每个任务中各个类别的联系，进而提取最符合这个任务的一个比较有代表性辨别性的特征向量，如猫和老虎相比较与猫和狗相比较，这两个任务所提取到的代表性特征向量是不相同的。

发明内容

为至少一定程度上解决现有技术中存在的技术问题之一，本发明的目的在于提供一种关系网络小样本图像分类方法、系统、装置及介质。

本发明所采用的技术方案是：

一种关系网络小样本图像分类方法，包括以下步骤：

划分数据集，在每个训练批次中，将数据集划分为训练集和测试集，训练集的样本标注有样本标签；

采用嵌入模块提取数据集中图像的图片特征表示；

由于嵌入模块提取的图片特征表示是任务无关型，利用自注意力模块学习任务间的上下文信息，将任务无关的图片特征表示转化为任务相关的图片特征表示；

根据任务相关的图片特征表示，以基于原学习的关系网络度量方式训练神经网络，采用训练后的神经网络进行图像分类。

进一步地，所述划分数据集，在每个训练批次中，将数据集划分为训练集和测试集，包括：

每一个批次对应一个任务，每个批次将数据集划分为训练集和测试集，将训练集中的支持集输入嵌入模块获取图片特征表示，训练集中的查询集带有标签，负责衡量与支持集中样本的距离；

训练集中的支持集和查询集负责不断更新网络模型的参数。

进一步地，所述嵌入模块采用残差网络来实现，所述残差网络包含4个依次连接基础模块，每个基础模块包含三个卷积小模块，在每个基础模块的输出端附加一个2×2最大池化层，在最后一个最大池化层的输出端附加一个2×2平均池化层；