arXiv:2603.13419cs.LG2026-03被引 1

扩散模型通过模型误差实现泛化,避免过拟合。

Diffusion Models Memorize in Training -- and Generalize in Inference

  • 模型在训练中逐步过拟合去噪目标,但泛化能力来自误差导致的平滑流场。
  • 验证集与训练集性能差距在中间噪声水平最明显,且推理阶段不复现此差距。
  • 适合关注生成模型泛化机制的科研人员阅读。

扩散模型在实践中表现良好,但最优扩散模型会完全记忆训练数据,因此无法泛化,这引发了一个问题:真实扩散模型的泛化性源于何处?我们发现,尽管在样本层面具有泛化能力,扩散模型仍会逐步过拟合去噪训练目标,从而在验证和训练样本间产生泛化差距,该差距在中间噪声水平最为显著。通过一个全解析的有误玩具模型,我们追踪了影响泛化差距的因素。研究发现,最优去噪流场在训练点附近高度集中,但模型误差抑制了对训练点的精确回忆,形成平滑、可泛化的流场。最终,训练中观察到的泛化差距不会传递到推理阶段,因为采样轨迹的中间状态远偏离于模型训练所依赖的噪声训练样本分布。这些发现揭示了扩散模型泛化的全新机制:流场通过模型误差实现泛化,使采样轨迹脱离噪声训练样本域,从而自然防止过拟合。

原文摘要 · Abstract (English)

Diffusion models generalize well in practice. However, an optimal diffusion model fully memorizes the training data and therefore fails to generalize, raising the question of what induces generalization in a real diffusion model. We show that, despite generalizing at the sample level, diffusion models progressively overfit the denoising training objective and thereby create a generalization gap between the performance on validation and training samples. This gap is most pronounced at intermediate noise levels. Using a fully analytic error-prone toy model, we trace the factors affecting the generalization gap. We find that the optimal denoising flow field localizes sharply around training points, but the model error suppresses the exact recall of training points, yielding a smooth, generalizing flow field. Finally, we find that the generalization gap observed in training does not translate to inference, which would result in a strong similarity between generated samples and training samples. This is because the intermediate states of sampling trajectories are sufficiently far from the distribution of noisy training samples the model is trained on. Together, these findings reveal a novel picture of how diffusion models generalize: the flow field generalizes through model error, which moves sampling trajectories outside the domain of noisy training samples and thereby naturally prevents overfitting.

扩散模型泛化机制过拟合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。