Manifold-Aligned Generative Transport
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Many high-dimensional datasets concentrate near a low-dimensional structure embedded in the ambient space. Generative models for such data must control off-support mass while remaining computationally practical. Diffusion models use iterative denoising at inference, whereas standard normalizing flows require invertible, dimension-preserving maps. We propose MAGT (Manifold-Aligned Generative Transport), a direct transport from a low-dimensional base distribution to the data space. Its core objective compares the data and generator-induced scores at a selected Gaussian smoothing level. A posterior identity expresses this score through a latent conditional mean, which is approximated by self-normalized importance sampling over a finite anchor set. After training, generation requires one evaluation of the transport, whose image also carries an intrinsic density with respect to manifold volume. We establish a minimax-optimal Wasserstein convergence rate for an explicitly constructed localized spline-RePU coordinate transport estimator, and treat finite-anchor approximation separately. Experiments on synthetic, image, and tabular benchmarks compare fidelity, support alignment, and sampling cost with diffusion, flow-matching, and adversarial baselines.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. First posted on arXiv 23 Feb 2026.
Builds on
0 of 0 references.
No references here yet.
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026TRACE: Transport Alignment Conformal Prediction via Diffusion and Flow Matching Models
- 2026Flow Matching is Adaptive to Manifold Structures
- 2024Categorical Flow Matching on Statistical Manifolds
- 2026Diffusion Image Generation with Explicit Modeling of Data Manifold Geometry
- 2026What Matters for Diffusion-Friendly Latent Manifold? Prior-Aligned Autoencoders for Latent Diffusion
- 2025Be Tangential to Manifold: Discovering Riemannian Metric for Diffusion Models
- 2026RiT: Vanilla Diffusion Transformers Suffice in Representation Space
- 2026Beckmann Transport Models: From Autonomous Flows to One-Step Maps