arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.09838cs.CV

Transformer Interpretability Beyond Attention Visualization

Tel Aviv University(特拉维夫大学);Facebook AI Research (FAIR)(Facebook AI研究院)

Hila Chefer, Shir Gur, Lior Wolf

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑