arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2301.08669cs.CVstat.ML
收藏

整体可解释的视觉Transformer

Holistically Explainable Vision Transformers

Max Planck Institute for Informatics(马克斯·普朗克信息学研究所);CISPA Helmholtz Center for Information Security(CISPA亥姆霍兹信息安全中心)

Moritz Böhle, Mario Fritz, Bernt Schiele

摘要:

Transformers在许多任务和领域的机器学习领域越来越占主导地位,这增加了理解其输出的重要性。虽然他们的注意力模块提供了对内部工作的部分洞察,但注意力得分已被证明不足以解释整个模型。为了解决这个问题,我们提出了B-cos Transformers,它本质上为他们的决策提供了全面的解释。具体来说,我们将每个模型组件(例如多层感知器,注意力层和标记化模块)制定为动态线性,这使我们能够通过单个线性变换忠实地总结整个Transformer。我们将我们提出的设计应用于视觉Transformer(ViTs),并表明所产生的模型(称为Bcos-ViTs)具有高度的可解释性,并且在ImageNet上与基线ViTs相比具有竞争力。代码将很快提供。

英文摘要:

Transformers increasingly dominate the machine learning landscape across many tasks and domains, which increases the importance for understanding their outputs. While their attention modules provide partial insight into their inner workings, the attention scores have been shown to be insufficient for explaining the models as a whole. To address this, we propose B-cos transformers, which inherently provide holistic explanations for their decisions. Specifically, we formulate each model component - such as the multi-layer perceptrons, attention layers, and the tokenisation module - to be dynamic linear, which allows us to faithfully summarise the entire transformer via a single linear transform. We apply our proposed design to Vision Transformers (ViTs) and show that the resulting models, dubbed Bcos-ViTs, are highly interpretable and perform competitively to baseline ViTs on ImageNet. Code will be made available soon.

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

↑