Paul, Jérôme
[UCL]
Dupont, Pierre
[UCL]
In the original Random Forest (RF) approach, Breiman proposes an embedded feature importance index. It is proportional to the decrease in tree accuracy, estimated on the out-of-bag (OOB) samples, when permuting a particular variable. Such multivariate index takes into account the interactions between variables but is not straightforward to interpret in a statistical sense. In particular, it is hard to decide which variables are statistically significant and, specifically, to assign a p-value to such a decision. We proposed a statistical procedure to measure variable importance, that tests if variables are significantly useful in combination with others in a forest in (Paul et al. 2013). The importance J_χ² of a variable is defined as the p-value, corrected for multiple testing, that the tree class vote distribution changes when permuting the feature. Those changes are estimated on the OOB samples and assessed by a Pearson's chi squared test. The outputted p-values offer a natural threshold to decide which features are statistically significant. Experiments conducted on synthetic and real high-dimensional datasets show that J_χ² correctly identifies relevant variables provided a large number of trees (typically 10,000). The feature ranking is also largely correlated with Breiman's index. Further analyses (Paul and Dupont, 2014) compare J_χ² to two alternative procedures proposed in (Huynh-Thu et al. 2012): 1Probe and mr-Test. They also allow to convert Breiman's importance into p-values. However, they are conceptually and computationally more complex than J_χ². Indeed, they resort on a resampling process that repeatedly builds RFs in order to estimate a null importance distribution from which p-values can be computed. Since J_χ² is directly estimated on the OOB samples with no need of additional resamplings, it appears that it requires an order of magnitude less trees than the two other approaches to yield similar sets of selected variables. One potential drawback of J_χ² is that the assumption of independence of the J_χ² test may become strongly violated when growing forest with exceedingly many trees while the number of independent samples in the dataset is left unchanged. One can address this potential issue by considering a Kolmogorov-Smirnov test while providing similar results to J_χ². We also evaluate here alternative statistical procedures based on the tree accuracy distributions with and without permuting variables. To sum up, we study several RF feature importance indices with the objective of relating them to well defined statistical tests. Such relation offers a statistical interpretation of those indices after translating them into p-values. It also provides a natural threshold to highlight relevant variables. Practical experiments, both on artificial and real data from DNA microarrays, show that one is able to retrieve important variables while drastically reducing the computational cost of recently proposed alternatives.
Bibliographic reference |
Paul, Jérôme ; Dupont, Pierre. Statistically interpretable importance indices for Random Forests.23rd Annual Machine Learning Conference of Belgium and the Netherlands (BENELEARN) (Brussels, Belgium, 06/06/2014). |
Permanent URL |
http://hdl.handle.net/2078.1/147350 |