Scalable Bayesian Regression in High Dimensions With Multiple Data Sources
提出一种适用于总维度p远大于样本量n的宽数据场景的贝叶斯回归方法,通过经验贝叶斯自动设定各数据源特有的收缩水平,并利用n×n内积矩阵实现高效计算,可处理百万级特征。
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.