揭示扩散模型如何通过信息限制实现泛化,避免高维数据记忆。
An exact information theory of generalization phase transitions in Bayesian diffusion models

- 用贝叶斯后验逆推噪声图像的原始训练样本,实现信息受限的生成。
- 发现泛化与记忆的相变边界:当互信息超过训练集对数时会记忆。
- 适用于研究生成模型机制、理解早期训练阶段的泛化行为的人。
扩散模型如何在有限训练数据下,从高维空间中学习复杂分布而不只是记忆,仍是根本性难题。为此,我们提出可解析计算的贝叶斯信息受限扩散(BIRD)模型,其中每个像素仅观察到噪声数据的部分信息。该模型通过贝叶斯后验反向推断当前受限观测来自哪个历史训练样本。此模型类推广了已有使用局部信息限制的解析扩散模型。我们证明,在不同架构(如UNet和DiT)下,空间局部BIRD模型在训练初期能准确近似真实扩散模型。在对数据分布最小假设下,我们识别出训练数据量、生成反演时间与信息限制量共同决定的泛化-记忆相变边界:当受限观测与训练数据的互信息超过训练点数的对数时,模型进入记忆态;否则为泛化态。多数据集实验验证了理论预测的相变位置。结果表明,生成过程始终处于记忆边缘——空间局部BIRD模型与早期训练扩散模型均通过随时间增强信息限制来逼近相变边界。总体而言,我们的工作揭示了信息限制在生成式AI中克服维度诅咒的关键作用。
原文摘要 · Abstract (English)
How diffusion models circumvent the curse of dimensionality to learn complex distributions over high dimensional spaces from a finite training set, instead of memorizing it, remains a fundamental mystery. To address this, we introduce analytically tractable Bayesian information restricted diffusion (BIRD) models, in which each pixel observes restricted information about noisy data. A BIRD model time-reverses diffusion by inferring which past training sample produced its current restricted observation using the Bayesian posterior. This model class generalizes existing analytical diffusion models that use spatially local information restriction. We show that spatially local BIRD models closely approximate trained diffusion models \textit{early in training}, across different architectures such as UNets and DiTs. Under minimal assumptions on the data distribution, we identify an information-theoretic phase boundary between memorization and generalization in the joint space of amount of training data, time in the reverse generative process, and amount of information restriction: a BIRD model memorizes when the mutual information between its restricted noisy observations and the training data exceeds the log number of training points, and it generalizes otherwise. Experiments across a range of datasets confirm our theoretically predicted location for the transition. We find that generation proceeds near the edge of memorization: both spatially local BIRD models and early-training diffusion models track the memorization-generalization phase boundary by increasingly restricting information over time. Overall, our results reveal a fundamental role for information restriction in generative AI to circumvent the curse of dimensionality.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。