Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

Decision-tree ensembles are a cornerstone of predictive modeling, and SHAP is a standard framework for interpreting their predictions. Among its variants, Background SHAP offers high accuracy by modeling missing features using a background dataset. Historically, this approach did not scale well, as the time complexity for explaining n instances using m background samples included an O(mn) component. Recent methods such as Woodelf and PLTreeSHAP reduce this to O(m+n), but introduce a preprocessing bottleneck that grows as 3^D with tree depth D, making them impractical for deep trees. We address this limitation with WoodelfHD, a Woodelf extension that reduces the 3^D factor to 2^D. The key idea is a Strassen-like multiplication scheme that exploits the structure of Woodelf matrices, reducing matrix-vector multiplication from O(k^2) to O(k*log(k)) via a fully vectorized, non-recursive implementation. In addition, we merge path nodes with identical features, reducing cache size and memory usage. When running on standard environments, WoodelfHD enables exact Background SHAP computation for trees with depths up to 21, where previous methods fail due to excessive memory usage. For ensembles of depths 12 and 15, it achieves speedups of 33x and 162x, respectively, over the state-of-the-art.

Ron Wettenstein, Alexander Nadel, Udi Boker• 2026

Related benchmarks

TaskDatasetResultRank
Background SHAPIEEE-CIS
Runtime3
34
Background SHAP Interaction ValuesIEEE-CIS 10,000 samples
Runtime (sec)2.6
16
Background Shapley values (BG SHAP)Theoretical decision tree ensemble
Space Complexity2
4
Path-Dependent Shapley values (PD SHAP)Theoretical decision tree ensemble
Space Complexity2
4
Background SHAPTheoretical Complexity Analysis
Time Complexity2
3
Background SHAP Interaction ValuesTheoretical Complexity Analysis
Time Complexity (Big O)2
3
Showing 6 of 6 rows

Other info

Follow for update