Current browse context:
stat.ME
Change to browse by:
References & Citations
Statistics > Methodology
Title: Variational Bayes for high-dimensional linear regression with sparse priors
(Submitted on 15 Apr 2019 (v1), last revised 19 Nov 2020 (this version, v3))
Abstract: We study a mean-field spike and slab variational Bayes (VB) approximation to Bayesian model selection priors in sparse high-dimensional linear regression. Under compatibility conditions on the design matrix, oracle inequalities are derived for the mean-field VB approximation, implying that it converges to the sparse truth at the optimal rate and gives optimal prediction of the response vector. The empirical performance of our algorithm is studied, showing that it works comparably well as other state-of-the-art Bayesian variable selection methods. We also numerically demonstrate that the widely used coordinate-ascent variational inference (CAVI) algorithm can be highly sensitive to the parameter updating order, leading to potentially poor performance. To mitigate this, we propose a novel prioritized updating scheme that uses a data-driven updating order and performs better in simulations. The variational algorithm is implemented in the R package 'sparsevb'.
Submission history
From: Kolyan Ray [view email][v1] Mon, 15 Apr 2019 15:58:44 GMT (5155kb,D)
[v2] Wed, 8 Jul 2020 21:33:39 GMT (6063kb,D)
[v3] Thu, 19 Nov 2020 15:06:11 GMT (6227kb,D)
Link back to: arXiv, form interface, contact.