The holdout randomization test for feature selection in black box models Journal Article


Authors: Tansey, W.; Veitch, V.; Zhang, H.; Rabadan, R.; Blei, D. M.
Article Title: The holdout randomization test for feature selection in black box models
Abstract: We propose the holdout randomization test (HRT), an approach to feature selection using black box predictive models. The HRT is a specialized version of the conditional randomization test (CRT) that uses data splitting for feasible computation. The HRT works with any predictive model and produces a valid p-value for each feature. To make the HRT more practical, we propose a set of extensions to maximize power and speed up computation. In simulations, these extensions lead to greater power than a competing knockoffs-based approach, without sacrificing control of the error rate. We apply the HRT to two case studies from the scientific literature where heuristics were originally used to select important features for predictive models. The results illustrate how such heuristics can be misleading relative to principled methods like the HRT. Code is available at . for this article are available online.
Keywords: false discovery rate; feature selection; black box models; conditional independence testing
Journal Title: Journal of Computational and Graphical Statistics
Volume: 31
Issue: 1
ISSN: 1061-8600
Publisher: Taylor & Francis Group  
Date Published: 2022-01-01
Start Page: 151
End Page: 162
Language: English
ACCESSION: WOS:000678932100001
DOI: 10.1080/10618600.2021.1923520
PROVIDER: wos
Notes: Article -- Source: Wos
Altmetric
Citation Impact
BMJ Impact Analytics
MSK Authors
  1. Wesley Tansey
    15 Tansey