科研之友
微信
新浪微博
Facebook
分享链接
摘要
为了更好地抑制卷积神经网络在图像分类训练过程中出现的过拟合现象，提高图像分类精度和算法的稳定性，提出了融合掩模和注意机制的CNN图像分类算法。算法使用去噪神经网络提取的去噪深度特征，作为深度CNN网络模型总框架的输入，基于掩模预训练方法训练网络模型；再使用不同注意类型的激活函数，在标准图像库Corel-1000上充分进行实验。实验结果表明：使用基于去噪神经网络的掩模预训练方法，不仅较大程度提升了图像分类效果，而且整个模型的性能更趋于稳定和健壮。加入混合注意机制后，实验效果得到进一步提升。算法通过降低图像背景噪声和深度特征噪声的干扰，能有效抑制训练过程中出现的过拟合现象；且不同注意类型激活函数的作用不尽相同，其中混合注意机制的实验效果最优。
- 单位
