Transformer Interpretability Beyond Attention Visualization

AI-generated keywords: Transformers Attention Layers Skip Connections Relevancy Scores Interpretability

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
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Hila Chefer, Shir Gur, Lior Wolf

License: CC BY 4.0

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.

Submitted to arXiv on 17 Dec. 2020

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2012.09838v2

In the field of text processing and computer vision classification tasks, self-attention techniques, particularly Transformers, have gained significant prominence. These methods rely on attention maps or heuristic propagation to visualize the image components that contribute to a specific classification. However, existing approaches face challenges when it comes to attention layers and skip connections. To address these challenges, this work proposes a novel method for computing relevancy in Transformer networks. The proposed method utilizes the Deep Taylor Decomposition principle to assign local relevance scores and then propagates them through the layers. This propagation involves attention layers and skip connections, which pose difficulties for existing methods. The formulation used in this solution ensures that total relevancy is maintained across layers. To evaluate the effectiveness of the proposed method, it is benchmarked on recent visual Transformer networks as well as a text classification problem. The results demonstrate a clear advantage over existing explainability methods. In further analysis, different variants of the proposed method are compared in terms of performance. It is found that considering all attention layers leads to the best results while maintaining relevance and gradient integration. However, considering only the last attention layer still yields moderate performance. Among single block visualizations, focusing on the block closest to the output proves more informative than the one closest to the input. The authors acknowledge funding from the European Research Council (ERC) under the European Union's Horizon 2020 research and innovation program. The first author's contribution is part of a project conducted at Tel Aviv University. Overall, this work presents specific solutions to challenges faced by existing methods in interpreting Transformers' attentions.
Created on 27 Sep. 2023

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 representation

Look 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.