科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对小样本图像分类任务中基于卷积神经网络的特征提取模块难以捕获远程语义信息和边特征相似度度量单一的问题，本文提出一种基于Swin Transformer的图神经网络小样本图像分类算法。首先，利用Swin Transformer网络来提取图像特征，并将该特征作为节点特征输入图神经网络。然后，通过增加额外的度量方式改进了边特征相似度量模块，形成双度量模块以计算节点特征之间的相似度，将得到的相似度作为边特征输入图神经网络。最后交替更新节点和边特征来获取图像标签的信息。在StanfordDogs、StanfordCars和CUB-200-2011三个数据集上的5-way 1-shot任务分类准确率分别达到了85.21%、91.10%和91.08%，在小样本图像分类任务中取得了显著的效果。
- 单位
