Pith. sign in

REVIEW 6 cited by

Fast TreeSHAP: Accelerating SHAP Value Computation for Trees

Not yet reviewed by Pith; the record is open.

This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.

SPECIMEN: schema-true, not a live event

T0 review · schema-true

One-sentence machine reading of the paper's core claim.

pith:XXXXXXXX · record.json · timestamp

arxiv 2109.09847 v3 pith:XF33626M submitted 2021-09-20 cs.LG stat.ML

classification cs.LGstat.ML
keywords treeshapfastfastershaptimecomputationcostdatasets
verification ladder T0 review T1 audit T2 compute T3 formal
0 comments
read the original abstract

SHAP (SHapley Additive exPlanation) values are one of the leading tools for interpreting machine learning models, with strong theoretical guarantees (consistency, local accuracy) and a wide availability of implementations and use cases. Even though computing SHAP values takes exponential time in general, TreeSHAP takes polynomial time on tree-based models. While the speedup is significant, TreeSHAP can still dominate the computation time of industry-level machine learning solutions on datasets with millions or more entries, causing delays in post-hoc model diagnosis and interpretation service. In this paper we present two new algorithms, Fast TreeSHAP v1 and v2, designed to improve the computational efficiency of TreeSHAP for large datasets. We empirically find that Fast TreeSHAP v1 is 1.5x faster than TreeSHAP while keeping the memory cost unchanged. Similarly, Fast TreeSHAP v2 is 2.5x faster than TreeSHAP, at the cost of a slightly higher memory usage, thanks to the pre-computation of expensive TreeSHAP steps. We also show that Fast TreeSHAP v2 is well-suited for multi-time model interpretations, resulting in as high as 3x faster explanation of newly incoming samples.

Discussion (0). Sign in to comment.

Forward citations

Cited by 6 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Budget-Efficient Automatic Algorithm Design via Code Graph

    cs.AI 2026-05 unverdicted novelty 7.0 of 10

    A code-graph and correction-based LLM search framework outperforms full-algorithm generation at equal token budgets on three combinatorial optimization problems.

  2. QuadraSHAP: Stable and Scalable Shapley Values for Product Games via Gauss-Legendre Quadrature

    cs.LG 2026-05 unverdicted novelty 7.0 of 10

    Shapley values in product games equal an exact one-dimensional integral of a polynomial, computable via Gauss-Legendre quadrature with linear cost in the number of features.

  3. QuadraSHAP: Stable and Scalable Shapley Values for Product Games via Gauss-Legendre Quadrature

    cs.LG 2026-05 conditional novelty 7.0 of 10

    Shapley values in product games equal the integral of a degree-(d-1) polynomial over [0,1], allowing provably exact or near-exact computation via Gauss-Legendre quadrature with O(d m_q) work.

  4. WOODELF-HD: Efficient Background SHAP for High-Depth Decision Trees

    cs.LG 2026-04 conditional novelty 7.0 of 10

    WoodelfHD reduces Background SHAP preprocessing for decision trees from 3^D to 2^D complexity, enabling exact computation on depths up to 21 with reported speedups of 33x to 162x.

  5. Rethinking XAI Evaluation: A Human-Centered Audit of Shapley Benchmarks in High-Stakes Settings

    cs.LG 2026-04 unverdicted novelty 6.0 of 10

    In high-stakes settings, Shapley explanations increase analyst confidence but do not improve decision accuracy, and standard metrics fail to predict human utility.

  6. From Decision Trees to Boolean Logic: A Fast and Unified SHAP Algorithm

    cs.LG 2025-11 unverdicted novelty 6.0 of 10

    WOODELF computes Background SHAP for tree ensembles in linear time via pseudo-Boolean formulas that encode trees, features, and background data, with reported speedups of 16x on CPU and 165x on GPU for million-row datasets.

Pith tools