Matched Forest: supervised learning for high-dimensional matched case–control studies

Abstract

Abstract Motivation Matched case–control analysis is widely used in biomedical studies to identify exposure variables associated with health conditions. The matching is used to improve the efficiency. Existing variable selection methods for matched case–control studies are challenged in high-dimensional settings where interactions among variables are also important. We describe a quite different method for high-dimensional matched case–control data, based on the potential outcome model, which is not only flexible regarding the number of matching and exposure variables but also able to detect interaction effects. Results We present Matched Forest (MF), an algorithm for variable selection in matched case–control data. The method preserves the case and control values in each instance but transforms the matched case–control data with added counterfactuals. A modified variable importance score from a supervised learner is used to detect important variables. The method is conceptually simple and can be applied with widely available software tools. Simulation studies show the effectiveness of MF in identifying important variables. MF is also applied to data from the biomedical domain and its performance is compared with alternative approaches. Availability and implementation R code for implementing MF is available at https://github.com/NooshinSh/Matched_Forest. Supplementary information Supplementary data are available at Bioinformatics online.