高维多数据源下的可扩展贝叶斯回归

Scalable Bayesian Regression in High Dimensions With Multiple Data Sources

Journal of Computational and Graphical Statistics · 2019
被引 5
ABS 3

中文导读

提出一种适用于总维度p远大于样本量n的宽数据场景的贝叶斯回归方法,通过经验贝叶斯自动设定各数据源特有的收缩水平,并利用n×n内积矩阵实现高效计算,可处理百万级特征。

Abstract

Applications of high-dimensional regression often involve multiple sources or types of covariates. We propose methodology for this setting, emphasizing the “wide data” regime with large total dimensionality p and sample size n≪p. We focus on a flexible ridge-type prior with shrinkage levels that are specific to each data type or source and that are set automatically by empirical Bayes. All estimation, including setting of shrinkage levels, is formulated mainly in terms of inner product matrices of size n×n. This renders computation efficient in the wide data regime and allows scaling to problems with millions of features. Furthermore, the proposed procedures are free of user-set tuning parameters. We show how sparsity can be achieved by post-processing of the Bayesian output via constrained minimization of a certain Kullback–Leibler divergence. This yields sparse solutions with adaptive, source-specific shrinkage, including a closed-form variant that scales to very large p. We present empirical results from a simulation study based on real data and a case study in Alzheimer’s disease involving millions of features and multiple data sources.

高维回归贝叶斯方法多数据源可扩展计算