arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.03683stat.MEmath.STstat.MLstat.TH
收藏

一种算法无关变量重要性推断的通用框架

A general framework for inference on algorithm-agnostic variable importance

Brian D. Williamson, Peter B. Gilbert, Noah R. Simon, Marco Carone

AI总结:

针对现有变量重要性评估依赖特定预测算法、无法反映特征内在预测价值的问题,提出一种算法无关的非参数推断通用框架,可构建有效置信区间、开展假设检验,经模拟和HIV抗体研究数据验证性能良好。

摘要:

在许多应用中,研究人员常需评估特征(或特征子集)对预测响应变量目标的相对贡献,也就是衡量特征的变量重要性。近期大多数变量重要性评估研究,都聚焦于在给定预测算法的框架下描述特征重要性,但这类评估未必能体现特征的预测潜力,还可能对特征的内在价值产生误导性反映。为解决这一局限,我们提出一个用于可解释、算法无关变量重要性的非参数推断通用框架。我们将变量重要性定义为总体层面的对比值:所有可用特征的最优预测能力,与剔除待研究特征后其余特征的最优预测能力之差。我们提出了一种非参数高效估计方法,即便使用机器学习技术,也能构建有效的置信区间;我们还概述了一种检验零重要性假设的有效策略。通过模拟实验,我们证明所提方法具有良好的操作特性,并利用一项针对HIV-1感染抗体的研究数据,展示了该方法的实际应用。

英文摘要:

In many applications, it is of interest to assess the relative contribution of features (or subsets of features) toward the goal of predicting a response -- in other words, to gauge the variable importance of features. Most recent work on variable importance assessment has focused on describing the importance of features within the confines of a given prediction algorithm. However, such assessment does not necessarily characterize the prediction potential of features, and may provide a misleading reflection of the intrinsic value of these features. To address this limitation, we propose a general framework for nonparametric inference on interpretable algorithm-agnostic variable importance. We define variable importance as a population-level contrast between the oracle predictiveness of all available features versus all features except those under consideration. We propose a nonparametric efficient estimation procedure that allows the construction of valid confidence intervals, even when machine learning techniques are used. We also outline a valid strategy for testing the null importance hypothesis. Through simulations, we show that our proposal has good operating characteristics, and we illustrate its use with data from a study of an antibody against HIV-1 infection.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑