Constrained Prescriptive Trees via Column Generation
Shivaram Subramanian, Wei Sun, Youssef Drissi, Markus Ettl
Abstract
With the abundance of available data, many enterprises seek to implement data-driven prescriptive analytics to help them make informed decisions. These prescriptive policies need to satisfy operational constraints, and proactively eliminate rule conflicts, both of which are ubiquitous in practice. It is also desirable for them to be simple and interpretable, so they can be easily verified and implemented. Existing approaches from the literature center around constructing variants of prescriptive decision trees to generate interpretable policies. However, none of the existing methods is able to handle constraints. In this paper, we propose a scalable method that solves the constrained prescriptive policy generation problem. We introduce a novel path-based mixed-integer program (MIP) formulation which identifies a (near) optimal policy efficiently via column generation. The policy generated can be represented as a multiway-split tree which is more interpretable and informative than binary-split trees due to its shorter rules. We demonstrate the efficacy of our method with extensive computational experiments on both synthetic and real datasets.
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 7474414f-4d8b-49ef-ae38-bd324e3695f9Cited by top-tier papers2
- Learning Prescriptive ReLU NetworksWei Sun, Asterios TsiourvasICML 2023 · 3 citations
- Necessary and Sufficient Conditions for Optimal Decision Trees using Dynamic ProgrammingJacobus G. M. van der Linden, Mathijs de Weerdt, Emir DemirovicNeurIPS 2023 · 2 citations
Builds on2
- A Scalable MIP-based Method for Learning Optimal Multivariate Decision TreesHaoran Zhu, Pavankumar Murali, Dzung T. Phan, Lam M. Nguyen et al.NeurIPS 2020 · 47 citations
- Model Distillation for Revenue Optimization: Interpretable Personalized PricingMax Biggs, Wei Sun, Markus EttlICML 2021 · 42 citations
Related papers
- Scalable Optimal Multiway-Split Decision Trees with ConstraintsShivaram Subramanian, Wei SunAAAI 2023 · 6 citations
- SPOT: Scalable Policy Optimization with Trees for Markov Decision ProcessesXuyuan Xiong, Pedro Chumpitaz-Flores, Kaixun Hua, Cheng HuaNeurIPS 2025
- Scalable Multi-Action Offline Policy Learning with an m-ary TreeShusei EshimaKDD 2026
- Optimal Counterfactual Explanations in Tree EnsemblesAxel Parmentier, Thibaut VidalICML 2021 · 66 citations
- Interpretable Off-Policy Learning via Hyperbox SearchDaniel Tschernutter, Tobias Hatt, Stefan FeuerriegelICML 2022 · 7 citations
