提出流模型在流形上采样的精确误差分析方法
Total Variation Rates for Riemannian Flow Matching
- 基于微分不等式构建流形上概率流的总变差控制框架
- 首次给出显式误差界,分离出数值离散与学习误差项
- 适用于球面和对称正定流形,可推导具体迭代复杂度
黎曼流匹配(RFM)将基于流的生成建模扩展到流形上的数据,通过学习一个时变切向量场,其流-微分方程将简单先验分布转化为数据分布。本文建立了非渐近的总变差(TV)收敛性分析,针对使用学习向量场与欧拉离散化在流形上的采样器。核心技术是建立两个流形微分方程流之间总变差演化的微分不等式,其时间导数由向量场偏差的散度与参考流的得分决定;控制这些项需引入考虑平行移动与曲率的新界。在群体流匹配场光滑性假设下,结合学习场的统一(紧致流形)或均方(哈达玛德流形)逼近保证,获得形式为 \mathrm{TV} \le C_{\mathrm{Lip}}\,h + C_{\varepsilon}\,\varepsilon(紧致流形上额外有 \varepsilon^2 高阶项)的显式误差界。实例化后,在超球面 $S^d$ 及 SPD$(n)$ 流形上,在温和矩条件下可得明确多项式迭代复杂度。
原文摘要 · Abstract (English)
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 $\mathrm{TV}\le C_{\mathrm{Lip}}\,h + C_{\varepsilon}\,\varepsilon$ (with an additional higher-order $\varepsilon^2$ term on compact manifolds), cleanly separating numerical discretization and learning errors. Here, $h$ is the step-size and $\varepsilon$ is the target accuracy. Instantiations yield \emph{explicit} polynomial iteration complexities on the hypersphere $S^d$, and on the SPD$(n)$ manifolds under mild moment conditions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。