Yu Hang, Wang Yuanjia, Zeng Donglin
Department of Statistics and Operation Research, University of North Carolina at Chapel Hill, North Carolina, United State.
Department of Biostatistics, Mailman School of Public Health, Columbia University, United State.
Stat (Int Stat Inst). 2020;9(1). doi: 10.1002/sta4.300. Epub 2020 Jul 6.
With growing interest to use black-box machine learning for complex data with many feature variables, it is critical to obtain a prediction model that only depends on a small set of features to maximize generalizability. Therefore, feature selection remains to be an important and challenging problem in modern applications. Most of existing methods for feature selection are based on either parametric or semiparametric models, so the resulting performance can severely suffer from model misspecification when high-order nonlinear interactions among the features are present. A very limited number of approaches for nonparametric feature selection were proposed, but they are computationally intensive and may not even converge. In this paper, we propose a novel and computationally efficient approach for nonparametric feature selection in regression field based on a tensor-product kernel function over the feature space. The importance of each feature is governed by a parameter in the kernel function which can be efficiently computed iteratively from a modified alternating direction method of multipliers (ADMM) algorithm. We prove the oracle selection property of the proposed method. Finally, we demonstrate the superior performance of our approach compared to existing methods via simulation studies and application to the prediction of Alzheimer's disease.
随着人们对将黑箱机器学习用于具有许多特征变量的复杂数据的兴趣日益增加,获得一个仅依赖于一小部分特征以最大化通用性的预测模型至关重要。因此,特征选择在现代应用中仍然是一个重要且具有挑战性的问题。现有的大多数特征选择方法基于参数模型或半参数模型,因此当特征之间存在高阶非线性相互作用时,所得性能可能会因模型误设而严重受损。提出的非参数特征选择方法数量非常有限,但它们计算量大,甚至可能不收敛。在本文中,我们基于特征空间上的张量积核函数,提出了一种用于回归领域非参数特征选择的新颖且计算高效的方法。每个特征的重要性由核函数中的一个参数控制,该参数可以通过修改后的交替方向乘子法(ADMM)算法迭代有效地计算出来。我们证明了所提方法的最优选择性质。最后,通过模拟研究以及在阿尔茨海默病预测中的应用,我们展示了我们的方法与现有方法相比的优越性能。