Computational Guarantees for Doubly Entropic Wasserstein Barycenters via Damped Sinkhorn Iterations

07/25/2023
by   Lenaïc Chizat, et al.
0

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.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
07/01/2019

Implementation of batched Sinkhorn iterations for entropy-regularized Wasserstein loss

In this report, we review the calculation of entropy-regularised Wassers...
research
03/21/2023

Doubly Regularized Entropic Wasserstein Barycenters

We study a general formulation of regularized Wasserstein barycenters th...
research
04/24/2018

Data-driven regularization of Wasserstein barycenters with an application to multivariate density registration

We present a framework to simultaneously align and smooth data in the fo...
research
10/17/2022

Statistical, Robustness, and Computational Guarantees for Sliced Wasserstein Distances

Sliced Wasserstein distances preserve properties of classic Wasserstein ...
research
05/26/2023

Exact Generalization Guarantees for (Regularized) Wasserstein Distributionally Robust Models

Wasserstein distributionally robust estimators have emerged as powerful ...
research
11/21/2022

A Bi-level Nonlinear Eigenvector Algorithm for Wasserstein Discriminant Analysis

Much like the classical Fisher linear discriminant analysis, Wasserstein...
research
12/30/2019

An Inner-Outer Iterative Method for Edge Preservation in Image Restoration and Reconstruction

We present a new inner-outer iterative algorithm for edge enhancement in...

Please sign up or login with your details

Forgot password? Click here to reset