BAM: Bottleneck Attention Module

By Jongchan Park, Sanghyun Woo, Joon-Young Lee, In-So Kweon. British Machine Vision Conference, 2018

1,311 citations recorded by the source corpus.

Recent advances in deep neural networks have been developed via architecture search for stronger representational power. In this work, we focus on the effect of attention in general deep neural networks. We propose a simple and effective attention module, named Bottleneck Attention Module (BAM), that can be integrated with any feed-forward convolutional neural networks. Our module infers an attention map along two separate pathways, channel and spatial. We place our module at each bottleneck of models where the downsampling of feature maps occurs. Our module constructs a hierarchical attention at bottlenecks with a number of parameters and it is trainable in an end-to-end manner jointly with any feed-forward models. We validate our BAM through extensive experiments on CIFAR-100, ImageNet-1K, VOC 2007 and MS COCO benchmarks. Our experiments show consistent improvement in classification and detection performances with various models, demonstrating the wide applicability of BAM. The code and models will be publicly available.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Radiology, Nuclear Medicine and Imaging, Model Architectures

View Paper: https://arxiv.org/abs/1807.06514

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Quoc V. Le
  2. Alan L. Yuille
  3. Jiashi Feng
  4. Irwan Bello
  5. Dacheng Tao
  6. Qibin Hou
  7. Barret Zoph
  8. Jianfeng Gao

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Ming-Hsuan Yang
  4. Xiaogang Wang
  5. Alan L. Yuille
  6. Jiashi Feng
  7. Yi Yang
  8. Andrew Zisserman

In Radiology, Nuclear Medicine and Imaging

  1. Junsuk Choe
  2. Quoc V. Le
  3. Irwan Bello
  4. Hyunjung Shim
  5. Barret Zoph
  6. Lorenzo Bonicelli
  7. Yunhe Wang
  8. Kai Han

In Model Architectures

  1. Quoc V. Le
  2. Alan L. Yuille
  3. Irwan Bello
  4. Barret Zoph
  5. Jonathon Shlens
  6. Qibin Hou
  7. Jiashi Feng
  8. Mingxing Tan

Citations in the corpus