Abstract
A decision tree recursively splits a feature space Rd and then assigns class labels based on the resulting partition. Decision trees have been part of the basic machine-learning toolkit for decades. A large body of work considers heuristic algorithms that compute a decision tree from training data, usually aiming to minimize in particular the size of the resulting tree. In contrast, little is known about the complexity of the underlying computational problem of computing a minimum-size tree for the given training data. We study this problem with respect to the number d of dimensions of the feature space Rd, which contains n training examples. We show that it can be solved in O(n2d+1) time, but under reasonable complexity-theoretic assumptions it is not possible to achieve f(d)⋅no(d/logd) running time. The problem is solvable in (dR)O(dR)⋅n1+o(1) time if there are exactly two classes and R is an upper bound on the number of tree leaves labeled with the first class.
Original language | English |
---|---|
Article number | 104322 |
Journal | Artificial Intelligence |
Volume | 343 |
DOIs | |
Publication status | Published - Jun 2025 |
Bibliographical note
Publisher Copyright:© 2025 The Author(s)
Funding
Main ideas for the results of this paper were developed in the relaxed atmosphere of Dagstuhl Seminar 21062 on Parameterized Complexity in Graph Drawing, organized by Robert Ganian, Fabrizio Montecchiani, Martin Noellenburg, and Meirav Zehavi. An extended abstract of this work appeared at AAAI 2023 [18] . In comparison, this version contains full proof details and more discussion. Stephen Kobourov acknowledges funding by the National Science Foundation, grant number NSF-CCF-2212130. Fabrizio Montecchiani acknowledges funding by MUR of Italy, under PRIN Project n. 2022ME9Z78-NextGRAAL: Next-generation algorithms for constrained GRAph visuALization, and by University of Perugia, Fondi di Ricerca di Ateneo, edizione 2022, project MiRA: Mixed Reality and AI Methodologies for Immersive Robotics. Manuel Sorge acknowledges funding by the Alexander von Humboldt Foundation. Jules Wulms acknowledges funding by the Vienna Science and Technology Fund (WWTF) under grant ICT19-035.
Funders | Funder number |
---|---|
National Science Foundation | NSF-CCF-2212130 |
MUR of Italy | 2022ME9Z78-NextGRAAL |
University of Perugia, Fondi di Ricerca di Ateneo, edizione 2022, project MiRA: Mixed Reality and AI Methodologies for Immersive Robotics | |
Alexander von Humboldt Foundation | |
Vienna Science and Technology Fund (WWTF) | ICT19-035 |
Keywords
- Decision trees
- Machine learning
- Parameterized complexity