PDF下载
基于稀疏化双线性卷积神经网络的细粒度图像分类

马力1 王永雄2

1.上海理工大学光电信息与计算机工程学院;2.上海康复器械工程技术研究中心 ​

摘要: 针对双线性卷积神经网络(B-CNN)在细粒度图像分类中因参数过多、复杂度过高而导致的过拟合问题,提出稀疏化B-CNN.首先对B-CNN的每个特征通道引入比例因子,在训练中采用正则化方法对其稀疏.然后利用比例因子的大小判别特征通道的重要性.最后将不重要特征通道按一定比例裁剪,消除网络过拟合,提高关键特征的显著性.稀疏化B-CNN属于弱监督学习,可实现端到端训练.在FGVC-aircraft、Stanford dogs、Stanford cars这3个细粒度图像数据集上的实验表明,稀疏化B-CNN的准确率高于B-CNN,也优于或基本接近其它通用的细粒度图像分类算法.
关键词: 细粒度图像分类;双线性卷积神经网络(B-CNN);过拟合;网络稀疏;网络剪枝
DOI:
基金资助:
文章地址: