476

On the Wasserstein median of probability measures

Journal of Computational And Graphical Statistics (JCGS), 2022
Main:25 Pages
9 Figures
Bibliography:1 Pages
Appendix:1 Pages
Abstract

Measures of central tendency such as the mean and the median are a primary way to summarize a given collection of random objects. In the field of optimal transport, the Wasserstein barycenter corresponds to the Fr\'echet or geometric mean of a set of probability measures, which is defined as a minimizer of the sum of its squared distances to each element of the set when the order is 2. We present the Wasserstein median, an equivalent of the Fr\'echet median under the 2- Wasserstein metric, as a robust alternative to the barycenter. The Wasserstein median is shown to be well defined and exist under mild conditions. We also propose an algorithm that makes use of any established routine for the Wasserstein barycenter in an iterative manner and prove its convergence. Our proposal is validated with simulated and real data examples when the objects of interest are univariate distributions, centered Gaussian distributions, and discrete measures on regular grids.

View on arXiv
Comments on this paper