Tetsuya Sakurai

Wasserstein Gradient Flow over Variational Parameter Space for Variational Inference

Variational Inference (VI) optimizes varia- tional parameters to closely align a variational distribution with the true posterior, being ap- proached through vanilla gradient descent in black-box VI or natural-gradient descent in natural-gradient VI. In this work, we reframe VI as the optimization o…