English

Mirror Flow Matching with Heavy-Tailed Priors for Generative Modeling on Convex Domains

Machine Learning 2025-10-13 v1 Machine Learning

Abstract

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 \emph{regularized mirror map} that controls dual tail behavior and guarantees finite moments, together with coupling to a Student-tt 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-tt 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.

Keywords

Cite

@article{arxiv.2510.08929,
  title  = {Mirror Flow Matching with Heavy-Tailed Priors for Generative Modeling on Convex Domains},
  author = {Yunrui Guan and Krishnakumar Balasubramanian and Shiqian Ma},
  journal= {arXiv preprint arXiv:2510.08929},
  year   = {2025}
}