English

Unbiased variable importance for random forests

Machine Learning 2020-05-18 v2 Machine Learning Applications

Abstract

The default variable-importance measure in random Forests, Gini importance, has been shown to suffer from the bias of the underlying Gini-gain splitting criterion. While the alternative permutation importance is generally accepted as a reliable measure of variable importance, it is also computationally demanding and suffers from other shortcomings. We propose a simple solution to the misleading/untrustworthy Gini importance which can be viewed as an overfitting problem: we compute the loss reduction on the out-of-bag instead of the in-bag training samples.

Cite

@article{arxiv.2003.02106,
  title  = {Unbiased variable importance for random forests},
  author = {Markus Loecher},
  journal= {arXiv preprint arXiv:2003.02106},
  year   = {2020}
}