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

Token 转换至关重要:迈向 Vision Transformer 的忠实事后解释

Token Transformation Matters: Towards Faithful Post-hoc Explanation for Vision Transformer

Illinois Institute of Technology(伊利诺伊理工学院);Carnegie Mellon University(卡内基梅隆大学)

Junyi Wu, Bin Duan, Weitai Kang, Hao Tang, Yan Yan

AI总结:

针对现有 Vision Transformer 解释方法忽略 token 转换信息的问题,提出事后解释方法 TokenTM,通过量化 token 转换效果并结合注意力权重,在分割和扰动测试中展现出优越性。

摘要:

虽然 Transformers 在各种计算机视觉应用中迅速普及,但对其内部机制的事后解释在很大程度上仍未被探索。Vision Transformers 通过将图像区域表示为转换后的 token 并通过注意力权重整合它们来提取视觉信息。然而,现有的事后解释方法仅考虑这些注意力权重,忽略了来自转换后 token 的关键信息,这未能准确说明模型预测背后的依据。为了将 token 转换的影响纳入解释,我们提出了 TokenTM,这是一种新颖的事后解释方法,利用了我们引入的 token 转换效果度量。具体而言,我们通过测量转换前后 token 长度的变化及其方向上的相关性来量化 token 转换效果。此外,我们开发了初始化和聚合规则,以整合所有层的注意力权重和 token 转换效果,捕获整个模型中整体的 token 贡献。在分割和扰动测试上的实验结果证明了我们提出的 TokenTM 相较于最先进的 Vision Transformer 解释方法的优越性。

英文摘要:

While Transformers have rapidly gained popularity in various computer vision applications, post-hoc explanations of their internal mechanisms remain largely unexplored. Vision Transformers extract visual information by representing image regions as transformed tokens and integrating them via attention weights. However, existing post-hoc explanation methods merely consider these attention weights, neglecting crucial information from the transformed tokens, which fails to accurately illustrate the rationales behind the models' predictions. To incorporate the influence of token transformation into interpretation, we propose TokenTM, a novel post-hoc explanation method that utilizes our introduced measurement of token transformation effects. Specifically, we quantify token transformation effects by measuring changes in token lengths and correlations in their directions pre- and post-transformation. Moreover, we develop initialization and aggregation rules to integrate both attention weights and token transformation effects across all layers, capturing holistic token contributions throughout the model. Experimental results on segmentation and perturbation tests demonstrate the superiority of our proposed TokenTM compared to state-of-the-art Vision Transformer explanation methods.

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

↑