English

Convergence Analysis of Probability Flow ODE for Score-based Generative Models

Machine Learning 2025-04-22 v3 Numerical Analysis Classical Analysis and ODEs Numerical Analysis

Abstract

Score-based generative models have emerged as a powerful approach for sampling high-dimensional probability distributions. Despite their effectiveness, their theoretical underpinnings remain relatively underdeveloped. In this work, we study the convergence properties of deterministic samplers based on probability flow ODEs from both theoretical and numerical perspectives. Assuming access to L2L^2-accurate estimates of the score function, we prove the total variation between the target and the generated data distributions can be bounded above by O(d3/4δ1/2)\mathcal{O}(d^{3/4}\delta^{1/2}) in the continuous time level, where dd denotes the data dimension and δ\delta represents the L2L^2-score matching error. For practical implementations using a pp-th order Runge-Kutta integrator with step size hh, we establish error bounds of O(d3/4δ1/2+d(dh)p)\mathcal{O}(d^{3/4}\delta^{1/2} + d\cdot(dh)^p) at the discrete level. Finally, we present numerical studies on problems up to 128 dimensions to verify our theory.

Keywords

Cite

@article{arxiv.2404.09730,
  title  = {Convergence Analysis of Probability Flow ODE for Score-based Generative Models},
  author = {Daniel Zhengyu Huang and Jiaoyang Huang and Zhengjiang Lin},
  journal= {arXiv preprint arXiv:2404.09730},
  year   = {2025}
}

Comments

37 pages, 7 figures; To appear in IEEE Transactions on Information Theory

R2 v1 2026-06-28T15:54:31.206Z