注意力机制与双线性网络的垃圾图像分类研究

摘要：

针对基于人工垃圾分类既费力又费时,且因分拣出错率高、速度慢等问题,以高效率网络模型为基础,提出了一种双线性注意力机制的卷积神经网络分类方法.首先,为EfficientNetB3模型添加通道注意力机制模块,学习通道间的非线性关系,提高关键特征的表达力.然后,采用双线性汇合的方法计算不同空间位置的外积,并对不同空间位置计算平均汇合以得到双线性特征.仿真结果表明,上述方法为卷积神经网络添加注意力模块和双行线汇合的方法可以提供更强的特征表示,在垃圾图像分类任务中提高分类精度,理论分析和试验验证均验证了所提算法的有效性.

关键词：垃圾图像分类;注意力机制;双线性网络;神经网络

所属期刊栏目：38

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

资助基金：北京市教育委员会社科计划一般项目;北京印刷学院普通重点教学改革项目

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

页数：共5页

页码：222-226