arXiv:2605.02167cs.LGcs.AI2026-05中稿 · ICML被引 2

让梯度归因更可靠,通过潜空间路径避开无效输入区域。

Manifold-Aligned Guided Integrated Gradients for Reliable Feature Attribution

论文配图:Manifold-Aligned Guided Integrated Gradients for Reliable Feature Attribution
图 1 · 摘自论文原文
  • 在变分自编码器潜空间构建归因路径,贴近数据分布
  • 相比传统方法,显著降低离曼德布罗特区域的噪声干扰
  • 适合需要可信解释的深度模型诊断场景

特征归因对诊断和信任深度神经网络至关重要,集成梯度(IG)因其公理性质被广泛应用。然而,当基线与输入间的积分路径经过梯度噪声区域时,IG可能产生不可靠解释。引导集成梯度虽通过自适应更新低梯度特征减轻敏感性,但输入空间引导仍导致中间输入偏离数据流形。为此,我们提出曼德尔-对齐引导集成梯度(MA-GIG),在预训练变分自编码器的潜空间构建归因路径。通过解码中间潜态,MA-GIG使路径偏向学习到的生成流形,减少对不自然输入区域的暴露。定性和定量评估表明,MA-GIG通过聚焦于接近输入的路径特征来聚合梯度,生成忠实解释。该方法有效降低了离流形噪声,在多个数据集和分类器上优于现有路径基归因方法。代码已公开于 https://github.com/leekwoon/ma-gig/。

原文摘要 · Abstract (English)

Feature attribution is central to diagnosing and trusting deep neural networks, and Integrated Gradients (IG) is widely used due to its axiomatic properties. However, IG can yield unreliable explanations when the integration path between a baseline and the input passes through regions with noisy gradients. While Guided Integrated Gradients reduces this sensitivity by adaptively updating low-gradient-magnitude features, input-space guidance still produces intermediate inputs that deviate from the data manifold. To address this limitation, we propose \emph{Manifold-Aligned Guided Integrated Gradients} (MA-GIG), which constructs attribution paths in the latent space of a pre-trained variational autoencoder. By decoding intermediate latent states, MA-GIG biases the path toward the learned generative manifold and reduces exposure to implausible input-space regions. Through qualitative and quantitative evaluations, we demonstrate that MA-GIG produces faithful explanations by aggregating gradients on path features proximal to the input. Consequently, our method reduces off-manifold noise and outperforms prior path-based attribution methods across multiple datasets and classifiers. Our code is available at https://github.com/leekwoon/ma-gig/.

特征归因可解释性潜空间生成模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。