Derivative-Free Diffusion Manifold-Constrained Gradient for Unified XAI

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kim, Won Jun, Chung, Hyungjin, Kim, Jaemin, Lee, Sangmin, Sim, Byeongsu, Ye, Jong Chul
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912494202650624
author Kim, Won Jun
Chung, Hyungjin
Kim, Jaemin
Lee, Sangmin
Sim, Byeongsu
Ye, Jong Chul
author_facet Kim, Won Jun
Chung, Hyungjin
Kim, Jaemin
Lee, Sangmin
Sim, Byeongsu
Ye, Jong Chul
contents Gradient-based methods are a prototypical family of explainability techniques, especially for image-based models. Nonetheless, they have several shortcomings in that they (1) require white-box access to models, (2) are vulnerable to adversarial attacks, and (3) produce attributions that lie off the image manifold, leading to explanations that are not actually faithful to the model and do not align well with human perception. To overcome these challenges, we introduce Derivative-Free Diffusion Manifold-Constrainted Gradients (FreeMCG), a novel method that serves as an improved basis for explainability of a given neural network than the traditional gradient. Specifically, by leveraging ensemble Kalman filters and diffusion models, we derive a derivative-free approximation of the model's gradient projected onto the data manifold, requiring access only to the model's outputs. We demonstrate the effectiveness of FreeMCG by applying it to both counterfactual generation and feature attribution, which have traditionally been treated as distinct tasks. Through comprehensive evaluation on both tasks, counterfactual explanation and feature attribution, we show that our method yields state-of-the-art results while preserving the essential properties expected of XAI tools.
format Preprint
id arxiv_https___arxiv_org_abs_2411_15265
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Derivative-Free Diffusion Manifold-Constrained Gradient for Unified XAI
Kim, Won Jun
Chung, Hyungjin
Kim, Jaemin
Lee, Sangmin
Sim, Byeongsu
Ye, Jong Chul
Computer Vision and Pattern Recognition
Machine Learning
Gradient-based methods are a prototypical family of explainability techniques, especially for image-based models. Nonetheless, they have several shortcomings in that they (1) require white-box access to models, (2) are vulnerable to adversarial attacks, and (3) produce attributions that lie off the image manifold, leading to explanations that are not actually faithful to the model and do not align well with human perception. To overcome these challenges, we introduce Derivative-Free Diffusion Manifold-Constrainted Gradients (FreeMCG), a novel method that serves as an improved basis for explainability of a given neural network than the traditional gradient. Specifically, by leveraging ensemble Kalman filters and diffusion models, we derive a derivative-free approximation of the model's gradient projected onto the data manifold, requiring access only to the model's outputs. We demonstrate the effectiveness of FreeMCG by applying it to both counterfactual generation and feature attribution, which have traditionally been treated as distinct tasks. Through comprehensive evaluation on both tasks, counterfactual explanation and feature attribution, we show that our method yields state-of-the-art results while preserving the essential properties expected of XAI tools.
title Derivative-Free Diffusion Manifold-Constrained Gradient for Unified XAI
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2411.15265