Dissecting Generation Modes for Abstractive Summarization Models via Ablation and Attribution
arXiv:2106.01518
Abstract
Despite the prominence of neural abstractive summarization models, we know little about how they actually form summaries and how to understand where their decisions come from. We propose a two-step method to interpret summarization model decisions. We first analyze the model's behavior by ablating the full model to categorize each decoder decision into one of several generation modes: roughly, is the model behaving like a language model, is it relying heavily on the input, or is it somewhere in between? After isolating decisions that do depend on the input, we explore interpreting these decisions using several different attribution methods. We compare these techniques based on their ability to select content and reconstruct the model's predicted token from perturbations of the input, thus revealing whether highlighted attributions are truly important for the generation of the next token. While this machinery can be broadly useful even beyond summarization, we specifically demonstrate its capability to identify phrases the summarization model has memorized and determine where in the training pipeline this memorization happened, as well as study complex generation phenomena like sentence fusion on a per-instance basis.
arXiv commentACL 2021; 16 pages
References in corpus (7)
- Axiomatic Attribution for Deep Networks
- Teaching Machines to Read and Comprehend
- Understanding Black-box Predictions via Influence Functions
- Towards Robust Interpretability with Self-Explaining Neural Networks
- Interpretation of Prediction Models Using the Input Gradient
- Sticking to the Facts: Confident Decoding for Faithful Data-to-Text Generation
- Prevent the Language Model from being Overconfident in Neural Machine Translation