Abstract:The interpretability of models has become a crucial issue in Machine Learning because of algorithmic decisions' growing impact on real-world applications. Tree ensemble methods, such as Random Forests or XgBoost, are powerful learning tools for classification tasks. However, while combining multiple trees may provide higher prediction quality than a single one, it sacrifices the interpretability property resulting in "black-box" models. In light of this, we aim to develop an interpretable representation of a tree-ensemble model that can provide valuable insights into its behavior. First, given a target tree-ensemble model, we develop a hierarchical visualization tool based on a heatmap representation of the forest's feature use, considering the frequency of a feature and the level at which it is selected as an indicator of importance. Next, we propose a mixed-integer linear programming (MILP) formulation for constructing a single optimal multivariate tree that accurately mimics the target model predictions. The goal is to provide an interpretable surrogate model based on oblique hyperplane splits, which uses only the most relevant features according to the defined forest's importance indicators. The MILP model includes a penalty on feature selection based on their frequency in the forest to further induce sparsity of the splits. The natural formulation has been strengthened to improve the computational performance of {mixed-integer} software. Computational experience is carried out on benchmark datasets from the UCI repository using a state-of-the-art off-the-shelf solver. Results show that the proposed model is effective in yielding a shallow interpretable tree approximating the tree-ensemble decision function.

Born-Again Tree Ensembles

Inherently Interpretable Tree Ensemble Learning

Unboxing Tree Ensembles for interpretability: a hierarchical visualization tool and a multivariate optimal re-built tree

On Computing Optimal Tree Ensembles

LionForests: Local Interpretation of Random Forests

Random Planted Forest: a directly interpretable tree ensemble

Fast Sparse Decision Tree Optimization via Reference Ensembles

Interpreting Tree Ensembles with inTrees

What influences the accuracy of decision tree ensembles?

A generalized decision tree ensemble based on the NeuralNetworks architecture: Distributed Gradient Boosting Forest (DGBF)

Learning accurate and interpretable decision trees

ForestPrune: Compact Depth-Controlled Tree Ensembles

Conceptual views on tree ensemble classifiers

Building more accurate decision trees with the additive tree

Bayesian Regression Trees for High-Dimensional Prediction and Variable Selection

A Unified Approach to Extract Interpretable Rules from Tree Ensembles via Integer Programming

Tree Ensembles with Rule Structured Horseshoe Regularization

Decision Predicate Graphs: Enhancing Interpretability in Tree Ensembles

Dive into Decision Trees and Forests: A Theoretical Demonstration

A Novel Tree-Based Dynamic Heterogeneous Ensemble Method for Credit Scoring.

Free Lunch in the Forest: Functionally-Identical Pruning of Boosted Tree Ensembles