English

Total Variation Rates for Riemannian Flow Matching

Machine Learning 2026-02-06 v1 Artificial Intelligence Machine Learning Statistics Theory Statistics Theory

Abstract

Riemannian flow matching (RFM) extends flow-based generative modeling to data supported on manifolds by learning a time-dependent tangent vector field whose flow-ODE transports a simple base distribution to the data law. We develop a nonasymptotic Total Variation (TV) convergence analysis for RFM samplers that use a learned vector field together with Euler discretization on manifolds. Our key technical ingredient is a differential inequality governing the evolution of TV between two manifold ODE flows, which expresses the time-derivative of TV through the divergence of the vector-field mismatch and the score of the reference flow; controlling these terms requires establishing new bounds that explicitly account for parallel transport and curvature. Under smoothness assumptions on the population flow-matching field and either uniform (compact manifolds) or mean-square (Hadamard manifolds) approximation guarantees for the learned field, we obtain explicit bounds of the form TVCLiph+Cεε\mathrm{TV}\le C_{\mathrm{Lip}}\,h + C_{\varepsilon}\,\varepsilon (with an additional higher-order ε2\varepsilon^2 term on compact manifolds), cleanly separating numerical discretization and learning errors. Here, hh is the step-size and ε\varepsilon is the target accuracy. Instantiations yield \emph{explicit} polynomial iteration complexities on the hypersphere SdS^d, and on the SPD(n)(n) manifolds under mild moment conditions.

Keywords

Cite

@article{arxiv.2602.05174,
  title  = {Total Variation Rates for Riemannian Flow Matching},
  author = {Yunrui Guan and Krishnakumar Balasubramanian and Shiqian Ma},
  journal= {arXiv preprint arXiv:2602.05174},
  year   = {2026}
}
R2 v1 2026-07-01T09:37:02.056Z