Debiaser Beware: Pitfalls of Centering Regularized Transport Maps
Aram-Alexandre Pooladian, Marco Cuturi, Jonathan Niles-Weed
Abstract
Estimating optimal transport (OT) maps (a.k.a. Monge maps) between two measures and is a problem fraught with computational and statistical challenges. A promising approach lies in using the dual potential functions obtained when solving an entropy-regularized OT problem between samples and , which can be used to recover an approximately optimal map. The negentropy penalization in that scheme introduces, however, an estimation bias that grows with the regularization strength. A well-known remedy to debias such estimates, which has gained wide popularity among practitioners of regularized OT, is to center them, by subtracting auxiliary problems involving and itself, as well as and itself. We do prove that, under favorable conditions on and , debiasing can yield better approximations to the Monge map. However, and perhaps surprisingly, we present a few cases in which debiasing is provably detrimental in a statistical sense, notably when the regularization strength is large or the number of samples is small. These claims are validated experimentally on synthetic and real datasets, and should reopen the debate on whether debiasing is needed when using entropic optimal transport.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- Minimax estimation of discontinuous optimal transport maps: The semi-discrete caseAram-Alexandre Pooladian, Vincent Divol, Jonathan Niles-WeedICML 2023 · 29 citations
- Progressive Entropic Optimal Transport SolversParnian Kassraie, Aram-Alexandre Pooladian, Michal Klein, James Thornton et al.NeurIPS 2024 · 14 citations
Builds on4
- Faster Wasserstein Distance Estimation with the Sinkhorn DivergenceLénaïc Chizat, Pierre Roussillon, Flavien Léger, François-Xavier Vialard et al.NeurIPS 2020 · 164 citations
- Entropic Optimal Transport between Unbalanced Gaussian Measures has a Closed FormHicham Janati, Boris Muzellec, Gabriel Peyré, Marco CuturiNeurIPS 2020 · 109 citations
- Convex Potential Flows: Universal Probability Distributions with Optimal Transport and Convex OptimizationChin-Wei Huang, Ricky T. Q. Chen, Christos Tsirigotis, Aaron C. CourvilleICLR 2021 · 107 citations
- Rates of Estimation of Optimal Transport Maps using Plug-in Estimators via Barycentric ProjectionsNabarun Deb, Promit Ghosal, Bodhisattva SenNeurIPS 2021 · 96 citations
Related papers
- Decreasing Entropic Regularization Averaged Gradient for Semi-Discrete Optimal TransportFerdinand Genans, Antoine Godichon-Baggioni, François-Xavier Vialard, Olivier WintenbergerNeurIPS 2025 · 2 citations
- Debiased Sinkhorn barycentersHicham Janati, Marco Cuturi, Alexandre GramfortICML 2020 · 62 citations
- Entropic Neural Optimal Transport via Diffusion ProcessesNikita Gushchin, Alexander Kolesov, Alexander Korotin, Dmitry P. Vetrov et al.NeurIPS 2023 · 59 citations
- Regularized Optimal Transport is Ground Cost AdversarialFrançois-Pierre Paty, Marco CuturiICML 2020 · 33 citations
- Computational Guarantees for Doubly Entropic Wasserstein BarycentersTomas Vaskevicius, Lénaïc ChizatNeurIPS 2023 · 5 citations
