Rashomon Sets of Falling Trees
Varun Babbar, Zachery Boner, Margo Seltzer, Cynthia Rudin
Abstract
Many real-world decisions require prioritizing high-risk cases, such as clinicians prioritizing high-risk patients before lower-risk ones. Falling rule lists (FRLs), which are ordered if--then rules with monotonically decreasing risks, provide an interpretable framework for such tasks; however, their single-path structure yields a highly restricted model class. We introduce falling trees, a new family of interpretable models that enforces the same monotonic risk constraint while permitting tree-structured branching. We present GraviTree, a novel dynamic-programming-with-bounds algorithm for learning the Rashomon set of falling trees under depth and branching constraints, together with bounds that use the falling constraint to provably reduce the search space. Our formulation can interpolate between rule lists and full decision trees, enabling user-desired model expressivity. Across clinical and public-risk datasets, falling trees match or outperform FRLs and other interpretable baselines, often producing lower-sparsity decisions for high-risk instances. Our results show that falling trees strike a practical balance between interpretability, expressiveness, and risk prioritization for high-stakes settings.
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 cb75d8a4-3bf4-46fc-a79a-97ddf3667a1bBuilds on13
- 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
- Fast Sparse Decision Tree Optimization via Reference EnsemblesHayden McTavish, Chudi Zhong, Reto Achermann, Ilias Karimalis et al.AAAI 2022 · 55 citations
- A Path to Simpler Models Starts With NoiseLesia Semenova, Harry Chen, Ronald Parr, Cynthia RudinNeurIPS 2023 · 41 citations
Related papers
- Near-Optimal Decision Trees in a SPLIT SecondVarun Babbar, Hayden McTavish, Cynthia Rudin, Margo I. SeltzerICML 2025
- SORTeD Rashomon Sets of Sparse Decision Trees: Anytime EnumerationElif Arslan, Jacobus G. M. van der Linden, Serge P. Hoogendoorn, Marco Rinaldi et al.NeurIPS 2025 · 8 citations
- Efficient Exploration of the Rashomon Set of Rule-Set ModelsMartino Ciaperoni, Han Xiao, Aristides GionisKDD 2024 · 3 citations
- From Rashomon Theory to PRAXIS: Efficient Decision Tree Rashomon SetsZakk Heile, Hayden McTavish, Varun Babbar, Margo Seltzer et al.ICML 2026 · 1 citation
- Efficient Decision Rule List Learning via Unified Sequence Submodular OptimizationLinxiao Yang, Jingbang Yang, Liang SunKDD 2024
