Abstract
Global optimization of decision trees has shown to be promising in terms of accuracy, size, and consequently human comprehensibility. However, many of the methods used rely on general-purpose solvers for which scalability remains an issue. Dynamic programming methods have been shown to scale much better because they exploit the tree structure by solving subtrees as independent subproblems. However, this only works when an objective can be optimized separately for subtrees. We explore this relationship in detail and show necessary and sufficient conditions for such separability and generalize previous dynamic programming approaches into a framework that can optimize any combination of separable objectives and constraints. Experiments on five application domains show the general applicability of this framework, while outperforming the scalability of general-purpose solvers by a large margin.
Original language | English |
---|---|
Title of host publication | Advances in Neural Information Processing Systems 36 (NeurIPS 2023) |
Subtitle of host publication | 37th Conference on Neural Information Processing Systems (NeurIPS 2023) |
Editors | A. Oh, T. Neumann, A. Globerson, K. Saenko, M. Hardt, S. Levine |
Publisher | Curran Associates, Inc. |
Pages | 9173-9212 |
Number of pages | 40 |
Volume | 36 |
Publication status | Published - 2023 |
Event | 37th Annual Conference on Neural Information Processing Systems - New Orleans, United States Duration: 10 Dec 2023 → 16 Dec 2023 Conference number: 37 |
Conference
Conference | 37th Annual Conference on Neural Information Processing Systems |
---|---|
Abbreviated title | NeurIPS 2023 |
Country/Territory | United States |
City | New Orleans |
Period | 10/12/23 → 16/12/23 |
Keywords
- optimal decision trees
- dynamic programming
- necessary and sufficient conditions
- group fairness
- policy generation
- cost-sensitive classification
- f1-score