A cautionary tale on fitting decision trees to data from additive models: generalization lower bounds
Not on the map. Why?
Abstract (arXiv)
Decision trees are important both as interpretable models amenable to high-stakes decision-making, and as building blocks of ensemble methods such as random forests and gradient boosting. Their statistical properties, however, are not well understood. The most cited prior works have focused on deriving pointwise consistency guarantees for CART in a classical nonparametric regression setting. We take a different approach, and advocate studying the generalization performance of decision trees with respect to different generative regression models. This allows us to elicit their inductive bias, that is, the assumptions the algorithms make (or do not make) to generalize to new data, thereby guiding practitioners on when and how to apply these methods. In this paper, we focus on sparse additive generative models, which have both low statistical complexity and some nonparametric flexibility. We prove a sharp squared error generalization lower bound for a large class of decision tree algorithms fitted to sparse additive models with C1 component functions. This bound is surprisingly much worse than the minimax rate for estimating such sparse additive models. The inefficiency is due not to greediness, but to the loss in power for detecting global structure when we average responses solely over each leaf, an observation that suggests opportunities to improve tree-based algorithms, for example, by hierarchical shrinkage. To prove these bounds, we develop new technical machinery, establishing a novel connection between decision tree estimation and rate-distortion theory, a sub-field of information theory.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 20 of its 57 references that are in the map. First posted on arXiv 18 Oct 2021.
Builds on
20 of 57 references.
References · 20
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
- 2017Towards A Rigorous Science of Interpretable Machine Learningcited by 1,400
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
- 2021Interpretable Machine Learning: Fundamental Principles and 10 Grand Challengescited by 364
- 2019Interpretable machine learning: definitions, methods, and applicationscited by 303
- 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 254
- 2015ranger: A Fast Implementation of Random Forests for High Dimensional Data in C++ and Rcited by 171
- 2015A Random Forest Guided Tourcited by 168
- 2015Interpretable classifiers using rules and Bayesian analysis: Building a better stroke prediction modelcited by 140
- 2020Generalized and Scalable Optimal Sparse Decision Treescited by 98
- 2014Consistency of random forestscited by 84
- 2017Iterative Random Forests to detect predictive and stable high-order interactionscited by 42
8 more
- 2018Local Linear Forestscited by 19
- 2017Data-driven Advice for Applying Machine Learning to Bioinformatics Problemscited by 16
- 2020Trees, forests, and impurity-based variable importancecited by 16
- 2020Estimation and Inference with Trees and Forests in High Dimensionscited by 14
- 2020Sparse learning with CARTcited by 12
- 2021Bridging Breiman's Brook: From Algorithmic Modeling to Statistical Learningcited by 7
- 2017Additive Models with Trend Filteringcited by 6
- 2019Linear Aggregation in Tree-based Estimatorscited by 5
Built on by
Citing papers per year: 2022: 4; 2023: 1; 2024: 1; 2025: 2; 2026: 1.
9 papers.
- 2022From Shapley Values to Generalized Additive Models and backcited by 59
- 2022Hierarchical Shrinkage: improving the accuracy and interpretability of tree-based methodscited by 18
- 2025ProxySPEX: Inference-Efficient Interpretability via Sparse Feature Interactions in LLMscited by 16
- 2022Fast Interpretable Greedy-Tree Sumscited by 12
- 2022An initial alignment between neural network and target is needed for gradient descent to learncited by 12
- 2023On the Convergence of CART under Sufficient Impurity Decrease Conditioncited by 5
- 2025Local MDI+: Local Feature Importances for Tree-Based Modelscited by 3
- 2024Analyze Additive and Interaction Effects via Collaborative Treescited by 0
- 2026Covariance-Driven Regression Trees: Reducing Overfitting in CARTcited by 0
- 2026Covariance-Driven Regression Trees: Reducing Overfitting in CART12 Jan 2026
- 2025Local MDI+: Local Feature Importances for Tree-Based Models10 Jun 2025
- 2025ProxySPEX: Inference-Efficient Interpretability via Sparse Feature Interactions in LLMs23 May 2025
- 2024Analyze Additive and Interaction Effects via Collaborative Trees19 May 2024
- 2023On the Convergence of CART under Sufficient Impurity Decrease Condition26 Oct 2023
- 2022From Shapley Values to Generalized Additive Models and back8 Sep 2022
- 2022An initial alignment between neural network and target is needed for gradient descent to learn25 Feb 2022
- 2022Hierarchical Shrinkage: improving the accuracy and interpretability of tree-based methods2 Feb 2022
- 2022Fast Interpretable Greedy-Tree Sums28 Jan 2022