Vanilla Gradient Descent for Oblique Decision Trees - Archive ouverte HAL
Communication Dans Un Congrès Année : 2024

Vanilla Gradient Descent for Oblique Decision Trees

Vanilla Gradient Descent for Oblique Decision Trees

Résumé

Decision Trees (DTs) constitute one of the major highly non-linear AI models, valued, e.g., for their efficiency on tabular data. Learning accurate DTs is, however, complicated, especially for oblique DTs, and does take a significant training time. Further, DTs suffer from overfitting, e.g., they proverbially "do not generalize" in regression tasks. Recently, some works proposed ways to make (oblique) DTs differentiable. This enables highly efficient gradient-descent algorithms to be used to learn DTs. It also enables generalizing capabilities by learning regressors at the leaves simultaneously with the decisions in the tree. Prior approaches to making DTs differentiable rely either on probabilistic approximations at the tree's internal nodes (soft DTs) or on approximations in gradient computation at the internal node (quantized gradient descent). In this work, we propose DTSemNet, a novel semantically equivalent and invertible encoding for (hard, oblique) DTs as Neural Networks (NNs), that uses standard vanilla gradient descent. Experiments across various classification and regression benchmarks show that oblique DTs learned using DTSemNet are more accurate than oblique DTs of similar size learned using state-ofthe-art techniques. Further, DT training time is significantly reduced. We also experimentally demonstrate that DTSemNet can learn DT policies as efficiently as NN policies in the Reinforcement Learning (RL) setup with physical inputs (dimensions ≤ 32).

Fichier principal
Vignette du fichier
PGES24.pdf (695.9 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04777062 , version 1 (12-11-2024)

Identifiants

Citer

Subrat Prasad Panda, Blaise Genest, Arvind Easwaran, Ponnuthurai Nagaratnam Suganthan. Vanilla Gradient Descent for Oblique Decision Trees. ECAI 2024 - 27TH EUROPEAN CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2024, Saint-Jacques de Compostelle, Spain. ⟨10.3233/FAIA240607⟩. ⟨hal-04777062⟩
9 Consultations
4 Téléchargements

Altmetric

Partager

More