Mixed local channel attention for object detection

被引:198
作者
Wan, Dahang
Lu, Rongsheng [1 ]
Shen, Siyuan
Xu, Ting
Lang, Xianli
Ren, Zhijie
机构
[1] Hefei Univ Technol, Sch Instrument Sci & Optoelect Engn, Hefei 230009, Peoples R China
关键词
Attention mechanism; Local channel attention; Object detection; Deep learning algorithm; Convolutional neural network; DATASET;
D O I
10.1016/j.engappai.2023.106442
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Attention mechanism, one of the most extensively utilized components in computer vision, can assist neural networks in emphasizing significant elements and suppressing irrelevant ones. However, the vast majority of channel attention mechanisms only contain channel feature information and ignore spatial feature information, resulting in poor model representation effect or object detection performance, and the spatial attention modules were often complex and expensive. In order to strike a balance between performance and complexity, this paper proposes a lightweight Mixed Local Channel Attention (MLCA) module to improve the performance of the object detection network, and it can simultaneously incorporate both channel information and spatial information, as well as local information and global information to improve the expression effect of the network. On this basis, the MobileNet-Attention-YOLO(MAY) algorithm for comparing the performance of various attention modules is presented. On the Pascal VOC and SMID datasets, MLCA achieves a better balance between model representation efficacy, performance, and complexity than alternative attention techniques. Against the Squeeze-and-Excitation(SE) attention mechanism on the PASCAL VOC dataset and the Coordinate Attention(CA) method on the SIMD dataset, the mAP is enhanced by 1.0 % and 1.5 %, respectively.
引用
收藏
页数:15
相关论文
共 87 条
[1]   Distributed Deep CNN-LSTM Model for Intrusion Detection Method in IoT-Based Vehicles [J].
Alferaidi, Ali ;
Yadav, Kusum ;
Alharbi, Yasser ;
Razmjooy, Navid ;
Viriyasitavat, Wattana ;
Gulati, Kamal ;
Kautish, Sandeep ;
Dhiman, Gaurav .
MATHEMATICAL PROBLEMS IN ENGINEERING, 2022, 2022
[2]  
Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
[3]   C plus EffxNet: A novel hybrid approach for COVID-19 diagnosis on CT images based on CBAM and EfficientNet [J].
Canayaz, Murat .
CHAOS SOLITONS & FRACTALS, 2021, 151
[4]  
Chen BH, 2019, Arxiv, DOI arXiv:1908.05819
[5]  
Chen D, 2021, Arxiv, DOI [arXiv:2103.11696, DOI 10.48550/ARXIV.2103.11696, 10.48550/arXiv.2103.11696]
[6]  
Chen K, 2019, Arxiv, DOI arXiv:1906.07155
[7]  
Chen YP, 2020, Arxiv, DOI arXiv:1912.03458
[8]   High-efficiency chaotic time series prediction based on time convolution neural network [J].
Cheng, Wei ;
Wang, Yan ;
Peng, Zheng ;
Ren, Xiaodong ;
Shuai, Yubei ;
Zang, Shengyin ;
Liu, Hao ;
Cheng, Hao ;
Wu, Jiagui .
CHAOS SOLITONS & FRACTALS, 2021, 152
[9]  
DeVries T, 2017, Arxiv, DOI arXiv:1708.04552
[10]  
Diba Ali., 2019, Spatio-Temporal Channel Correlation Networks for Action Classification