Computing Wasserstein barycenter via operator splitting: the method of averaged marginals - Archive ouverte HAL
Article Dans Une Revue SIAM Journal on Mathematics of Data Science Année : 2024

Computing Wasserstein barycenter via operator splitting: the method of averaged marginals

Daniel Mimouni
Paul Malisani
  • Fonction : Auteur
  • PersonId : 763594
  • IdRef : 164602798

Résumé

The Wasserstein barycenter (WB) is an important tool for summarizing sets of probability measures. It finds applications in applied probability, clustering, image processing, etc. When the measures’ supports are finite, computing a (balanced) WB can be done by solving a linear optimization problem whose dimensions generally exceed standard solvers’ capabilities. In the more general setting where measures have different total masses, we propose a convex nonsmooth optimization formulation for the so-called unbalanced WB problem. Due to their colossal dimensions, we introduce a decomposition scheme based on the Douglas–Rachford splitting method that can be applied to both balanced and unbalanced WB problem variants. Our algorithm, which has the interesting interpretation of being built upon averaging marginals, operates a series of simple (and exact) projections that can be parallelized and even randomized, making it suitable for large-scale datasets. Numerical comparisons against state-of-the-art methods on several data sets from the literature illustrate the method’s performance.
Fichier non déposé

Dates et versions

hal-04709465 , version 1 (25-09-2024)

Identifiants

Citer

Daniel Mimouni, Paul Malisani, Jiamin Zhu, Welington de Oliveira. Computing Wasserstein barycenter via operator splitting: the method of averaged marginals. SIAM Journal on Mathematics of Data Science, In press, 6 (4), pp.1000-1026. ⟨10.1137/23M1584228⟩. ⟨hal-04709465⟩
30 Consultations
0 Téléchargements

Altmetric

Partager

More