English

Attention is not Explanation

Computation and Language 2019-05-10 v3 Artificial Intelligence

Abstract

Attention mechanisms have seen wide adoption in neural NLP models. In addition to improving predictive performance, these are often touted as affording transparency: models equipped with attention provide a distribution over attended-to input units, and this is often presented (at least implicitly) as communicating the relative importance of inputs. However, it is unclear what relationship exists between attention weights and model outputs. In this work, we perform extensive experiments across a variety of NLP tasks that aim to assess the degree to which attention weights provide meaningful `explanations' for predictions. We find that they largely do not. For example, learned attention weights are frequently uncorrelated with gradient-based measures of feature importance, and one can identify very different attention distributions that nonetheless yield equivalent predictions. Our findings show that standard attention modules do not provide meaningful explanations and should not be treated as though they do. Code for all experiments is available at https://github.com/successar/AttentionExplanation.

Keywords

Cite

@article{arxiv.1902.10186,
  title  = {Attention is not Explanation},
  author = {Sarthak Jain and Byron C. Wallace},
  journal= {arXiv preprint arXiv:1902.10186},
  year   = {2019}
}

Comments

Accepted as NAACL 2019 Long Paper

R2 v1 2026-06-23T07:52:15.678Z