The influence of dimensions on the complexity of computing decision trees

Stephen Kobourov, Maarten Löffler, Fabrizio Montecchiani*, Marcin Pilipczuk, Ignaz Rutter, Raimund Seidel, Manuel Sorge, Jules Wulms

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

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/log⁡d) 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 languageEnglish
Article number104322
JournalArtificial Intelligence
Volume343
DOIs
Publication statusPublished - 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.

FundersFunder number
National Science FoundationNSF-CCF-2212130
MUR of Italy2022ME9Z78-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

    Fingerprint

    Dive into the research topics of 'The influence of dimensions on the complexity of computing decision trees'. Together they form a unique fingerprint.

    Cite this