Namhoon Lee

Learn To Pay Attention

2018 · arXiv (Cornell University)

We propose an end-to-end-trainable attention module for convolutional neural network (CNN) architectures built for image classification. The module takes as input the 2D feature vector maps which form the intermediate representations of the input image …