arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.01482stat.MLcs.AIcs.LG
收藏

一个波解释一切:特征归因的统一视角

One Wave To Explain Them All: A Unifying Perspective On Feature Attribution

Mines Paris - PSL University(巴黎矿业学院 - 巴黎文理研究大学);RTE France(法国电网公司)

Gabriel Kasmi, Amandine Brunetto, Thomas Fel, Jayneel Parekh

AI总结:

本文提出小波归因方法WAM,利用小波域的空间与尺度局部化特性统一处理任意维度输入的特征归因,在音频、图像和体数据上匹配或优于现有基于梯度的方法,并连接模型鲁棒性与透明度。

摘要:

特征归因方法旨在通过识别影响模型决策的输入特征来提高深度神经网络的透明度。基于像素的热图已成为将特征归因于高维输入(如图像、音频表示和体数据)的标准方法。虽然直观且方便,但这些基于像素的归因无法捕捉数据的底层结构。此外,计算归因时域的选择常常被忽视。本工作表明,小波域能够提供信息丰富且有意义的归因。它可以处理任意输入维度,并提供了一种统一的特征归因方法。我们的方法,即小波归因方法(Wavelet Attribution Method,WAM),利用小波系数的空间和尺度局部化特性,提供能够同时捕捉模型决策过程中“在哪里”和“是什么”的解释。我们展示了WAM在包括音频、图像和体数据在内的多种模态上,在定量上匹配或优于现有的基于梯度的方法。此外,我们还讨论了WAM如何将归因与模型鲁棒性和透明度的更广泛方面联系起来。项目页面:https://gabrielkasmi.github.io/wam/

英文摘要:

Feature attribution methods aim to improve the transparency of deep neural networks by identifying the input features that influence a model's decision. Pixel-based heatmaps have become the standard for attributing features to high-dimensional inputs, such as images, audio representations, and volumes. While intuitive and convenient, these pixel-based attributions fail to capture the underlying structure of the data. Moreover, the choice of domain for computing attributions has often been overlooked. This work demonstrates that the wavelet domain allows for informative and meaningful attributions. It handles any input dimension and offers a unified approach to feature attribution. Our method, the Wavelet Attribution Method (WAM), leverages the spatial and scale-localized properties of wavelet coefficients to provide explanations that capture both the where and what of a model's decision-making process. We show that WAM quantitatively matches or outperforms existing gradient-based methods across multiple modalities, including audio, images, and volumes. Additionally, we discuss how WAM bridges attribution with broader aspects of model robustness and transparency. Project page: https://gabrielkasmi.github.io/wam/

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

↑