English

EAANet: Efficient Attention Augmented Convolutional Networks

Computer Vision and Pattern Recognition 2022-06-07 v1

Abstract

Humans can effectively find salient regions in complex scenes. Self-attention mechanisms were introduced into Computer Vision (CV) to achieve this. Attention Augmented Convolutional Network (AANet) is a mixture of convolution and self-attention, which increases the accuracy of a typical ResNet. However, The complexity of self-attention is O(n2) in terms of computation and memory usage with respect to the number of input tokens. In this project, we propose EAANet: Efficient Attention Augmented Convolutional Networks, which incorporates efficient self-attention mechanisms in a convolution and self-attention hybrid architecture to reduce the model's memory footprint. Our best model show performance improvement over AA-Net and ResNet18. We also explore different methods to augment Convolutional Network with self-attention mechanisms and show the difficulty of training those methods compared to ResNet. Finally, we show that augmenting efficient self-attention mechanisms with ResNet scales better with input size than normal self-attention mechanisms. Therefore, our EAANet is more capable of working with high-resolution images.

Keywords

Cite

@article{arxiv.2206.01821,
  title  = {EAANet: Efficient Attention Augmented Convolutional Networks},
  author = {Runqing Zhang and Tianshu Zhu},
  journal= {arXiv preprint arXiv:2206.01821},
  year   = {2022}
}

Comments

8 pages, 4 figures. Not published

R2 v1 2026-06-24T11:38:54.168Z