Fixed-Support Wasserstein Barycenters: Computational Hardness and Fast Algorithm

02/12/2020
by   Tianyi Lin, et al.
0

We study the fixed-support Wasserstein barycenter problem (FS-WBP), which consists in computing the Wasserstein barycenter of m discrete probability measures supported on a finite metric space of size n. We show first that the constraint matrix arising from the standard linear programming (LP) representation of the FS-WBP is not totally unimodular when m ≥ 3 and n ≥ 3. This result resolves an open question pertaining to the relationship between the FS-WBP and the minimum-cost flow (MCF) problem since it proves that the FS-WBP in the standard LP form is not an MCF problem when m ≥ 3 and n ≥ 3. We also develop a provably fast deterministic variant of the celebrated iterative Bregman projection (IBP) algorithm, named FastIBP, with a complexity bound of Õ(mn^7/3ε^-4/3), where ε∈ (0, 1) is the tolerance. This complexity bound is better than the best known complexity bound of Õ(mn^2ε^-2) for the IBP algorithm in terms of ε, and that of Õ(mn^5/2ε^-1) from other accelerated algorithms in terms of n. Finally, we conduct extensive experiments with both synthetic and real data and demonstrate the favorable performance of the FastIBP algorithm in practice.

READ FULL TEXT

Authors

02/12/2020

Computational Hardness and Fast Algorithm for Fixed-Support Wasserstein Barycenter

We study in this paper the fixed-support Wasserstein barycenter problem ...
02/12/2020

Revisiting Fixed Support Wasserstein Barycenter: Computational Hardness and Efficient Algorithms

We study the fixed-support Wasserstein barycenter problem (FS-WBP), whic...
06/12/2020

Projection Robust Wasserstein Distance and Riemannian Optimization

Projection robust Wasserstein (PRW) distance, or Wasserstein projection ...
01/19/2019

On Efficient Optimal Transport: An Analysis of Greedy and Accelerated Mirror Descent Algorithms

We provide theoretical analyses for two algorithms that solve the regula...
09/12/2018

A Fast Globally Linearly Convergent Algorithm for the Computation of Wasserstein Barycenters

In this paper, we consider the problem of computing a Wasserstein baryce...
10/10/2019

On Scalable Variant of Wasserstein Barycenter

We study a variant of Wasserstein barycenter problem, which we refer to ...
09/30/2015

Fast Discrete Distribution Clustering Using Wasserstein Barycenter with Sparse Support

In a variety of research areas, the weighted bag of vectors and the hist...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.