Stochastic variance-reduced Gaussian variational inference on the Bures-Wasserstein manifold
Optimization in the Bures-Wasserstein space has been gaining popularity in the machine learning community since it draws connections between variational inference and Wasserstein gradient flows. The variational inference objective function of Kullback–Leibler divergence can be written as the sum of…