arXiv:2508.17689cs.LGstat.ML2025-08NeurIPS被引 23

揭示扩散模型记忆与泛化之间的临界点,预测模型规模何时开始复制训练数据。

On the Edge of Memorization in Diffusion Models

  • 通过对比记忆型与泛化型模型的损失差,构建理论框架分析模型行为
  • 发现模型参数量在临界点时发生记忆主导的相变,实验验证预测准确性
  • 为版权和隐私问题提供理论依据,适合关注模型安全性的研究者

扩散模型何时复现训练数据,何时能生成新样本?这一记忆与泛化的权衡对版权和数据隐私有重要影响。本文构建一个基于合成或自然图像结构数据的可控制实验环境,提出假设:欠参数模型的记忆或泛化行为由记忆型模型与泛化型模型之间的训练损失差异决定。理论上刻画了在关键参数量下,完全泛化模型的加权损失超过欠参数记忆模型的转折点。通过精心设计的实验,验证了该转折点对应梯度下降训练中扩散模型的相变行为。最终,理论可解析预测记忆占主导的模型尺寸。本工作为未来理论与实证研究提供了可解析且具实际意义的框架。代码见 https://github.com/DruvPai/diffusion_mem_gen。

原文摘要 · Abstract (English)

When do diffusion models reproduce their training data, and when are they able to generate samples beyond it? A practically relevant theoretical understanding of this interplay between memorization and generalization may significantly impact real-world deployments of diffusion models with respect to issues such as copyright infringement and data privacy. In this work, to disentangle the different factors that influence memorization and generalization in practical diffusion models, we introduce a scientific and mathematical "laboratory" for investigating these phenomena in diffusion models trained on fully synthetic or natural image-like structured data. Within this setting, we hypothesize that the memorization or generalization behavior of an underparameterized trained model is determined by the difference in training loss between an associated memorizing model and a generalizing model. To probe this hypothesis, we theoretically characterize a crossover point wherein the weighted training loss of a fully generalizing model becomes greater than that of an underparameterized memorizing model at a critical value of model (under)parameterization. We then demonstrate via carefully-designed experiments that the location of this crossover predicts a phase transition in diffusion models trained via gradient descent, validating our hypothesis. Ultimately, our theory enables us to analytically predict the model size at which memorization becomes predominant. Our work provides an analytically tractable and practically meaningful setting for future theoretical and empirical investigations. Code for our experiments is available at https://github.com/DruvPai/diffusion_mem_gen.

扩散模型记忆机制模型泛化理论分析

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