Saved in:
Bibliographic Details
Main Authors: Swain, Swadesh, Singhi, Shree
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.04118
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913635670949888
author Swain, Swadesh
Singhi, Shree
author_facet Swain, Swadesh
Singhi, Shree
contents Integrated Gradients (IG) is a widely used algorithm for attributing the outputs of a deep neural network to its input features. Due to the absence of closed-form integrals for deep learning models, inaccurate Riemann Sum approximations are used to calculate IG. This often introduces undesirable errors in the form of high levels of noise, leading to false insights in the model's decision-making process. We introduce a framework, RiemannOpt, that minimizes these errors by optimizing the sample point selection for the Riemann Sum. Our algorithm is highly versatile and applicable to IG as well as its derivatives like Blur IG and Guided IG. RiemannOpt achieves up to 20% improvement in Insertion Scores. Additionally, it enables its users to curtail computational costs by up to four folds, thereby making it highly functional for constrained environments.
format Preprint
id arxiv_https___arxiv_org_abs_2410_04118
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Riemann Sum Optimization for Accurate Integrated Gradients Computation
Swain, Swadesh
Singhi, Shree
Machine Learning
Artificial Intelligence
Optimization and Control
Integrated Gradients (IG) is a widely used algorithm for attributing the outputs of a deep neural network to its input features. Due to the absence of closed-form integrals for deep learning models, inaccurate Riemann Sum approximations are used to calculate IG. This often introduces undesirable errors in the form of high levels of noise, leading to false insights in the model's decision-making process. We introduce a framework, RiemannOpt, that minimizes these errors by optimizing the sample point selection for the Riemann Sum. Our algorithm is highly versatile and applicable to IG as well as its derivatives like Blur IG and Guided IG. RiemannOpt achieves up to 20% improvement in Insertion Scores. Additionally, it enables its users to curtail computational costs by up to four folds, thereby making it highly functional for constrained environments.
title Riemann Sum Optimization for Accurate Integrated Gradients Computation
topic Machine Learning
Artificial Intelligence
Optimization and Control
url https://arxiv.org/abs/2410.04118