Reto Achermann
Assistant Professor
Systems Research Group
TUM School of Computation, Information and Technology
Fast Sparse Decision Tree Optimization via Reference Ensembles
Authors
Hayden McTavish, Chudi Zhong, Reto Achermann, Ilias Karimalis, Jacques Chen, Cynthia Rudin and Margo Seltzer
Venue
Proceedings of the Thirty-Sixth AAAI Conference on Artificial Intelligence (AAAI'22)
Links
Abstract
Sparse decision tree optimization has been one of the most fundamental problems in AI since its inception and is a challenge at the core of interpretable machine learning. Sparse decision tree optimization is computationally hard, and despite steady effort since the 1960's, breakthroughs have been made on the problem only within the past few years, primarily on the problem of finding optimal sparse decision trees. However, current state-of-the-art algorithms often require impractical amounts of computation time and memory to find optimal or near-optimal trees for some real-world datasets, particularly those having several continuous-valued features. Given that the search spaces of these decision tree optimization problems are massive, can we practically hope to find a sparse decision tree that competes in accuracy with a black box machine learning model? We address this problem via smart guessing strategies that can be applied to any optimal branch-and-bound-based decision tree algorithm. The guesses come from knowledge gleaned from black box models. We show that by using these guesses, we can reduce the run time by multiple orders of magnitude while providing bounds on how far the resulting trees can deviate from the black box's accuracy and expressive power. Our approach enables guesses about how to bin continuous features, the size of the tree, and lower bounds on the error for the optimal decision tree. Our experiments show that in many cases we can rapidly construct sparse decision trees that match the accuracy of black box models. To summarize: when you are having trouble optimizing, just guess.
Bibtex
@inproceedings{McTavish:2022:FSD,
author = {McTavish, Hayden and Zhong, Chudi and Achermann, Reto and Karimalis, Ilias and Chen, Jacques and Rudin, Cynthia and Seltzer, Margo},
booktitle = {Proceedings of the Thirty-Sixth AAAI Conference on Artificial Intelligence},
doi = {10.1609/aaai.v36i9.21194},
id = {McTavish:2022:FSD},
location = {Virtual},
publisher = {Association for the Advancement of Artificial Intelligence},
series = {AAAI'22},
title = {Fast Sparse Decision Tree Optimization via Reference Ensembles},
url = {https://doi.org/10.1609/aaai.v36i9.21194},
year = {2022}
}
Prof. Reto Achermann
I01: Chair of Distributed Systems and Operating Systems (aka Systems Research Group)
1st Floor, 7th Finger
School of Computation, Information, and Technology (CIT)
Technical University of Munich (TUM)
Boltzmannstr. 3
85748 Garching bei München
Germany
firstname.lastname [at] cit.tum.de


