Abstract:This paper studies the convergence rates of optimal transport (OT) map estimators, a topic of growing interest in statistics, machine learning, and various scientific fields. Despite recent advancements, existing results rely on regularity assumptions that are very restrictive in practice and much stricter than those in Brenier's Theorem, including the compactness and convexity of the probability support and the bi-Lipschitz property of the OT maps. We aim to broaden the scope of OT map estimation and fill this gap between theory and practice. Given the strong convexity assumption on Brenier's potential, we first establish the non-asymptotic convergence rates for the original plug-in estimator without requiring restrictive assumptions on probability measures. Additionally, we introduce a sieve plug-in estimator and establish its convergence rates without the strong convexity assumption on Brenier's potential, enabling the widely used cases such as the rank functions of normal or t-distributions. We also establish new Poincaré-type inequalities, which are proved given sufficient conditions on the local boundedness of the probability density and mild topological conditions of the support, and these new inequalities enable us to achieve faster convergence rates for the Donsker function class. Moreover, we develop scalable algorithms to efficiently solve the OT map estimation using neural networks and present numerical experiments to demonstrate the effectiveness and robustness.

Stability and Sample Complexity of Divergence Regularized Optimal Transport

Statistical Convergence Rates of Optimal Transport Map Estimation between General Distributions

The stability and statistic of domain decomposition algorithm with mini-batch learning for optimal transport

Asymptotic analysis of domain decomposition for optimal transport

Limit Theorems for Entropic Optimal Transport Maps and the Sinkhorn Divergence

An Homogeneous Unbalanced Regularized Optimal Transport model with applications to Optimal Transport with Boundary

Stability and statistical inference for semidiscrete optimal transport maps

Regularization for Wasserstein distributionally robust optimization

Convergence Rates of the Regularized Optimal Transport : Disentangling Suboptimality and Entropy

Weak limits of entropy regularized Optimal Transport; potentials, plans and divergences

Lower Complexity Adaptation for Empirical Entropic Optimal Transport

Unifying Distributionally Robust Optimization via Optimal Transport Theory

Estimation of entropy-regularized optimal transport maps between non-compactly supported measures

Regularity theory and geometry of unbalanced optimal transport

Quantitative stability in optimal transport for general power costs

Duality-based Dynamical Optimal Transport of Discrete Time Systems

Convergence rate of entropy-regularized multi-marginal optimal transport costs

A Sinkhorn-type Algorithm for Constrained Optimal Transport

Interpolating between Optimal Transport and KL regularized Optimal Transport using Rényi Divergences

Quantum optimal transport with convex regularization