Stochastic Wasserstein Barycenters

Sebastian Claici, Edward Chien, Justin Solomon

Abstract: We present a stochastic algorithm to compute the barycenter of a set of probability distributions under the Wasserstein metric from optimal transport. Unlike previous approaches, our method extends...