Rate-Optimal Noise Annealing in Semi-Dual Neural Optimal Transport: Tangential Identifiability, Off-Manifold Ambiguity, and Guaranteed Recovery
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
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2018Which Training Methods for GANs do actually Converge?cited by 672
- 2019Optimal transport mapping via input convex neural networkscited by 202
- 2022Neural Optimal Transportcited by 123
- 2021Generative Modeling with Optimal Transport Mapscited by 80
- 2024Statistical optimal transportcited by 44
- 2021Neural Monge Map estimation and its applicationscited by 40
- 2023Generative Modeling through the Semi-dual Formulation of Unbalanced Optimal Transportcited by 39
- 2019Quantitative stability of optimal transport maps and linearization of the 2-Wasserstein spacecited by 25
- 2024Scalable Wasserstein Gradient Flow for Generative Modeling through Unbalanced Optimal Transportcited by 22
- 2021Intrinsic Dimension Estimation Using Wasserstein Distancescited by 20
Built on by
Citing papers per year: 2026: 1.
1 paper.