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.

A New Method for Learning Decision Tree Classifier

Rough set based decision tree

Decision tree classifier: a detailed survey

Classification Based on Decision Tree Algorithm for Machine Learning

Novel Design of Decision-Tree-Based Support Vector Machines Multi-Class Classifier

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

A Decision Support System Using Two-Level Classifier for Smart Grid

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

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

Dive into Decision Trees and Forests: A Theoretical Demonstration

MTForest: Ensemble Decision Trees based on Multi-Task Learning

A linear multivariate decision tree with branch-and-bound components

Decision tree methods: applications for classification and prediction

dtControl: Decision Tree Learning Algorithms for Controller Representation

Tree in Tree: from Decision Trees to Decision Graphs

Student Academic Performance Prediction by using Decision Tree Algorithm

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

for Further Innproving the Performance of Decision Tree Learning

Learning accurate and interpretable decision trees

Better trees: an empirical study on hyperparameter tuning of classification decision tree induction algorithms