Efficient and Accurate Explanation Estimation with Distribution Compression

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Baniecki, Hubert, Casalicchio, Giuseppe, Bischl, Bernd, Biecek, Przemyslaw
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909464097980416
author Baniecki, Hubert
Casalicchio, Giuseppe
Bischl, Bernd
Biecek, Przemyslaw
author_facet Baniecki, Hubert
Casalicchio, Giuseppe
Bischl, Bernd
Biecek, Przemyslaw
contents We discover a theoretical connection between explanation estimation and distribution compression that significantly improves the approximation of feature attributions, importance, and effects. While the exact computation of various machine learning explanations requires numerous model inferences and becomes impractical, the computational cost of approximation increases with an ever-increasing size of data and model parameters. We show that the standard i.i.d. sampling used in a broad spectrum of algorithms for post-hoc explanation leads to an approximation error worthy of improvement. To this end, we introduce Compress Then Explain (CTE), a new paradigm of sample-efficient explainability. It relies on distribution compression through kernel thinning to obtain a data sample that best approximates its marginal distribution. CTE significantly improves the accuracy and stability of explanation estimation with negligible computational overhead. It often achieves an on-par explanation approximation error 2-3x faster by using fewer samples, i.e. requiring 2-3x fewer model evaluations. CTE is a simple, yet powerful, plug-in for any explanation method that now relies on i.i.d. sampling.
format Preprint
id arxiv_https___arxiv_org_abs_2406_18334
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient and Accurate Explanation Estimation with Distribution Compression
Baniecki, Hubert
Casalicchio, Giuseppe
Bischl, Bernd
Biecek, Przemyslaw
Machine Learning
We discover a theoretical connection between explanation estimation and distribution compression that significantly improves the approximation of feature attributions, importance, and effects. While the exact computation of various machine learning explanations requires numerous model inferences and becomes impractical, the computational cost of approximation increases with an ever-increasing size of data and model parameters. We show that the standard i.i.d. sampling used in a broad spectrum of algorithms for post-hoc explanation leads to an approximation error worthy of improvement. To this end, we introduce Compress Then Explain (CTE), a new paradigm of sample-efficient explainability. It relies on distribution compression through kernel thinning to obtain a data sample that best approximates its marginal distribution. CTE significantly improves the accuracy and stability of explanation estimation with negligible computational overhead. It often achieves an on-par explanation approximation error 2-3x faster by using fewer samples, i.e. requiring 2-3x fewer model evaluations. CTE is a simple, yet powerful, plug-in for any explanation method that now relies on i.i.d. sampling.
title Efficient and Accurate Explanation Estimation with Distribution Compression
topic Machine Learning
url https://arxiv.org/abs/2406.18334