Leveraging Predictive Equivalence in Decision Trees
Hayden McTavish, Zachery Boner, Jon Donnelly, Margo I. Seltzer, Cynthia Rudin
Abstract
Decision trees are widely used for interpretable machine learning due to their clearly structured reasoning process. However, this structure belies a challenge we refer to as predictive equivalence: a given tree's decision boundary can be represented by many different decision trees. The presence of models with identical decision boundaries but different evaluation processes makes model selection challenging. The models will have different variable importance and behave differently in the presence of missing values, but most optimization procedures will arbitrarily choose one such model to return. We present a boolean logical representation of decision trees that does not exhibit predictive equivalence and is faithful to the underlying decision boundary. We apply our representation to several downstream machine learning tasks. Using our representation, we show that decision trees are surprisingly robust to test-time missingness of feature values; we address predictive equivalence's impact on quantifying variable importance; and we present an algorithm to optimize the cost of reaching predictions.
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.
Cited by top-tier papers1
Ask how each one uses itBuilds on15
- Predictive Multiplicity in ClassificationCharles T. Marx, Flávio P. Calmon, Berk UstunICML 2020 · 197 citations
- Generalized and Scalable Optimal Sparse Decision TreesJimmy Lin, Chudi Zhong, Diane Hu, Cynthia Rudin et al.ICML 2020 · 174 citations
- Learning Optimal Decision Trees Using Caching Branch-and-Bound SearchGaël Aglin, Siegfried Nijssen, Pierre SchausAAAI 2020 · 134 citations
- Exploring the Whole Rashomon Set of Sparse Decision TreesRui Xin, Chudi Zhong, Zhi Chen, Takuya Takagi et al.NeurIPS 2022 · 117 citations
- What's a good imputation to predict with missing values?Marine Le Morvan, Julie Josse, Erwan Scornet, Gaël VaroquauxNeurIPS 2021 · 95 citations
Related papers
- Prediction models that learn to avoid missing valuesLena Stempfle, Anton Matsson, Newton Mwai Kinyanjui, Fredrik D. JohanssonICML 2025
- A Comparative Study and Component Analysis of Query Plan Representation Techniques in ML4DB StudiesYue Zhao, Zhaodonghui Li, Gao CongVLDB 2024 · 19 citations
- Towards a Unified Framework for Uncertainty-aware Nonlinear Variable Selection with Theoretical GuaranteesWenying Deng, Beau Coker, Rajarshi Mukherjee, Jeremiah Z. Liu et al.NeurIPS 2022 · 5 citations
- Feature Importance Metrics in the Presence of Missing DataHenrik von Kleist, Joshua Wendland, Ilya Shpitser, Carsten MarrICML 2025
- Feature Learning for Interpretable, Performant Decision TreesJack H. Good, Torin Kovach, Kyle Miller, Artur DubrawskiNeurIPS 2023 · 16 citations
