CAM: compound attention module for improved vision models
In recent years, the attention mechanism has played a significant role in enhancing algorithm performance in deep learning-based visual tasks. Most methods focus on developing more complex attention mechanisms to improve network performance, which inevitably increases the computational complexity of the model. To balance performance and computational complexity, this paper proposes the CAM (Compound Attention Module) attention mechanism, which delivers substantial performance improvements with only a slight increase in parameters. The CAM module operates in two dimensions: space and channel. It is a lightweight, plug-and-play module with minimal computational overhead. We validate our CAM module through extensive experiments on image classification and object detection tasks using the CIFAR-100 and VisDrone2019 datasets. Experimental results demonstrate that the model consistently improves image classification and object detection performance, outperforming similar modules.
Authors
- Fei Tan (ORCID: https://orcid.org/0000-0001-7057-581X)
- Zhiqiang Du (ORCID: https://orcid.org/0000-0002-8314-4354)
- H. M. Liu
Institutions
- Beijing Academy of Artificial Intelligence (CN)
- Sichuan University of Science and Engineering (CN)
Publication Details
- Published
- 2024-10-24
- DOI
- https://doi.org/10.1117/12.3050534
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00