Manifold Gaussian Variational Bayes on the Precision Matrix
We propose an optimization algorithm for variational inference (VI) in complex models. Our approach relies on natural gradient updates where the variational space is a Riemann manifold. We develop an efficient algorithm for gaussian variational inference whose updates satisfy the positive definite constraint on the variational covariance matrix. Our manifold gaussian variational Bayes on the precision matrix (MGVBP) solution provides simple update rules, is straightforward to implement, and the use of the precision matrix parameterization has a significant computational advantage. Due to its black-box nature, MGVBP stands as a ready-to-use solution for VI in complex models. Over five data sets, we empirically validate our feasible approach on different statistical and econometric models, discussing its performance with respect to baseline methods.
Authors
- Martin Magris (ORCID: https://orcid.org/0000-0002-4237-2457)
- Alexandros Iosifidis (ORCID: https://orcid.org/0000-0003-4807-1345)
- Mostafa Shabani (ORCID: https://orcid.org/0000-0001-7552-3525)
Institutions
- Instituto Tecnológico Autónomo de México (MX)
- Aarhus University (DK)
Publication Details
- Journal
- Neural Computation
- Published
- 2024-08-19
- DOI
- https://doi.org/10.1162/neco_a_01686
- Primary Topic
- Bayesian Methods and Mixture Models
- Type
- article
- Field-Weighted Citation Impact
- 0.00