English

Geometric Regularity in Deterministic Sampling Dynamics of Diffusion-based Generative Models

Machine Learning 2025-12-12 v3 Statistical Mechanics Computer Vision and Pattern Recognition Machine Learning

Abstract

Diffusion-based generative models employ stochastic differential equations (SDEs) and their equivalent probability flow ordinary differential equations (ODEs) to establish a smooth transformation between complex high-dimensional data distributions and tractable prior distributions. In this paper, we reveal a striking geometric regularity in the deterministic sampling dynamics of diffusion generative models: each simulated sampling trajectory along the gradient field lies within an extremely low-dimensional subspace, and all trajectories exhibit an almost identical boomerang shape, regardless of the model architecture, applied conditions, or generated content. We characterize several intriguing properties of these trajectories, particularly under closed-form solutions based on kernel-estimated data modeling. We also demonstrate a practical application of the discovered trajectory regularity by proposing a dynamic programming-based scheme to better align the sampling time schedule with the underlying trajectory structure. This simple strategy requires minimal modification to existing deterministic numerical solvers, incurs negligible computational overhead, and achieves superior image generation performance, especially in regions with only 5 - 10 function evaluations.

Keywords

Cite

@article{arxiv.2506.10177,
  title  = {Geometric Regularity in Deterministic Sampling Dynamics of Diffusion-based Generative Models},
  author = {Defang Chen and Zhenyu Zhou and Can Wang and Siwei Lyu},
  journal= {arXiv preprint arXiv:2506.10177},
  year   = {2025}
}

Comments

57 pages. Accepted by Journal of Statistical Mechanics: Theory and Experiment (2025). The short version was published in ICML 2024. arXiv admin note: text overlap with arXiv:2405.11326

R2 v1 2026-07-01T03:12:09.414Z