Timezone: »

Continuous Regularized Wasserstein Barycenters
Lingxiao Li · Aude Genevay · Mikhail Yurochkin · Justin M Solomon

Wed Dec 09 09:00 AM -- 11:00 AM (PST) @ Poster Session 3 #897

Wasserstein barycenters provide a geometrically meaningful way to aggregate probability distributions, built on the theory of optimal transport. They are difficult to compute in practice, however, leading previous work to restrict their supports to finite sets of points. Leveraging a new dual formulation for the regularized Wasserstein barycenter problem, we introduce a stochastic algorithm that constructs a continuous approximation of the barycenter. We establish strong duality and use the corresponding primal-dual relationship to parametrize the barycenter implicitly using the dual potentials of regularized transport problems. The resulting problem can be solved with stochastic gradient descent, which yields an efficient online algorithm to approximate the barycenter of continuous distributions given sample access. We demonstrate the effectiveness of our approach and compare against previous work on synthetic examples and real-world applications.

Author Information

Lingxiao Li (MIT)
Aude Genevay (MIT)
Mikhail Yurochkin (IBM Research, MIT-IBM Watson AI Lab)
Justin M Solomon (MIT)

More from the Same Authors