Convergence Analysis of the Wasserstein Proximal Algorithm beyond Geodesic Convexity

Shuailong Zhu, Xiaohui Chen

The proximal algorithm is a powerful tool to minimize nonlinear and nonsmooth functionals in a general metric space. Motivated by the recent progress in studying the training dynam…