CBAM: Convolutional Block Attention Module

  • 2018-07-17 16:05:59
  • Sanghyun Woo, Jongchan Park, Joon-Young Lee, In So Kweon
  • 9

Abstract

We propose Convolutional Block Attention Module (CBAM), a simple yeteffective attention module for feed-forward convolutional neural networks.Given an intermediate feature map, our module sequentially infers attentionmaps along two separate dimensions, channel and spatial, then the attentionmaps are multiplied to the input feature map for adaptive feature refinement.Because CBAM is a lightweight and general module, it can be integrated into anyCNN architectures seamlessly with negligible overheads and is end-to-endtrainable along with base CNNs. We validate our CBAM through extensiveexperiments on ImageNet-1K, MS~COCO detection, and VOC~2007 detection datasets.Our experiments show consistent improvements in classification and detectionperformances with various models, demonstrating the wide applicability of CBAM.The code and models will be publicly available.

 

Quick Read (beta)

loading the full paper ...