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

驾驶动作预测的无噪声解释方法

Noise-Free Explanation for Driving Action Prediction

University of Manchester(曼彻斯特大学);Imperial College London(帝国理工学院)

Hongbo Zhu, Theodor Wulff, Rahul Singh Maharjan, Jinpei Han, Angelo Cangelosi

AI总结:

针对现有注意力可解释方法存在片面性、易产生虚假噪声归因的问题,提出SNNA方法,通过值向量范数加权注意力、梯度引导标签信号及扰动平均梯度生成无噪声归因,在驾驶动作预测多标签任务上的定性定量评估均优于现有SOTA方法。

摘要:

尽管注意力机制在各类人工智能(AI)领域的基于Transformer的架构中已取得长足进展,但其内部工作机制仍有待探究。现有可解释方法各有侧重但较为片面,主要分析注意力机制或基于梯度的归因,却忽略了输入特征值的大小或跳跃连接模块。此外,它们不可避免地会产生与模型决策无关的虚假噪声像素归因,妨碍人类对可视化结果的信任。为此,我们提出了一种易于实现且有效的方案来弥补这一缺陷:Smooth Noise Norm Attention(SNNA,平滑噪声范数注意力)。我们通过变换后值向量的范数对注意力进行加权,并利用注意力梯度引导特定标签的信号,随后随机采样输入扰动并对相应梯度取平均,以生成无噪声的归因结果。与以往研究在二分类或多分类任务上评估解释方法不同,本文探索了更复杂的多标签分类场景,即驾驶动作预测任务,并专门为此训练了一个模型。定性与定量评估结果均表明,与其他SOTA(state-of-the-art,当前最优)的基于注意力的可解释方法相比,SNNA在生成更清晰的视觉解释图和对输入像素重要性排序方面更具优势。

英文摘要:

Although attention mechanisms have achieved considerable progress in Transformer-based architectures across various Artificial Intelligence (AI) domains, their inner workings remain to be explored. Existing explainable methods have different emphases but are rather one-sided. They primarily analyse the attention mechanisms or gradient-based attribution while neglecting the magnitudes of input feature values or the skip-connection module. Moreover, they inevitably bring spurious noisy pixel attributions unrelated to the model's decision, hindering humans' trust in the spotted visualization result. Hence, we propose an easy-to-implement but effective way to remedy this flaw: Smooth Noise Norm Attention (SNNA). We weigh the attention by the norm of the transformed value vector and guide the label-specific signal with the attention gradient, then randomly sample the input perturbations and average the corresponding gradients to produce noise-free attribution. Instead of evaluating the explanation method on the binary or multi-class classification tasks like in previous works, we explore the more complex multi-label classification scenario in this work, i.e., the driving action prediction task, and trained a model for it specifically. Both qualitative and quantitative evaluation results show the superiority of SNNA compared to other SOTA attention-based explainable methods in generating a clearer visual explanation map and ranking the input pixel importance.

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

↑