信息与控制

北大核心,JST,CSCD,WJCI,

国内刊号:21-1138/TP

国际刊号:1002-0411

信息与控制杂志2025年第2期:一种轻量级空间位置注意力模块及其在图像分类网络中的应用

发布日期:

作者:许云峰, 刘畅达, 梅卫, 徐振峰, 张妍

关键词:注意力机制, 空间位置信息, 图像分类

针对以往的注意力方法忽略了空间位置信息对跨维度交互的重要性的问题,提出了一种轻量级的空间位置注意力模块(spatial location attention module,SLAM),该模块通过3个分支结构计算输入特征图在水平方向、垂直方向和通道方向上的位置信息注意力权重,并沿3个空间方向进行特征聚合得到注意力权重特征图,实现特征图中空间和位置信息注意力权重的自适应调整。基于该模块改进了ResNet18、ResNet50和MobileNetV2网络,并且针对图像分类任务进行了大量的实验,结果表明SLAM能够显著提高模型的性能,且性能优于其他注意力方法。在ImageNet-1K和Stanford-Cars数据集的分类任务上,基于SLAM改进的ResNet18、ResNet50和MobileNetV2网络的Top-1准确率最高分别提升2.62%和2.4%。在废钢评级任务中,基于SLAM改进的YOLOv5s和YOLOv8s网络与基于CBAM (convolutional block attention module)和CA (coordinate attention)改进的这两个网络相比在召回率、F1分数、mAP 0.5:0.95和mAP 0.5等4个指标上均有提高。

来源:2025年第2期

《信息与控制》期刊编辑部

查看信息与控制杂志2025年第2期

声明

严正声明:本站非期刊官网,非中介代理。

本站仅提供学术规范服务:快速预审、润色编辑服务、中英文查重、降重、去重服务、推荐合适的期刊投稿等学术规范服务。 如需提供学术规范服务请联系在线编辑。

联系我们

  • 地址:辽宁省沈阳市创新路135号
  • 电话:024-23970049
  • E-mail:xk@sia.cn

咨询工作人员