Random Coordinate Descent on the Wasserstein Space of Probability Measures

Yewei Xu, Qin Li

Optimization over the space of probability measures endowed with the Wasserstein-2 geometry is central to modern machine learning and mean-field modeling. However, traditional meth…