Linearized Wasserstein Barycenters: Synthesis, Analysis, Representational Capacity, and Applications
Abstract
We propose the linear barycentric coding model (LBCM) which utilizes the linear optimal transport (LOT) metric for analysis and synthesis of probability measures. We provide a closed-form solution to the variational problem characterizing the probability measures in the LBCM and establish equivalence of the LBCM to the set of 2-Wasserstein barycenters in the special case of compatible measures. Computational methods for synthesizing and analyzing measures in the LBCM are developed with finite sample guarantees. One of our main theoretical contributions is to identify an LBCM, expressed in terms of a simple family, which is sufficient to express all probability measures on the closed unit interval. We show that a natural analogous construction of an LBCM in 2 dimensions fails, and we leave it as an open problem to identify the proper extension in more than 1 dimension. We conclude by demonstrating the utility of LBCM for covariance estimation and data imputation.
Cite
Text
Werenski et al. "Linearized Wasserstein Barycenters: Synthesis, Analysis, Representational Capacity, and Applications." Proceedings of The 28th International Conference on Artificial Intelligence and Statistics, 2025.Markdown
[Werenski et al. "Linearized Wasserstein Barycenters: Synthesis, Analysis, Representational Capacity, and Applications." Proceedings of The 28th International Conference on Artificial Intelligence and Statistics, 2025.](https://mlanthology.org/aistats/2025/werenski2025aistats-linearized/)BibTeX
@inproceedings{werenski2025aistats-linearized,
title = {{Linearized Wasserstein Barycenters: Synthesis, Analysis, Representational Capacity, and Applications}},
author = {Werenski, Matthew and Mallery, Brendan and Aeron, Shuchin and Murphy, James M.},
booktitle = {Proceedings of The 28th International Conference on Artificial Intelligence and Statistics},
year = {2025},
pages = {4555-4563},
volume = {258},
url = {https://mlanthology.org/aistats/2025/werenski2025aistats-linearized/}
}