Transformer Interpretability Beyond Attention Visualization
AI-generated Key Points
- Self-attention techniques, particularly Transformers, are prominent in text processing and computer vision classification tasks
- Existing approaches face challenges with attention layers and skip connections
- The proposed method uses Deep Taylor Decomposition to compute relevancy in Transformer networks
- Total relevancy is maintained across layers in the proposed method
- The proposed method outperforms existing explainability methods in visual Transformer networks and text classification problems
- Considering all attention layers leads to the best results while maintaining relevance and gradient integration
- Focusing on the block closest to the output is more informative than the one closest to the input in single block visualizations
- Funding for this work was provided by the European Research Council (ERC) under Horizon 2020 program
Authors: Hila Chefer, Shir Gur, Lior Wolf
Abstract: Self-attention techniques, and specifically Transformers, are dominating the field of text processing and are becoming increasingly popular in computer vision classification tasks. In order to visualize the parts of the image that led to a certain classification, existing methods either rely on the obtained attention maps or employ heuristic propagation along the attention graph. In this work, we propose a novel way to compute relevancy for Transformer networks. The method assigns local relevance based on the Deep Taylor Decomposition principle and then propagates these relevancy scores through the layers. This propagation involves attention layers and skip connections, which challenge existing methods. Our solution is based on a specific formulation that is shown to maintain the total relevancy across layers. We benchmark our method on very recent visual Transformer networks, as well as on a text classification problem, and demonstrate a clear advantage over the existing explainability methods.
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
Assess the quality of the AI-generated content by voting
Score: 0
Why do we need votes?
Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.