Computational Guarantees for Doubly Entropic Wasserstein Barycenters via Damped Sinkhorn Iterations

Lénaïc Chizat, Tomas Vaškevičius
7/25/2023
math.OCcs.LGstat.ML

No confident code implementation found for this paper.

Abstract

We study the computation of doubly regularized Wasserstein barycenters, a recently introduced family of entropic barycenters governed by inner and outer regularization strengths. Previous research has demonstrated that various regularization parameter choices unify several notions of entropy-penalized barycenters while also revealing new ones, including a special case of debiased barycenters. In this paper, we propose and analyze an algorithm for computing doubly regularized Wasserstein barycenters. Our procedure builds on damped Sinkhorn iterations followed by exact maximization/minimization steps and guarantees convergence for any choice of regularization parameters. An inexact variant of our algorithm, implementable using approximate Monte Carlo sampling, offers the first non-asymptotic convergence guarantees for approximating Wasserstein barycenters between discrete point clouds in the free-support/grid-free setting.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{chizat2023computational,
  title  = {Computational Guarantees for Doubly Entropic Wasserstein Barycenters via Damped Sinkhorn Iterations},
  author = {Lénaïc Chizat and Tomas Vaškevičius},
  year   = {2023},
  eprint = {2307.13370},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2307.13370v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2307.13370.svg)](https://codeofpaper.com/paper/2307.13370)

Discussion