Manifold-Aligned Guided Integrated Gradients for Reliable Feature Attribution

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kim, Soyeon, Lim, Seongwoo, Lee, Kyowoon, Choi, Jaesik
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914574091943936
author Kim, Soyeon
Lim, Seongwoo
Lee, Kyowoon
Choi, Jaesik
author_facet Kim, Soyeon
Lim, Seongwoo
Lee, Kyowoon
Choi, Jaesik
contents 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/.
format Preprint
id arxiv_https___arxiv_org_abs_2605_02167
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Manifold-Aligned Guided Integrated Gradients for Reliable Feature Attribution
Kim, Soyeon
Lim, Seongwoo
Lee, Kyowoon
Choi, Jaesik
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
68T07, 68T05, 68T45
I.2.6; I.5.2; I.2.10
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/.
title Manifold-Aligned Guided Integrated Gradients for Reliable Feature Attribution
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
68T07, 68T05, 68T45
I.2.6; I.5.2; I.2.10
url https://arxiv.org/abs/2605.02167