提出数据分布相变理论,指导扩散模型用局部网络更高效生成图像
Local Diffusion Models and Phases of Data Distributions
- 基于统计物理定义数据分布相,区分不同阶段的去噪需求
- 发现去噪过程存在快速相变,此时局部去噪器失效
- 验证局部马尔可夫性可诊断相变,适合优化模型架构
受统计物理启发,扩散模型通过逐步去噪生成复杂数据分布。真实数据如图像在低维空间中具有局部结构,但传统扩散模型忽略此特性,学习全局得分函数,计算成本高。本文受非平衡统计物理进展启发,提出数据分布相的通用框架,用于分析扩散模型中去噪器的局部性要求。若两个分布可通过局部操作(如局部去噪)沿扩散演化路径相互连接,则属于同一数据分布相。我们证明反向去噪过程包含早期平凡相、晚期数据相,以及两者之间的快速相变阶段,该阶段局部去噪器必须失效。进一步表明,局部去噪性能与空间马尔可夫性密切相关,为此提供了可操作的相变诊断标准。在真实数据集上的数值实验验证了该准则。研究结果为设计更简单高效的扩散模型提供指导:远离相变点时可用小型局部神经网络计算得分函数;仅在相变窄时间区间内需全局网络。这为研究数据分布相、生成式人工智能的广义科学,以及受物理启发的神经网络设计开辟新方向。
原文摘要 · Abstract (English)
As a class of generative artificial intelligence frameworks inspired by statistical physics, diffusion models have shown extraordinary performance in synthesizing complicated data distributions through a denoising process gradually guided by score functions. Real-life data, like images, is often spatially structured in low-dimensional spaces. However, ordinary diffusion models ignore this local structure and learn spatially global score functions, which are often computationally expensive. In this work, motivated by recent advances in non-equilibrium statistical physics, we develop a generic framework for defining phases of data distributions and use it to analyze the locality requirements of denoisers in diffusion models. We define two distributions as belonging to the same data distribution phase if they can be mutually connected via spatially local operations such as local denoisers, along the same evolution path as the diffusion. We demonstrate that the reverse denoising process consists of an early trivial phase and a late data phase, sandwiching a rapid phase transition where local denoisers must fail. We further demonstrate that the performance of local denoisers is closely tied to spatial Markovianity, which provides an operational criterion for diagnosing such phase transitions. We validate this criterion through numerical experiments on real-world datasets. Our work suggests guidance for simpler and more efficient architectures of diffusion models: far from the phase transition point, we can use small local neural networks to compute the score function; global neural networks are only necessary around the narrow time interval of phase transitions. This result also opens up new directions for studying phases of data distributions, the broader science of generative artificial intelligence, and guiding the design of neural networks inspired by physics concepts.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。