Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Guided Integrated Gradients: An Adaptive Path Method for Removing Noise

About

Integrated Gradients (IG) is a commonly used feature attribution method for deep neural networks. While IG has many desirable properties, the method often produces spurious/noisy pixel attributions in regions that are not related to the predicted class when applied to visual models. While this has been previously noted, most existing solutions are aimed at addressing the symptoms by explicitly reducing the noise in the resulting attributions. In this work, we show that one of the causes of the problem is the accumulation of noise along the IG path. To minimize the effect of this source of noise, we propose adapting the attribution path itself -- conditioning the path not just on the image but also on the model being explained. We introduce Adaptive Path Methods (APMs) as a generalization of path methods, and Guided IG as a specific instance of an APM. Empirically, Guided IG creates saliency maps better aligned with the model's prediction and the input image that is being explained. We show through qualitative and quantitative experiments that Guided IG outperforms other, related methods in nearly every experiment.

Andrei Kapishnikov, Subhashini Venugopalan, Besim Avci, Ben Wedin, Michael Terry, Tolga Bolukbasi• 2021

Related benchmarks

TaskDatasetResultRank
Feature Attribution EvaluationImageNet standard (val)
AUC83.8
39
Attribution Quality EvaluationImageNet (val)
SIC AUC0.771
30
Feature Attribution EvaluationOpen Images 5000 random images (val)
AUC0.719
13
Feature Attribution EvaluationDiabetic Retinopathy (DR) 165 sample images (val)
AUC86.3
13
Attribution Quality EvaluationOpen Images (val)
SIC AUC0.866
10
Feature AttributionSynthetic half-moons dataset with Gaussian noises (std dev 0.05-0.65)
AUC-Purity0.361
10
Feature AttributionPascal VOC (test)
AUC-Comp0.2
8
Showing 7 of 7 rows

Other info

Code

Follow for update