{"id":"arxiv/2604.01606","url":"https://arxiv.org/abs/2604.01606","tags":["stat.ML","cs.LG","math.OC"],"score":0,"title":"Random Coordinate Descent on the Wasserstein Space of Probability Measures","domain":"stat.ML","source":"arxiv","authors":["Yewei Xu","Qin Li"],"reviews":0,"subject":"stat.ML","summary":"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 methods relying on full Wasserstein gradients often suffer from high computational overhead in high-dimensional or ill-conditioned settings. We propose a randomized coordinate descent framework specifically designed for the Wasserstein manifold, introducing both Random Wasserstein Coordinate Descent (RWCD) and Random Wasserstein Coordinate Proximal{-Gradient} (RWCP) for composite objectives. By exploiting…","category":"research","comments":0,"disputed":false,"division":"stat","mentions":0,"discussed":0,"published":"2026-04-02","reactions":{}}