Sensitivity Verification for Additive Decision Tree Ensembles
Arhaan Ahmad, Tanay Vineet Tayal, Ashutosh Gupta, S. Akshay
Abstract
Tree ensemble models, such as Gradient Boosted Decision Trees (GBDTs) and random forests, are widely popular models for a variety of machine learning tasks. The power of these models comes from the ensemble of decision trees, which makes analysis of such models significantly harder than for single trees. As a result, recent work has focused on developing exact and approximate techniques for questions such as robustness verification, fairness and explainability for such models of tree ensembles. In this paper, we focus on a specific problem of feature sensitivity for additive decision tree ensembles and build a formal verification framework for it, where we also take into account the confidence of the tree ensemble in its output. We start by showing theoretical (NP-)hardness of the problem and explain how it relates to other verification problems. Next, we provide a novel encoding of the problem using pseudo-Boolean constraints. Based on this encoding, we develop a tunable algorithm to perform sensitivity analysis, which can trade off precision for running time. We implement our algorithm and study its performance on a suite of GBDT benchmarks from the literature. Our experiments show the practical utility of our approach and its improved performance compared to existing approaches.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 6dea555a-3b7d-4ed4-af44-4244bd79f96fCited by top-tier papers2
- Data-Aware and Scalable Sensitivity Analysis for Decision Tree EnsemblesNamrita Varshney, Ashutosh Gupta, Arhaan Ahmad, Tanay Vineet Tayal et al.ICLR 2026 · 2 citations
- Quantifying Sensitivity for Tree Ensembles: A Symbolic and Compositional ApproachAjinkya Naik, Chaitanya Garg, S. Akshay, Ashutosh Gupta et al.CAV 2026
Builds on3
- General Cutting Planes for Bound-Propagation-Based Neural Network VerificationHuan Zhang, Shiqi Wang, Kaidi Xu, Linyi Li et al.NeurIPS 2022 · 154 citations
- Globally-Robust Neural NetworksKlas Leino, Zifan Wang, Matt FredriksonICML 2021 · 150 citations
- Versatile Verification of Tree EnsemblesLaurens Devos, Wannes Meert, Jesse DavisICML 2021 · 16 citations
Related papers
- OC-space: a Unifying Perspective on Verification of Tree EnsemblesTimo Martens, Laurens Devos, Lorenzo Cascioli, Wannes Meert et al.ICML 2026
- Abstract Interpretation of Decision Tree Ensemble ClassifiersFrancesco Ranzato, Marco ZanellaAAAI 2020 · 50 citations
- Verifiable Boosted Tree EnsemblesStefano Calzavara, Lorenzo Cazzaro, Claudio Lucchese, Giulio Ermanno PibiriS&P 2025
- On Lp-norm Robustness of Ensemble Decision Stumps and TreesYihan Wang, Huan Zhang, Hongge Chen, Duane S. Boning et al.ICML 2020 · 11 citations
- Subgroup Robustness Grows On Trees: An Empirical Baseline InvestigationJosh Gardner, Zoran Popovic, Ludwig SchmidtNeurIPS 2022 · 27 citations
