Abstract:Decision Trees (DTs) are a class of supervised learning models that are widely used for both classification and regression applications. They are well-known for their interpretability and robustness, which have led them to remain popular even 60 years after they were first proposed. However, because traditional tree algorithms use greedy methods that are prone to suboptimality, several works have explored the usage of evolutionary algorithms instead. Although these algorithms are often reported to outperform the traditional greedy approach, their computational cost is much higher, since the evolutionary component requires a large number (millions or billions) of function evaluations in order to produce a single tree. Aiming to reduce this computational cost, in this work we propose an encoding that allows the training and evaluation of DTs using only matrix operations. The proposed procedure is shown to be much faster than the traditional tree implementation for complete trees with depths ranging from 2 to 6, and for datasets ranging in size from 100 to 100,000 observations. In particular, the results show speedups of nearly up to 20 times, especially when the dataset is large and the desired tree is small enough to be interpretable. The proposed procedure also benefits from GPU parallelization, although it is still highly performing without it. Furthermore, we propose an evolutionary algorithm, called Coral Reef Optimization for Decision Trees (CRO-DT), that integrates this encoding with a pre-existing ensemble algorithm to evolve better univariate trees. The results obtained show that the proposed CRO-DT is competitive with traditional and modern tree algorithms, consistently producing models of good quality across 14 tested UCI Datasets. We conclude that for most relevant situations, the proposed matrix encoding provides significant speedups over the traditional implementation, and also may serve as a basis for high quality evolutionary DT algorithms.

GradTree: Learning Axis-Aligned Decision Trees with Gradient Descent

Vanilla Gradient Descent for Oblique Decision Trees

Learning Binary Decision Trees by Argmin Differentiation

Stochastic Gradient Trees

End-to-end Learning of Deterministic Decision Trees

Deep Trees for (Un)structured Data: Tractability, Performance, and Interpretability

Efficient evolution of decision trees via fully matrix-based fitness evaluation

Branches: A Fast Dynamic Programming and Branch & Bound Algorithm for Optimal Decision Trees

Efficient non-greedy optimization of decision trees

An improved column-generation-based matheuristic for learning classification trees

Decision Stream: Cultivating Deep Decision Trees

Multi-Layered Gradient Boosting Decision Trees.

A Reinforcement Learning Approach to Online Learning of Decision Trees

MurTree: Optimal Decision Trees Via Dynamic Programming and Search.

Learning decision trees through Monte Carlo tree search: An empirical evaluation

A New Method for Learning Decision Tree Classifier

One-Stage Tree: End-to-end Tree Builder and Pruner.

Poster: gbdt-rs: Fast and Trustworthy Gradient Boosting Decision Tree

Fair Adversarial Gradient Tree Boosting

Learning accurate and interpretable decision trees