首页 | 官方网站   微博 | 高级检索  
     

局部特征融合的小样本分类
引用本文:董杨洋,宋蓓蓓,孙文方.局部特征融合的小样本分类[J].中国图象图形学报,2023,28(7):2093-2104.
作者姓名:董杨洋  宋蓓蓓  孙文方
作者单位:1.长安大学信息工程学院,西安 710064;2.西安电子科技大学空间科学与技术学院,西安 710126
基金项目:国家自然科学基金项目(62001059);陕西省重点研发计划资助(2021GY-019)
摘    要:目的 小样本学习是一项具有挑战性的任务,旨在利用有限数量的标注样本数据对新的类别数据进行分类。基于度量的元学习方法是当前小样本分类的主流方法,但往往仅使用图像的全局特征,且模型分类效果很大程度上依赖于特征提取网络的性能。为了能够充分利用图像的局部特征以及提高模型的泛化能力,提出一种基于局部特征融合的小样本分类方法。方法 首先,将输入图像进行多尺度网格分块处理后送入特征提取网络以获得局部特征;其次,设计了一个基于Transformer架构的局部特征融合模块来得到包含全局信息的局部增强特征,以提高模型的泛化能力;最后,以欧几里得距离为度量,计算查询集样本特征向量与支持集类原型之间的距离,实现分类。结果 在小样本分类中常用的3个数据集上与当前先进的方法进行比较,在5-way 1-shot和5-way 5-shot的设置下相对次优结果,所提方法在MiniImageNet数据集上的分类精度分别提高了2.96%和2.9%,在CUB(Caltech-UCSD Birds-200-2011)数据集上的分类精度分别提高了3.22%和1.77%,而在TieredImageNet数据集上的分类精度与最优结果相当,实验结果表明了所提方法的有效性。结论 提出的小样本分类方法充分利用了图像的局部特征,同时改善了模型的特征提取能力和泛化能力,使小样本分类结果更为准确。

关 键 词:小样本学习  度量学习  局部特征  Transformer  特征融合
收稿时间:2022/2/24 0:00:00
修稿时间:2022/5/23 0:00:00

Local feature fusion network-based few-shot image classification
Dong Yangyang,Song Beibei,Sun Wenfang.Local feature fusion network-based few-shot image classification[J].Journal of Image and Graphics,2023,28(7):2093-2104.
Authors:Dong Yangyang  Song Beibei  Sun Wenfang
Affiliation:School of Information Engineering, Chang''an University, Xi''an 710064, China; School of Aerospace Science and Technology, Xidian University, Xi''an 710126, China
Abstract:
Keywords:few-shot learning  metric learning  local feature  Transformer  feature fusion
点击此处可从《中国图象图形学报》浏览原始摘要信息
点击此处可从《中国图象图形学报》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号