Image classification has witnessed remarkable progress due to the advent of Convolutional Neural Networks [1] (CNNs). However, the capacity to focus on salient information within images remains a challenge. This research explores the integration of Channel Attention and Spatial Attention mechanisms into baseline CNN architectures to enhance classification performance. We conduct experiments on standard image classification datasets to evaluate the efficacy of integrating CBAM [13] (Convolutional Block Attention Module), BAM [11] (Bottleneck Attention Module), and scSE [12] (spatial and channel Squeeze & Excitation) against baseline models. Experimental results demonstrate that leveraging these attention modules significantly improves classification accuracy.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

The Effectiveness of Channel and Spatial Attention for Improving Image Classification

  • Le Thao Tram Nguyen,
  • Minh Phung Bui,
  • Quoc Huy Nguyen

摘要

Image classification has witnessed remarkable progress due to the advent of Convolutional Neural Networks [1] (CNNs). However, the capacity to focus on salient information within images remains a challenge. This research explores the integration of Channel Attention and Spatial Attention mechanisms into baseline CNN architectures to enhance classification performance. We conduct experiments on standard image classification datasets to evaluate the efficacy of integrating CBAM [13] (Convolutional Block Attention Module), BAM [11] (Bottleneck Attention Module), and scSE [12] (spatial and channel Squeeze & Excitation) against baseline models. Experimental results demonstrate that leveraging these attention modules significantly improves classification accuracy.