Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SORTeD Rashomon Sets of Sparse Decision Trees: Anytime Enumeration

About

Sparse decision tree learning provides accurate and interpretable predictive models that are ideal for high-stakes applications by finding the single most accurate tree within a (soft) size limit. Rather than relying on a single "best" tree, Rashomon sets-trees with similar performance but varying structures-can be used to enhance variable importance analysis, enrich explanations, and enable users to choose simpler trees or those that satisfy stakeholder preferences (e.g., fairness) without hard-coding such criteria into the objective function. However, because finding the optimal tree is NP-hard, enumerating the Rashomon set is inherently challenging. Therefore, we introduce SORTD, a novel framework that improves scalability and enumerates trees in the Rashomon set in order of the objective value, thus offering anytime behavior. Our experiments show that SORTD reduces runtime by up to two orders of magnitude compared with the state of the art. Moreover, SORTD can compute Rashomon sets for any separable and totally ordered objective and supports post-evaluating the set using other separable (and partially ordered) objectives. Together, these advances make exploring Rashomon sets more practical in real-world applications.

Elif Arslan, Jacobus G. M. van der Linden, Serge Hoogendoorn, Marco Rinaldi, Emir Demirovi\'c• 2025

Related benchmarks

TaskDatasetResultRank
Decision Tree Rashomon Set Calculationhelena
Runtime100.3
11
Decision Tree Rashomon Set CalculationCovertype
Runtime6.47e+4
7
Decision Tree Rashomon Set constructionjasmine
Runtime (s)5.21e+3
7
Decision Tree Rashomon Set constructionDiabetes
Runtime (s)683.3
7
Decision Tree Rashomon Set constructionmagic
Runtime (s)268.7
6
Decision Tree Rashomon Set CalculationCOMPAS
Runtime7.23
6
Decision Tree Rashomon Set constructionBank
Runtime (s)1.81e+3
6
Decision Tree Rashomon Set constructionShopping
Runtime (s)2.42e+4
6
Decision Tree Rashomon Set constructiontaxi
Runtime (s)364.9
6
Decision Tree Rashomon Set constructionChurn
Construction Time (s)1.24e+5
6
Showing 10 of 48 rows

Other info

Follow for update