MDA for random forests: inconsistency, and a practical solution via the Sobol-MDA

02/26/2021
by   Clément Bénard, et al.
0

Variable importance measures are the main tools to analyze the black-box mechanism of random forests. Although the Mean Decrease Accuracy (MDA) is widely accepted as the most efficient variable importance measure for random forests, little is known about its theoretical properties. In fact, the exact MDA definition varies across the main random forest software. In this article, our objective is to rigorously analyze the behavior of the main MDA implementations. Consequently, we mathematically formalize the various implemented MDA algorithms, and then establish their limits when the sample size increases. In particular, we break down these limits in three components: the first two are related to Sobol indices, which are well-defined measures of a variable contribution to the output variance, widely used in the sensitivity analysis field, as opposed to the third term, whose value increases with dependence within input variables. Thus, we theoretically demonstrate that the MDA does not target the right quantity when inputs are dependent, a fact that has already been noticed experimentally. To address this issue, we define a new importance measure for random forests, the Sobol-MDA, which fixes the flaws of the original MDA. We prove the consistency of the Sobol-MDA and show its good empirical performance through experiments on both simulated and real data. An open source implementation in R and C++ is available online.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
01/13/2020

Trees, forests, and impurity-based variable importance

Tree ensemble methods such as random forests [Breiman, 2001] are very po...
research
05/25/2021

SHAFF: Fast and consistent SHApley eFfect estimates via random Forests

Interpretability of learning algorithms is crucial for applications invo...
research
08/07/2023

Variable importance for causal forests: breaking down the heterogeneity of treatment effects

Causal random forests provide efficient estimates of heterogeneous treat...
research
07/28/2014

Understanding Random Forests: From Theory to Practice

Data analysis and machine learning have become an integrative part of th...
research
12/06/2022

The Importance of Variable Importance

Variable importance is defined as a measure of each regressor's contribu...
research
03/04/2020

Unbiased variable importance for random forests

The default variable-importance measure in random Forests, Gini importan...
research
11/11/2019

Simplifying Random Forests: On the Trade-off between Interpretability and Accuracy

We analyze the trade-off between model complexity and accuracy for rando...

Please sign up or login with your details

Forgot password? Click here to reset