Provably Accurate Shapley Value Estimation via Leverage Score Sampling

arXiv:2410.01917 · cs.LG, cs.AI · Submitted 2024-10-02 · Read on arXiv

cs.LG, cs.AI

Submitted: 2024-10-02

Updated: 2026-10-08

Comments: Accepted to ICLR 2025

Code: https://github.com/rtealwitter/leverageshap

License: http://creativecommons.org/licenses/by/4.0/

The gist: Originally introduced in game theory, Shapley values have emerged as a central tool in explainable machine learning, where they are used to attribute model predictions to specific input features.

Terminology

Abstract

Originally introduced in game theory, Shapley values have emerged as a central tool in explainable machine learning, where they are used to attribute model predictions to specific input features. However, computing Shapley values exactly is expensive: for a general model with n features, O(2 n) model evaluations are necessary. To address this issue, approximation algorithms are widely used. One of the most popular is the Kernel SHAP algorithm, which is model agnostic and remarkably effective in practice. At the time of our original submission, no strong non-asymptotic complexity guarantees were known for Kernel SHAP. We address this issue by introducing Leverage SHAP, a lightweight modification of Kernel SHAP that provides provably accurate Shapley value estimates with just O(n n) model evaluations. Our approach takes advantage of a connection between Shapley value estimation and agnostic active learning by employing leverage score sampling, a powerful regression tool. Beyond theoretical guarantees, we find that Leverage SHAP achieves an approximately 50% reduction in error compared to the highly optimized implementation of Kernel SHAP in the widely used SHAP library [Lundberg & Lee, 2017].

Sources

Related papers