English

Wasserstein barycenters can be computed in polynomial time in fixed dimension

Optimization and Control 2021-11-16 v2 Computational Geometry Data Structures and Algorithms Machine Learning

Abstract

Computing Wasserstein barycenters is a fundamental geometric problem with widespread applications in machine learning, statistics, and computer graphics. However, it is unknown whether Wasserstein barycenters can be computed in polynomial time, either exactly or to high precision (i.e., with polylog(1/ε)\textrm{polylog}(1/\varepsilon) runtime dependence). This paper answers these questions in the affirmative for any fixed dimension. Our approach is to solve an exponential-size linear programming formulation by efficiently implementing the corresponding separation oracle using techniques from computational geometry.

Keywords

Cite

@article{arxiv.2006.08012,
  title  = {Wasserstein barycenters can be computed in polynomial time in fixed dimension},
  author = {Jason M. Altschuler and Enric Boix-Adsera},
  journal= {arXiv preprint arXiv:2006.08012},
  year   = {2021}
}

Comments

15 pages + refs, 5 figs. Improved exposition. Title has been updated for clarity

R2 v1 2026-06-23T16:19:02.842Z