Joint Distribution-Informed Shapley Values for Sparse Counterfactual Explanations

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: You, Lei, Bian, Yijun, Cao, Lele
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911473105633280
author You, Lei
Bian, Yijun
Cao, Lele
author_facet You, Lei
Bian, Yijun
Cao, Lele
contents Counterfactual explanations (CE) aim to reveal how small input changes flip a model's prediction, yet many methods modify more features than necessary, reducing clarity and actionability. We introduce \emph{COLA}, a model- and generator-agnostic post-hoc framework that refines any given CE by computing a coupling via optimal transport (OT) between factual and counterfactual sets and using it to drive a Shapley-based attribution (\emph{$p$-SHAP}) that selects a minimal set of edits while preserving the target effect. Theoretically, OT minimizes an upper bound on the $W_1$ divergence between factual and counterfactual outcomes and that, under mild conditions, refined counterfactuals are guaranteed not to move farther from the factuals than the originals. Empirically, across four datasets, twelve models, and five CE generators, COLA achieves the same target effects with only 26--45\% of the original feature edits. On a small-scale benchmark, COLA shows near-optimality.
format Preprint
id arxiv_https___arxiv_org_abs_2410_05419
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Joint Distribution-Informed Shapley Values for Sparse Counterfactual Explanations
You, Lei
Bian, Yijun
Cao, Lele
Machine Learning
Artificial Intelligence
Methodology
Counterfactual explanations (CE) aim to reveal how small input changes flip a model's prediction, yet many methods modify more features than necessary, reducing clarity and actionability. We introduce \emph{COLA}, a model- and generator-agnostic post-hoc framework that refines any given CE by computing a coupling via optimal transport (OT) between factual and counterfactual sets and using it to drive a Shapley-based attribution (\emph{$p$-SHAP}) that selects a minimal set of edits while preserving the target effect. Theoretically, OT minimizes an upper bound on the $W_1$ divergence between factual and counterfactual outcomes and that, under mild conditions, refined counterfactuals are guaranteed not to move farther from the factuals than the originals. Empirically, across four datasets, twelve models, and five CE generators, COLA achieves the same target effects with only 26--45\% of the original feature edits. On a small-scale benchmark, COLA shows near-optimality.
title Joint Distribution-Informed Shapley Values for Sparse Counterfactual Explanations
topic Machine Learning
Artificial Intelligence
Methodology
url https://arxiv.org/abs/2410.05419