用统计滤波提升视觉Transformer注意力图的可解释性
There is More to Attention: Statistical Filtering Enhances Explanations in Vision Transformers
- 将注意力图与统计滤波结合,去除噪声干扰
- 生成更清晰、更符合人类注视习惯的解释图
- 适合需要可信视觉解释的研究与应用
可解释人工智能(XAI)随着大规模Transformer模型的兴起变得愈发重要,但许多为卷积神经网络(CNN)设计的解释方法在视觉Transformer(ViT)上表现不佳。现有ViT解释方法多依赖注意力权重,其在各层中捕捉的是标记间的交互,常产生噪声较大的解释图。尽管已有方法引入MLP块进行特征归因,但我们认为注意力信号在经过恰当过滤后仍具价值且可解释。本文提出一种新方法,将注意力图与最初用于CNN的统计滤波相结合,以消除噪声或无信息模式,生成更忠实的解释。进一步提出类别特异性变体,获得更具判别力的解释。在多个主流基准上评估表明,该方法优于或相当于是当前最佳方法,生成的解释图更清晰、更易读。除基于扰动的忠实度指标外,还引入人类注视数据评估与人类感知的一致性,强调人类可解释性在XAI中的核心地位。本方法在多个数据集上持续表现优异,兼具高效性与人类合理性。
原文摘要 · Abstract (English)
Explainable AI (XAI) has become increasingly important with the rise of large transformer models, yet many explanation methods designed for CNNs transfer poorly to Vision Transformers (ViTs). Existing ViT explanations often rely on attention weights, which tend to yield noisy maps as they capture token-to-token interactions within each layer.While attribution methods incorporating MLP blocks have been proposed, we argue that attention remains a valuable and interpretable signal when properly filtered. We propose a method that combines attention maps with a statistical filtering, initially proposed for CNNs, to remove noisy or uninformative patterns and produce more faithful explanations. We further extend our approach with a class-specific variant that yields discriminative explanations. Evaluation against popular state-of-the-art methods demonstrates that our approach produces sharper and more interpretable maps. In addition to perturbation-based faithfulness metrics, we incorporate human gaze data to assess alignment with human perception, arguing that human interpretability remains essential for XAI. Across multiple datasets, our approach consistently outperforms or is comparable to the SOTA methods while remaining efficient and human plausible.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。