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.

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

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

Optimized Feature Generation for Tabular Data via LLMs with Decision Tree Reasoning

Strong Optimal Classification Trees

Mixed integer linear optimization formulations for learning optimal binary classification trees

Mixed-Integer Linear Optimization for Semi-Supervised Optimal Classification Trees

Optimal Mixed Integer Linear Optimization Trained Multivariate Classification Trees

Machine Learning-Enhanced Ant Colony Optimization for Column Generation

A New Method for Learning Decision Tree Classifier

GradTree: Learning Axis-Aligned Decision Trees with Gradient Descent

Learning optimal decision trees using constraint programming

The Max-Cut Decision Tree: Improving on the Accuracy and Running Time of Decision Trees

Rolling Lookahead Learning for Optimal Classification Trees

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

Learning accurate and interpretable decision trees

Efficient non-greedy optimization of decision trees

Efficient Tree Classifiers for Large Scale Datasets.

Accelerating Column Generation Algorithm Using Machine-Learning-Based Column Elimination

Enhancing Column Generation by a Machine-Learning-Based Pricing Heuristic for Graph Coloring

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

Learning Binary Decision Trees by Argmin Differentiation