Forward-backward Gaussian variational inference via JKO in the Bures-Wasserstein Space

Michael Diao, Krishnakumar Balasubramanian, Sinho Chewi +1

Variational inference (VI) seeks to approximate a target distribution $\pi$ by an element of a tractable family of distributions. Of key interest in statistics and machine learning i…