arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.02426cs.LGcs.CV

打破维度诅咒:扩散模型高效学习低维分布

Breaking the Curse of Dimensionality: Diffusion Models Efficiently Learn Low-Dimensional Distributions

University of Science and Technology of China(中国科学技术大学)

Peng Wang, Huijie Zhang, Zekai Zhang, Siyi Chen, Yi Ma, Qing Qu

AI总结:

提出新数学框架,证明扩散模型通过等价于子空间聚类,能以线性于内在维度的样本复杂度学习低维分布,避免维度诅咒。

摘要:

尽管扩散模型在广泛的生成任务中取得了经验上的成功,但其学习数据分布能力的基本原理仍不清楚。在这项工作中,我们开发了一个新的数学框架,解释了扩散模型如何能够从有限数量的训练样本中有效学习低维分布,而不受维度诅咒的影响。具体来说,受图像数据内在低维结构的启发,我们在理论上分析了一个数据分布被建模为低秩高斯混合的场景。在合适的网络参数化下,我们表明优化扩散模型的训练目标等价于在训练样本上解决经典子空间聚类问题,其中每个子空间基对应于一个高斯分量的低秩协方差。这种等价性使我们能够证明,学习底层分布的样本复杂度与数据的内在维度呈线性关系,而不是与环境维度呈指数关系。我们的理论发现得到了经验证据的进一步支持,这些证据展示了在合成和真实世界图像数据集上的泛化相变现象。此外,我们建立了学习到的子空间基与图像数据语义属性之间的对应关系,为可控图像生成提供了原则性基础。

英文摘要:

Despite their empirical success across a wide range of generative tasks, the fundamental principles underlying the ability of diffusion models to learn data distributions are poorly understood. In this work, we develop a new mathematical framework that explains how diffusion models can effectively learn low-dimensional distributions from a finite number of training samples without suffering from the curse of dimensionality. Specifically, motivated by the intrinsic low-dimensional structure of image data, we theoretically analyze a setting in which the data distribution is modeled as a mixture of low-rank Gaussians. Under suitable network parameterization, we show that optimizing the training objective of diffusion models is equivalent to solving the canonical subspace clustering problem over the training samples, where each subspace basis corresponds to the low-rank covariance of a Gaussian component. This equivalence allows us to show that the sample complexity for learning the underlying distribution scales linearly with the intrinsic dimension of the data, rather than exponentially with the ambient dimension. Our theoretical findings are further supported by empirical evidence that demonstrates phase transition phenomena in generalization on both synthetic and real-world image datasets. Moreover, we establish a correspondence between the learned subspace bases and semantic attributes of image data, providing a principled foundation for controllable image generation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑