计算机科学与探索

作者：陈晓雷, 卢禹冰, 曹宝宁, 林冬梅

单位：兰州理工大学 电气工程与信息工程学院，兰州 730000

关键词：注意力机制,一维卷积,图像分类,目标检测

目前大多数注意力机制模块在提高深度学习模型应用精度的同时也带来了模型复杂度的增加。针对这一问题，提出了一种轻量化高精度双通道注意力机制模块（EDCA）。EDCA将特征图分别沿通道、宽度和高度三个方向压缩并重新排列组合，采用一维卷积获取组合后的权重信息，接着将权重信息分割并应用于对应维度以获得特征关注度。在图像分类数据集miniImageNet与目标检测数据集Pascal VOC2007上对EDCA进行充分实验，实验结果表明，与SENet、CBAM、SGE、ECA-Net、Coordinate Attention相比，EDCA所需的计算量以及参数量更少。在miniImageNet数据集上使用ResNet50+EDCA时，Top-1精度较以上方法分别提升0.024 3、0.021 8、0.022 1、0.022 5、0.014 1；在Pascal VOC2007数据集上使用MobileNetV3+YOLOv4+EDCA时，AP50较SENet、CBAM、ECA-Net、Coordinate Attention分别提升0.009 4、0.004 6、0.005 9、0.001 4。

来源：2023年第4期

《计算机科学与探索》期刊编辑部

查看计算机科学与探索杂志2023年第4期