Mirror Flow Matching with Heavy-Tailed Priors for Generative Modeling on Convex Domains
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
We study generative modeling on convex domains using flow matching and mirror maps, and identify two fundamental challenges. First, standard log-barrier mirror maps induce heavy-tailed dual distributions, leading to ill-posed dynamics. Second, coupling with Gaussian priors performs poorly when matching heavy-tailed targets. To address these issues, we propose Mirror Flow Matching based on a regularized mirror map that controls dual tail behavior and guarantees finite moments, together with coupling to a Student-t prior that aligns with heavy-tailed targets and stabilizes training. We provide theoretical guarantees, including spatial Lipschitzness and temporal regularity of the velocity field, Wasserstein convergence rates for flow matching with Student-t priors and primal-space guarantees for constrained generation, under varepsilon-accurate learned velocity fields. Empirically, our method outperforms baselines in synthetic convex-domain simulations and achieves competitive sample quality on real-world constrained generative tasks.
Key facts. Cited by 8 papers in the map (data through 14 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 30 of its 36 references that are in the map. First posted on arXiv 10 Oct 2025.
Builds on
30 of 36 references.
Closest ancestors · 9
Same direction, newest first
- 2025Constraint-Aware Flow Matching via Randomized Exploration18 Aug
- 2025Physics-Constrained Flow Matching: Sampling Generative Models with Hard Constraints4 Jun
- 2024On the Convergence and Straightness of Rectified Flow19 Oct
- 2024Improving the Training of Rectified Flows30 May
- 2024Reflected Flow Matching26 May
- 2023Flow Matching on General Geometries7 Feb
- 2023Improving and generalizing flow-based generative models with minibatch optimal transport1 Feb
- 2022Flow Matching for Generative Modeling6 Oct
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep
Other references · 21
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
- 2022Building Normalizing Flows with Stochastic Interpolantscited by 995
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
- 2023Rethinking FID: Towards a Better Evaluation Metric for Image Generationcited by 310
- 2023The probability flow ODE is provably fastcited by 146
- 2024FlowPolicy: Enabling Fast and Robust 3D Flow-based Policy via Consistency Flow Matching for Robot Manipulationcited by 104
- 2023Error Bounds for Flow Matching Methodscited by 91
- 2024Constrained Synthesis with Projected Diffusion Modelscited by 89
- 2023Reflected Diffusion Modelscited by 87
- 2023Diffusion Models for Constrained Domainscited by 63
- 2023Mirror Diffusion Models for Constrained and Watermarked Generationcited by 42
- 2024Heavy-Tailed Diffusion Modelscited by 40
9 more
- 2021The Mirror Langevin Algorithm Converges with Vanishing Biascited by 37
- 2023Metropolis Sampling for Constrained Diffusion Modelscited by 28
- 2024Evaluating the design space of diffusion-based generative modelscited by 26
- 2022Mirror Descent Strikes Again: Optimal Stochastic Convex Optimization under Infinite Noise Variancecited by 24
- 2024Constrained Diffusion Models via Dual Trainingcited by 20
- 2023Gaussian Interpolation Flowscited by 17
- 2024Preference Alignment with Flow Matchingcited by 11
- 2025Non-asymptotic Analysis of Diffusion Annealed Langevin Monte Carlo for Generative Modellingcited by 11
- 2024Neural Approximate Mirror Maps for Constrained Diffusion Modelscited by 9
Built on by
Citing papers per year: 2025: 2; 2026: 6.
8 papers.
- 2025Discrete State Diffusion Models: A Sample Complexity Perspectivecited by 5
- 2026Low-Dimensional Adaptation of Rectified Flow: A Diffusion and Stochastic Localization Perspectivecited by 3
- 2026Do Heavy Tails Help Diffusion? On the Subtle Trade-off Between Initialization and Trainingcited by 3
- 2025Generative Modeling with Continuous Flows: Sample Complexity of Flow Matchingcited by 2
- 2026Debiased Counterfactual Generation via Flow Matching from Observationscited by 1
- 2026PolyFlow: Safe and Efficient Polytope-Constrained Flow Matching with Constraint Embedding and Projection-free Updatecited by 1
- 2026Fine-Tuning Generative Models for Extreme Events via CVaR-Penalized Wasserstein Gradient Flowscited by 1
- 2026Manifold-Stable Flow Matchingcited by 0
- 2026Manifold-Stable Flow Matching28 Sep 2026
- 2026Fine-Tuning Generative Models for Extreme Events via CVaR-Penalized Wasserstein Gradient Flows12 Aug 2026
- 2026PolyFlow: Safe and Efficient Polytope-Constrained Flow Matching with Constraint Embedding and Projection-free Update11 Jun 2026
- 2026Do Heavy Tails Help Diffusion? On the Subtle Trade-off Between Initialization and Training13 May 2026
- 2026Debiased Counterfactual Generation via Flow Matching from Observations8 May 2026
- 2026Low-Dimensional Adaptation of Rectified Flow: A Diffusion and Stochastic Localization Perspective21 Jan 2026
- 2025Generative Modeling with Continuous Flows: Sample Complexity of Flow Matching1 Dec 2025
- 2025Discrete State Diffusion Models: A Sample Complexity Perspective12 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplings
- 2026Uncertainty in a Single Pass: A Closed-Form Identity for One-Step Flow Matching
- 2025Theoretical Guarantees for High Order Trajectory Refinement in Generative Flows
- 2024Flow map matching with stochastic interpolants: A mathematical framework for consistency models
- 2025Source-Guided Flow Matching
- 2025Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling
- 2025Reweighted Flow Matching via Unbalanced OT for Label-free Long-tailed Generation
- 2025A Theoretical Analysis of Discrete Flow Matching Generative Models