Surrogate minimal depth as an importance measure for variables in random forests

Stephan Seifert*, Sven Gundlach, Silke Szymczak

*Corresponding author for this work
4 Citations (Scopus)

Abstract

Motivation: It has been shown that the machine learning approach random forest can be successfully applied to omics data, such as gene expression data, for classification or regression and to select variables that are important for prediction. However, the complex relationships between predictor variables, in particular between causal predictor variables, make the interpretation of currently applied variable selection techniques difficult. Results: Here we propose a new variable selection approach called surrogate minimal depth (SMD) that incorporates surrogate variables into the concept of minimal depth (MD) variable importance. Applying SMD, we show that simulated correlation patterns can be reconstructed and that the increased consideration of variable relationships improves variable selection. When compared with existing state-of-the-art methods and MD, SMD has higher empirical power to identify causal variables while the resulting variable lists are equally stable. In conclusion, SMD is a promising approach to get more insight into the complex interplay of predictor variables and outcome in a high-dimensional data setting. Availability and implementation: https://github.com/StephanSeifert/SurrogateMinimalDepth. Supplementary information: Supplementary data are available at Bioinformatics online.

Original languageEnglish
JournalBioinformatics
Volume35
Issue number19
Pages (from-to)3663-3671
Number of pages9
ISSN1367-4803
DOIs
Publication statusPublished - 01.10.2019

Fingerprint

Dive into the research topics of 'Surrogate minimal depth as an importance measure for variables in random forests'. Together they form a unique fingerprint.

Cite this