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 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.
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