黑箱模型中特征选择的留出随机化检验

The Holdout Randomization Test for Feature Selection in Black Box Models

Journal of Computational and Graphical Statistics · 2021
被引 21
ABS 3

中文导读

提出留出随机化检验(HRT),一种利用黑箱预测模型进行特征选择的方法,通过数据拆分生成有效p值,并在模拟和案例中优于启发式方法。

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 https://github.com/tansey/hrt. Supplementary materials for this article are available online.

特征选择机器学习统计检验黑箱模型