Saccader: Improving Accuracy of Hard Attention Models for Vision
Gamaleldin Elsayed, Simon Kornblith, Quoc V Le
Abstract
Although deep convolutional neural networks achieve state-of-the-art performance across nearly all image classification tasks, their decisions are difficult to interpret. One approach that offers some level of interpretability by design is \textit{hard attention}, which uses only relevant portions of the image. However, training hard attention models with only class label supervision is challenging, and hard attention has proved difficult to scale to complex datasets. Here, we propose a novel hard attention model, which we term Saccader. Key to Saccader is a pretraining step that requires only class labels and provides initial attention locations for policy gradient optimization. Our best models narrow the gap to common ImageNet baselines, achieving $75\%$ top-1 and $91\%$ top-5 while attending to less than one-third of the image.
BibTeX
@inproceedings{NEURIPS2019_8dd48d6a,
author = {Elsayed, Gamaleldin and Kornblith, Simon and Le, Quoc V},
booktitle = {Advances in Neural Information Processing Systems},
editor = {H. Wallach and H. Larochelle and A. Beygelzimer and F. d\textquotesingle Alch\'{e}-Buc and E. Fox and R. Garnett},
pages = {},
publisher = {Curran Associates, Inc.},
title = {Saccader: Improving Accuracy of Hard Attention Models for Vision},
url = {https://proceedings.neurips.cc/paper_files/paper/2019/file/8dd48d6a2e2cad213179a3992c0be53c-Paper.pdf},
volume = {32},
year = {2019}
}