A Unified Approach for Computing Wasserstein Barycenters of Discrete and Continuous Measures

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Xu, Peng, Zhu, Changbo, Chen, Xiaohui
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918494994432000
author Xu, Peng
Zhu, Changbo
Chen, Xiaohui
author_facet Xu, Peng
Zhu, Changbo
Chen, Xiaohui
contents Computing the unregularized Wasserstein barycenter for measure-valued data is a challenging optimization task. Recent algorithms have been tailored to either discrete measures as point clouds or continuous measures discretized on regular grids. In this work, we propose a primal mirror descent algorithm for computing the exact Wasserstein barycenter in the Fisher-Rao geometry. Our algorithm is a unified approach that is flexible enough to simultaneously cover discrete and absolutely continuous input measures, with convergence guarantees in both settings. In particular, when all input measures are discrete, our algorithm, initialized from any probability density, solves a sequence of semi-discrete optimal transport subproblems and produces absolutely continuous iterates that converge to the discrete barycenter. We use synthetic and real data examples to demonstrate the promising result in terms of accuracy and computational cost.
format Preprint
id arxiv_https___arxiv_org_abs_2605_11270
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle A Unified Approach for Computing Wasserstein Barycenters of Discrete and Continuous Measures
Xu, Peng
Zhu, Changbo
Chen, Xiaohui
Optimization and Control
Computing the unregularized Wasserstein barycenter for measure-valued data is a challenging optimization task. Recent algorithms have been tailored to either discrete measures as point clouds or continuous measures discretized on regular grids. In this work, we propose a primal mirror descent algorithm for computing the exact Wasserstein barycenter in the Fisher-Rao geometry. Our algorithm is a unified approach that is flexible enough to simultaneously cover discrete and absolutely continuous input measures, with convergence guarantees in both settings. In particular, when all input measures are discrete, our algorithm, initialized from any probability density, solves a sequence of semi-discrete optimal transport subproblems and produces absolutely continuous iterates that converge to the discrete barycenter. We use synthetic and real data examples to demonstrate the promising result in terms of accuracy and computational cost.
title A Unified Approach for Computing Wasserstein Barycenters of Discrete and Continuous Measures
topic Optimization and Control
url https://arxiv.org/abs/2605.11270