English

Understanding Generalization in Diffusion Distillation via Probability Flow Distance

Machine Learning 2026-02-13 v3 Computer Vision and Pattern Recognition

Abstract

Diffusion distillation provides an effective approach for learning lightweight and few-steps diffusion models with efficient generation. However, evaluating their generalization remains challenging: theoretical metrics are often impractical for high-dimensional data, while no practical metrics rigorously measure generalization. In this work, we bridge this gap by introducing probability flow distance (\texttt{PFD}), a theoretically grounded and computationally efficient metric to measure generalization. Specifically, \texttt{PFD} quantifies the distance between distributions by comparing their noise-to-data mappings induced by the probability flow ODE. Using \texttt{PFD} under the diffusion distillation setting, we empirically uncover several key generalization behaviors, including: (1) quantitative scaling behavior from memorization to generalization, (2) epoch-wise double descent training dynamics, and (3) bias-variance decomposition. Beyond these insights, our work lays a foundation for generalization studies in diffusion distillation and bridges them with diffusion training.

Keywords

Cite

@article{arxiv.2505.20123,
  title  = {Understanding Generalization in Diffusion Distillation via Probability Flow Distance},
  author = {Huijie Zhang and Zijian Huang and Siyi Chen and Jinfan Zhou and Zekai Zhang and Peng Wang and Qing Qu},
  journal= {arXiv preprint arXiv:2505.20123},
  year   = {2026}
}

Comments

41 pages, 15 figures

R2 v1 2026-07-01T02:39:59.338Z