arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04563cs.CVcs.AI

注意力引导的CAM:由自注意力引导的视觉Transformer可视化解释

Attention Guided CAM: Visual Explanations of Vision Transformer Guided by Self-Attention

Korea Institute of Science and Technology(韩国科学技术研究院);Sogang University(西江大学)

Saebom Leem, Hyunseok Seo

AI总结:

提出一种注意力引导的ViT可视化方法,通过选择性聚合分类输出到自注意力的梯度并辅以自注意力分数,实现仅用类别标签的高性能弱监督定位与忠实解释。

摘要:

视觉Transformer(ViT)是计算机视觉领域应用最广泛的模型之一,在各种任务上表现出色。为了在各种应用中充分利用基于ViT的架构,需要具有良好定位性能的适当可视化方法,但用于CNN模型的方法由于其独特结构在ViT中尚不可用。在这项工作中,我们提出了一种应用于ViT的注意力引导可视化方法,为其决策提供高层次的语义解释。我们的方法选择性地聚合从分类输出直接传播到每个自注意力的梯度,收集从输入图像每个位置提取的图像特征的贡献。这些梯度还由归一化的自注意力分数(即成对补丁相关性分数)引导,用于补充自注意力机制有效检测到的补丁级上下文信息的梯度。我们的方法仅使用类别标签即可提供精细的高层语义解释和出色的定位性能。因此,我们的方法在弱监督定位任务上优于先前领先的ViT可解释性方法,并在捕获目标类别对象的完整实例方面表现出强大能力。同时,我们的方法提供了忠实解释模型的可视化,这在扰动比较测试中得到了证明。

英文摘要:

Vision Transformer(ViT) is one of the most widely used models in the computer vision field with its great performance on various tasks. In order to fully utilize the ViT-based architecture in various applications, proper visualization methods with a decent localization performance are necessary, but these methods employed in CNN-based models are still not available in ViT due to its unique structure. In this work, we propose an attention-guided visualization method applied to ViT that provides a high-level semantic explanation for its decision. Our method selectively aggregates the gradients directly propagated from the classification output to each self-attention, collecting the contribution of image features extracted from each location of the input image. These gradients are additionally guided by the normalized self-attention scores, which are the pairwise patch correlation scores. They are used to supplement the gradients on the patch-level context information efficiently detected by the self-attention mechanism. This approach of our method provides elaborate high-level semantic explanations with great localization performance only with the class labels. As a result, our method outperforms the previous leading explainability methods of ViT in the weakly-supervised localization task and presents great capability in capturing the full instances of the target class object. Meanwhile, our method provides a visualization that faithfully explains the model, which is demonstrated in the perturbation comparison test.

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

↑