基于多分支神经网络模型的弱监督细粒度图像分类方法

摘要：

针对传统基于注意力机制的神经网络不能联合关注局部特征和旋转不变特征的问题,提出一种基于多分支神经网络模型的弱监督细粒度图像分类方法.首先,用轻量级类激活图(CAM)网络定位有潜在语义信息的局部区域,设计可变形卷积的残差网络ResNet-50和旋转不变编码的方向响应网络(ORN);其次,利用预训练模型分别初始化特征网络,并输入原图和以上局部区域分别对模型进行微调;最后,组合三个分支内损失和分支间损失优化整个网络,对测试集进行分类预测.所提方法在CUB-200-2011和FGVC_Aircraft数据集上的分类准确率分别达到87.7％和90.8％,与多注意力卷积神经网络(MA-CNN)方法相比,分别提高了1.2个百分点和0.9个百分点;在Aircraft_2数据集上的分类准确率达到91.8％,比ResNet-50网络提高了4.1个百分点.实验结果表明,所提方法有效提高了弱监督细粒度图像分类的准确率.

关键词：细粒度图像分类、深度学习、弱监督、可变形卷积、类激活图、方向响应网络

所属期刊栏目：40

分类号：TP391.4(计算技术、计算机技术)

资助基金：国家自然科学基金资助项目;湖北省自然科学基金资助项目;冶金自动化与检测技术教育部工程研究中心开放基金资助项目

在线出版日期：2020-06-02（万方平台首次上网日期，不代表论文的发表时间）

页数：共6页

页码：1295-1300