AI RESEARCH

Random Coordinate Descent on the Wasserstein Space of Probability Measures

arXiv CS.LG

ArXi:2604.01606v1 Announce Type: cross 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,