arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.09481stat.MEstat.ML

基于Shapley值的总体特征重要性高效非参数统计推断

Efficient nonparametric statistical inference on population feature importance using Shapley values

Brian D. Williamson, Jean Feng

AI总结:

针对Shapley总体变量重要性度量计算复杂度随变量数指数增长的问题,提出仅需采样Θ(n)个特征子集的高效估计器,证明其渐近最优收敛性,构建有效统计推断方法,在模拟和住院死亡率预测任务中表现良好。

摘要:

预测任务中变量在真实总体层面的重要性可为理解潜在数据生成机制提供有用信息,并能帮助确定后续实验中需要收集哪些测量值。对该重要性进行有效的统计推断是理解目标总体的关键组成部分。我们提出了一种计算高效的流程,用于估计Shapley总体变量重要性度量(SPVIM)并对其进行有效的统计推断。尽管真实SPVIM的计算复杂度随变量数量呈指数级增长,我们提出了一种估计器,在给定n个观测值的情况下,仅需随机采样Θ(n)个特征子集。我们证明了该估计器以渐近最优速率收敛。此外,通过推导估计器的渐近分布,我们构建了有效的置信区间和假设检验。我们的流程在模拟实验中表现出良好的有限样本性能,且在住院死亡率预测任务中,应用不同机器学习算法时能得到相似的变量重要性估计结果。

英文摘要:

The true population-level importance of a variable in a prediction task provides useful knowledge about the underlying data-generating mechanism and can help in deciding which measurements to collect in subsequent experiments. Valid statistical inference on this importance is a key component in understanding the population of interest. We present a computationally efficient procedure for estimating and obtaining valid statistical inference on the Shapley Population Variable Importance Measure (SPVIM). Although the computational complexity of the true SPVIM scales exponentially with the number of variables, we propose an estimator based on randomly sampling only $Θ(n)$ feature subsets given $n$ observations. We prove that our estimator converges at an asymptotically optimal rate. Moreover, by deriving the asymptotic distribution of our estimator, we construct valid confidence intervals and hypothesis tests. Our procedure has good finite-sample performance in simulations, and for an in-hospital mortality prediction task produces similar variable importance estimates when different machine learning algorithms are applied.

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

↑