Scalable Spike-and-Slab
Niloy Biswas, Lester Mackey, Xiao-Li Meng
Abstract
Spike-and-slab priors are commonly used for Bayesian variable selection, due to their inter-pretability and favorable statistical properties. However, existing samplers for spike-and-slab posteriors incur prohibitive computational costs when the number of variables is large. In this article, we propose Scalable Spike-and-Slab ( S 3 ), a scalable Gibbs sampling implementation for high-dimensional Bayesian regression with the continuous spike-and-slab prior of George & McCulloch (1993). For a dataset with n observations and p covariates, S 3 has order max n 2 p t , np computational cost at iteration t where p t never exceeds the number of covariates switching spike-and-slab states between iterations t and t − 1 of the Markov chain. This improves upon the order n 2 p per-iteration cost of state-of-the-art implementations as, typically, p t is substantially smaller than p . We apply S 3 on synthetic and real-world datasets, demonstrating orders of magnitude speed-ups over existing exact samplers and significant gains in inferential quality over approximate samplers with comparable cost.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 00eab425-3e92-4f9b-bf31-412ac4f57bacCited by top-tier papers1
Ask how each one uses itBuilds on1
Related papers
- Thompson Sampling for High-Dimensional Sparse Linear Contextual BanditsSunrit Chakraborty, Saptarshi Roy, Ambuj TewariICML 2023 · 15 citations
- Flexible mean field variational inference using mixtures of non-overlapping exponential familiesJeffrey P. SpenceNeurIPS 2020 · 5 citations
- Efficient Variational Inference for Sparse Deep Learning with Theoretical GuaranteeJincheng Bai, Qifan Song, Guang ChengNeurIPS 2020 · 55 citations
- Scalable Variational Bayesian Kernel Selection for Sparse Gaussian Process RegressionTong Teng, Jie Chen, Yehong Zhang, Bryan Kian Hsiang LowAAAI 2020 · 24 citations
- Information Directed Sampling for Sparse Linear BanditsBotao Hao, Tor Lattimore, Wei DengNeurIPS 2021 · 22 citations
