Skip to main navigation Skip to search Skip to main content

Efficient permutation testing of variable importance measures by the example of random forests

  • Technical University of Munich
  • University of Munich

Research output: Contribution to journalArticlepeer-review

29 Scopus citations

Abstract

Hypothesis testing of variable importance measures (VIMPs) is still the subject of ongoing research. This particularly applies to random forests (RF), for which VIMPs are a popular feature. Among recent developments, heuristic approaches to parametric testing have been proposed whose distributional assumptions are based on empirical evidence. Other formal tests under regularity conditions were derived analytically. But these approaches can be computationally expensive or even practically infeasible. This problem also occurs with non-parametric permutation tests, which are, however, distribution-free and can generically be applied to any kind of prediction model and VIMP. Embracing this advantage, it is proposed to use sequential permutation tests and sequential p-value estimation to reduce the computational costs associated with conventional permutation tests. These costs can be particularly high in case of complex prediction models. Therefore, RF's popular and widely used permutation VIMP (pVIMP) serves as a practical and relevant application example. The results of simulation studies confirm the theoretical properties of the sequential tests, that is, the type-I error probability is controlled at a nominal level and a high power is maintained with considerably fewer permutations needed compared to conventional permutation testing. The numerical stability of the methods is investigated in two additional application studies. In summary, theoretically sound sequential permutation testing of VIMP is possible at greatly reduced computational costs. Recommendations for application are given. A respective implementation for RF's pVIMP is provided through the accompanying R package rfvimptest.

Original languageEnglish
Article number107689
JournalComputational Statistics and Data Analysis
Volume181
DOIs
StatePublished - May 2023

Keywords

  • Machine learning
  • Prediction model
  • Sequential permutation test
  • Variable selection
  • p-value

Fingerprint

Dive into the research topics of 'Efficient permutation testing of variable importance measures by the example of random forests'. Together they form a unique fingerprint.

Cite this