Simple approximative algorithms for free-support Wasserstein barycenters

Johannes von Lindheim ()
Additional contact information
Johannes von Lindheim: Technische Universität Berlin

Computational Optimization and Applications, 2023, vol. 85, issue 1, No 7, 213-246

Abstract: Abstract Computing Wasserstein barycenters of discrete measures has recently attracted considerable attention due to its wide variety of applications in data science. In general, this problem is NP-hard, calling for practical approximative algorithms. In this paper, we analyze a well-known simple framework for approximating Wasserstein- $${\varvec{p}}$$ p barycenters, where we mainly consider the most common case $${\varvec{p}}={\varvec{2}}$$ p = 2 and $${\varvec{p}}={\varvec{1}}$$ p = 1 , which is not as well discussed. The framework produces sparse support solutions and shows good numerical results in the free-support setting. Depending on the desired level of accuracy, this requires only $${\varvec{N}}-{\varvec{1}}$$ N - 1 or $${\varvec{N(N}}-{\varvec{1)/2 }}$$ N ( N - 1 ) / 2 standard two-marginal optimal transport (OT) computations between the $${\varvec{N}}$$ N input measures, respectively, which is fast, memory-efficient and easy to implement using any OT solver as a black box. What is more, these methods yield a relative error of at most $${\varvec{N}}$$ N and $${\varvec{2}}$$ 2 , respectively, for both $${\varvec{p}}={\varvec{ 1, 2}}$$ p = 1 , 2 . We show that these bounds are practically sharp. In light of the hardness of the problem, it is not surprising that such guarantees cannot be close to optimality in general. Nevertheless, these error bounds usually turn out to be drastically lower for a given particular problem in practice and can be evaluated with almost no computational overhead, in particular without knowledge of the optimal solution. In our numerical experiments, this guaranteed errors of at most a few percent.

Keywords: Optimal transport; Discrete Wasserstein barycenter; Approximative algorithm; Error analysis; Sparsity; 65D18; 68U10; 90B80 (search for similar items in EconPapers)
Date: 2023
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
http://link.springer.com/10.1007/s10589-023-00458-3 Abstract (text/html)
Access to the full text of the articles in this series is restricted.

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:spr:coopap:v:85:y:2023:i:1:d:10.1007_s10589-023-00458-3

Ordering information: This journal article can be ordered from
http://www.springer.com/math/journal/10589

DOI: 10.1007/s10589-023-00458-3

Access Statistics for this article

Computational Optimization and Applications is currently edited by William W. Hager

More articles in Computational Optimization and Applications from Springer
Bibliographic data for series maintained by Sonal Shukla () and Springer Nature Abstracting and Indexing ().