Current browse context:
stat.ML
Change to browse by:
References & Citations
Statistics > Machine Learning
Title: Extending Mean-Field Variational Inference via Entropic Regularization: Theory and Computation
(Submitted on 14 Apr 2024 (v1), last revised 16 Apr 2024 (this version, v2))
Abstract: Variational inference (VI) has emerged as a popular method for approximate inference for high-dimensional Bayesian models. In this paper, we propose a novel VI method that extends the naive mean field via entropic regularization, referred to as $\Xi$-variational inference ($\Xi$-VI). $\Xi$-VI has a close connection to the entropic optimal transport problem and benefits from the computationally efficient Sinkhorn algorithm. We show that $\Xi$-variational posteriors effectively recover the true posterior dependency, where the dependence is downweighted by the regularization parameter. We analyze the role of dimensionality of the parameter space on the accuracy of $\Xi$-variational approximation and how it affects computational considerations, providing a rough characterization of the statistical-computational trade-off in $\Xi$-VI. We also investigate the frequentist properties of $\Xi$-VI and establish results on consistency, asymptotic normality, high-dimensional asymptotics, and algorithmic stability. We provide sufficient criteria for achieving polynomial-time approximate inference using the method. Finally, we demonstrate the practical advantage of $\Xi$-VI over mean-field variational inference on simulated and real data.
Submission history
From: Bohan Wu [view email][v1] Sun, 14 Apr 2024 01:40:11 GMT (2726kb,D)
[v2] Tue, 16 Apr 2024 18:43:22 GMT (2725kb,D)
Link back to: arXiv, form interface, contact.