Scale-Dependent Radial Geometry and Metric Mismatch in Wasserstein Propagation for Reverse Diffusion

arXiv cs.LG / 3/23/2026

📰 NewsIdeas & Deep AnalysisModels & Research

Key Points

  • The paper shows that propagating sampling error in the Euclidean geometry underlying W2 during reverse diffusion can be suboptimal, as contraction can occur first at large separations in a radial geometry under weak log-concavity.
  • It formalizes a radial lower profile for the learned reverse drift, linking far-field contraction reserve and near-field Euclidean load, and uses positivity of the reserve to define admissible switch times.
  • A one-switch routing argument is proposed: before the switch, contraction is achieved with reflection coupling in a concave transport metric adapted to the radial profile; at the switch, the analysis transfers from this metric back to W2 under a p-moment budget to handle the remaining window.
  • Under discretizations with L2 score-error control and standard assumptions, the authors derive explicit non-asymptotic end-to-end W2 guarantees, a scalar switch-selection objective, and a sharp bound on the conversion exponent within an affine-tail concave class.

Abstract

Existing analyses of reverse diffusion often propagate sampling error in the Euclidean geometry underlying \(\Wtwo\) along the entire reverse trajectory. Under weak log-concavity, however, Gaussian smoothing can create contraction first at large separations while short separations remain non-dissipative. The first usable contraction is therefore radial rather than Euclidean, creating a metric mismatch between the geometry that contracts early and the geometry in which the terminal error is measured. We formalize this mismatch through an explicit radial lower profile for the learned reverse drift. Its far-field limit gives a contraction reserve, its near-field limit gives the Euclidean load governing direct \(\Wtwo\) propagation, and admissible switch times are characterized by positivity of the reserve on the remaining smoothing window. We exploit this structure with a one-switch routing argument. Before the switch, reflection coupling yields contraction in a concave transport metric adapted to the radial profile. At the switch, we convert once from this metric back to \(\Wtwo\) under a \(p\)-moment budget, and then propagate the converted discrepancy over the remaining short window in Euclidean geometry. For discretizations of the learned reverse SDE under \(L^2\) score-error control, a one-sided Lipschitz condition of score error, and standard well-posedness and coupling hypotheses, we obtain explicit non-asymptotic end-to-end \(\Wtwo\) guarantees, a scalar switch-selection objective, and a sharp structural limit on the conversion exponent within the affine-tail concave class.