mlsanityData to 7 Oct 2026

Rate-Optimal Noise Annealing in Semi-Dual Neural Optimal Transport: Tangential Identifiability, Off-Manifold Ambiguity, and Guaranteed Recovery

arXiv 2602.04110 ↗First posted 4 Feb 2026cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

Semi-dual neural optimal transport learns a transport map via a max-min objective, yet training can converge to incorrect or degenerate maps. We fully characterize these spurious solutions in the common regime where data concentrate on low-dimensional manifold: the objective is underconstrained off the data manifold, while the on-manifold transport signal remains identifiable. Following Choi, Choi, and Kwon (2025), we study additive-noise smoothing as a remedy and prove new map recovery guarantees as the noise vanishes. Our main practical contribution is a computable terminal noise level varepsilonstat(N) that attains the optimal statistical rate, with scaling governed by the intrinsic dimension m of the data. The formula arises from a theoretical unified analysis of (i) quantitative stability of optimal plans, (ii) smoothing-induced bias, and (iii) finite-sample error, yielding rates that depend on m rather than the ambient dimension. Finally, we show that the reduced semi-dual objective becomes increasingly ill-conditioned as varepsilon downarrow 0. This provides a principled stopping rule: annealing below varepsilonstat(N) can textitworsen optimization conditioning without improving statistical accuracy.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 14 of its 22 references that are in the map. First posted on arXiv 4 Feb 2026.

Builds on

14 of 22 references.

References · 14

  1. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  2. 2022Flow Matching for Generative Modelingcited by 5,426
  3. 2018Which Training Methods for GANs do actually Converge?cited by 672
  4. 2019Optimal transport mapping via input convex neural networkscited by 202
  5. 2022Neural Optimal Transportcited by 123
  6. 2021Generative Modeling with Optimal Transport Mapscited by 80
  7. 2024Statistical optimal transportcited by 44
  8. 2021Neural Monge Map estimation and its applicationscited by 40
  9. 2023Generative Modeling through the Semi-dual Formulation of Unbalanced Optimal Transportcited by 39
  10. 2019Quantitative stability of optimal transport maps and linearization of the 2-Wasserstein spacecited by 25
  11. 2024Scalable Wasserstein Gradient Flow for Generative Modeling through Unbalanced Optimal Transportcited by 22
  12. 2021Intrinsic Dimension Estimation Using Wasserstein Distancescited by 20
2 more
  1. 2022An Optimal Transport Perspective on Unpaired Image Super-Resolutioncited by 12
  2. 2025Overcoming Spurious Solutions in Semi-Dual Neural Optimal Transport: A Smoothing Approach for Learning the Optimal Transport Plancited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.