提出扩散模型收敛速度新理论,证明可自动适应低维数据结构。
O(d/T) Convergence Theory for Diffusion Probabilistic Models under Minimal Assumptions
- 基于最小假设,分析反向扩散过程的误差传播机制。
- 在高维数据下,生成分布与目标分布距离为 O(d/T)。
- 通过系数优化,可提升至 O(k/T),k为数据内在维度。
基于得分的扩散模型通过学习逆转将目标分布逐步扰动为噪声的扩散过程来生成新数据,在各类生成任务中表现卓越。尽管其经验性能优异,现有理论分析常依赖强假设或收敛速度较慢。本文针对广泛应用的SDE-based采样器DDPM,建立在最弱假设下的快速收敛理论。证明:若得分函数估计$\ ext{L}_2$精度足够,生成分布与目标分布的总变差距离上界为O(d/T),其中d为数据维度,T为步数,且该结论对任意具有有限一阶矩的目标分布成立。此外,通过精心设计系数,收敛率可进一步提升至O(k/T),k为数据分布的内在维度,表明DDPM能自动适应自然图像等数据的低维结构特征。该成果源于一套新颖的分析工具,可精细刻画反向过程中每一步的误差传播规律。
原文摘要 · Abstract (English)
Score-based diffusion models, which generate new data by learning to reverse a diffusion process that perturbs data from the target distribution into noise, have achieved remarkable success across various generative tasks. Despite their superior empirical performance, existing theoretical guarantees are often constrained by stringent assumptions or suboptimal convergence rates. In this paper, we establish a fast convergence theory for the denoising diffusion probabilistic model (DDPM), a widely used SDE-based sampler, under minimal assumptions. Our analysis shows that, provided $\ell_{2}$-accurate estimates of the score functions, the total variation distance between the target and generated distributions is upper bounded by $O(d/T)$ (ignoring logarithmic factors), where $d$ is the data dimensionality and $T$ is the number of steps. This result holds for any target distribution with finite first-order moment. Moreover, we show that with careful coefficient design, the convergence rate improves to $O(k/T)$, where $k$ is the intrinsic dimension of the target data distribution. This highlights the ability of DDPM to automatically adapt to unknown low-dimensional structures, a common feature of natural image distributions. These results are achieved through a novel set of analytical tools that provides a fine-grained characterization of how the error propagates at each step of the reverse process.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。