[LIBRO][B] An introduction to optimization on smooth manifolds

N Boumal - 2023 - books.google.com
Optimization on Riemannian manifolds-the result of smooth geometry and optimization
merging into one elegant modern framework-spans many areas of science and engineering …

Variational inference via Wasserstein gradient flows

M Lambert, S Chewi, F Bach… - Advances in Neural …, 2022 - proceedings.neurips.cc
Abstract Along with Markov chain Monte Carlo (MCMC) methods, variational inference (VI)
has emerged as a central computational approach to large-scale Bayesian inference …

Forward-backward Gaussian variational inference via JKO in the Bures-Wasserstein space

MZ Diao, K Balasubramanian… - … on Machine Learning, 2023 - proceedings.mlr.press
Variational inference (VI) seeks to approximate a target distribution $\pi $ by an element of a
tractable family of distributions. Of key interest in statistics and machine learning is Gaussian …

Rates of estimation of optimal transport maps using plug-in estimators via barycentric projections

N Deb, P Ghosal, B Sen - Advances in Neural Information …, 2021 - proceedings.neurips.cc
Optimal transport maps between two probability distributions $\mu $ and $\nu $ on $\R^ d $
have found extensive applications in both machine learning and statistics. In practice, these …

[PDF][PDF] Statistical optimal transport

S Chewi, J Niles-Weed, P Rigollet - arxiv preprint arxiv:2407.18163, 2024 - arxiv.org
Statistical Optimal Transport arxiv:2407.18163v2 [math.ST] 7 Nov 2024 Page 1 Statistical
Optimal Transport Sinho Chewi Yale Jonathan Niles-Weed NYU Philippe Rigollet MIT …

Learning Gaussian mixtures using the Wasserstein–Fisher–Rao gradient flow

Y Yan, K Wang, P Rigollet - The Annals of Statistics, 2024 - projecteuclid.org
Learning Gaussian mixtures using the Wasserstein-Fisher-Rao gradient flow Page 1 The
Annals of Statistics 2024, Vol. 52, No. 4, 1774–1795 https://doi.org/10.1214/24-AOS2416 © …

The schrödinger bridge between gaussian measures has a closed form

C Bunne, YP Hsieh, M Cuturi… - … Conference on Artificial …, 2023 - proceedings.mlr.press
The static optimal transport $(\mathrm {OT}) $ problem between Gaussians seeks to recover
an optimal map, or more generally a coupling, to morph a Gaussian into another. It has been …

SVGD as a kernelized Wasserstein gradient flow of the chi-squared divergence

S Chewi, T Le Gouic, C Lu… - Advances in Neural …, 2020 - proceedings.neurips.cc
Abstract Stein Variational Gradient Descent (SVGD), a popular sampling algorithm, is often
described as the kernelized gradient flow for the Kullback-Leibler divergence in the …

Averaging on the Bures-Wasserstein manifold: dimension-free convergence of gradient descent

J Altschuler, S Chewi, PR Gerber… - Advances in Neural …, 2021 - proceedings.neurips.cc
We study first-order optimization algorithms for computing the barycenter of Gaussian
distributions with respect to the optimal transport metric. Although the objective is …

Wasserstein barycenters are NP-hard to compute

JM Altschuler, E Boix-Adsera - SIAM Journal on Mathematics of Data Science, 2022 - SIAM
Computing Wasserstein barycenters (aka optimal transport barycenters) is a fundamental
problem in geometry which has recently attracted considerable attention due to many …