Skip to main content

Leveraging Predictive Equivalence in Decision Trees

Publication ,  Conference
McTavish, H; Boner, Z; Donnelly, J; Seltzer, M; Rudin, C
Published in: Proceedings of Machine Learning Research
January 1, 2025

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.

Duke Scholars

Published In

Proceedings of Machine Learning Research

EISSN

2640-3498

Publication Date

January 1, 2025

Volume

267

Start / End Page

43440 / 43475
 

Citation

APA
Chicago
ICMJE
MLA
NLM
McTavish, H., Boner, Z., Donnelly, J., Seltzer, M., & Rudin, C. (2025). Leveraging Predictive Equivalence in Decision Trees. In Proceedings of Machine Learning Research (Vol. 267, pp. 43440–43475).
McTavish, H., Z. Boner, J. Donnelly, M. Seltzer, and C. Rudin. “Leveraging Predictive Equivalence in Decision Trees.” In Proceedings of Machine Learning Research, 267:43440–75, 2025.
McTavish H, Boner Z, Donnelly J, Seltzer M, Rudin C. Leveraging Predictive Equivalence in Decision Trees. In: Proceedings of Machine Learning Research. 2025. p. 43440–75.
McTavish, H., et al. “Leveraging Predictive Equivalence in Decision Trees.” Proceedings of Machine Learning Research, vol. 267, 2025, pp. 43440–75.
McTavish H, Boner Z, Donnelly J, Seltzer M, Rudin C. Leveraging Predictive Equivalence in Decision Trees. Proceedings of Machine Learning Research. 2025. p. 43440–43475.

Published In

Proceedings of Machine Learning Research

EISSN

2640-3498

Publication Date

January 1, 2025

Volume

267

Start / End Page

43440 / 43475