ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks

Code

• github.com/BangguWu/ECANet
  OfficialIn paperpytorch★ 1,399
• github.com/rwightman/pytorch-image-models
  pytorch★ 36,538
• github.com/xmu-xiaoma666/External-Attention-pytorch
  pytorch★ 12,169
• github.com/e96031413/PyTorch_YOLOv4-tiny
  pytorch★ 9
• github.com/e96031413/AA-YOLO
  pytorch★ 9
• github.com/mnikitin/channel-attention
  mxnet★ 0
• github.com/mindspore-courses/External-Attention-MindSpore/blob/main/model/attention/ECAAttention.py
  mindspore★ 0
• github.com/JinLi711/Attention-Augmented-Convolution
  tf★ 0
• github.com/JinLi711/Convolution_Variants
  tf★ 0
• github.com/digantamisra98/Reproducibilty-Challenge-ECANET
  pytorch★ 0

Abstract

Recently, channel attention mechanism has demonstrated to offer great potential in improving the performance of deep convolutional neural networks (CNNs). However, most existing methods dedicate to developing more sophisticated attention modules for achieving better performance, which inevitably increase model complexity. To overcome the paradox of performance and complexity trade-off, this paper proposes an Efficient Channel Attention (ECA) module, which only involves a handful of parameters while bringing clear performance gain. By dissecting the channel attention module in SENet, we empirically show avoiding dimensionality reduction is important for learning channel attention, and appropriate cross-channel interaction can preserve performance while significantly decreasing model complexity. Therefore, we propose a local cross-channel interaction strategy without dimensionality reduction, which can be efficiently implemented via 1D convolution. Furthermore, we develop a method to adaptively select kernel size of 1D convolution, determining coverage of local cross-channel interaction. The proposed ECA module is efficient yet effective, e.g., the parameters and computations of our modules against backbone of ResNet50 are 80 vs. 24.37M and 4.7e-4 GFLOPs vs. 3.86 GFLOPs, respectively, and the performance boost is more than 2% in terms of Top-1 accuracy. We extensively evaluate our ECA module on image classification, object detection and instance segmentation with backbones of ResNets and MobileNetV2. The experimental results show our module is more efficient while performing favorably against its counterparts.

Tasks

Benchmark Results

Reproductions