扩散模型记忆机制被破解!损失曲率无需采样即可预测训练数据反刍起点
Loss Curvature as a Sampling-Free Proxy for Memorization in Diffusion Models
arXiv LG (cs.LG) 🕐 今天 12:00

📖 AI 总结

该研究探讨扩散模型记忆训练数据的机制,从模型层面和样本层面两个视角展开分析。作者提出,输入损失曲率——即扩散模型训练损失Hessian矩阵的Frobenius范数——可作为量化记忆程度的有效指标。实验表明,在训练过程中追踪训练集与验证集样本的平均输入曲率,能够识别模型何时开始复现训练数据。然而,该指标不足以衡量单个样本被记忆的可能性。为此,作者引入扩散模型中记忆的反事实定义,将监督学习中的相关概念加以推广,并从理论上证明该反事实记忆受留一模型输入曲率的控制,从而在样本局部损失景观与其生成可能性之间建立了直接的理论联系。研究通过多项实验验证了所推导的边界,表明依赖输入的曲率可作为记忆的有效代理指标。这一工作为无需采样即可评估扩散模型记忆风险提供了新的理论工具与实证依据。

🔑 关键词速览

输入损失曲率 (Input Loss Curvature)扩散模型训练损失关于输入的 Hessian 矩阵的 Frobenius 范数,用于量化损失景观的局部曲率。
反事实记忆 (Counterfactual Memorization)从监督学习扩展至扩散模型的定义,衡量单个样本被模型记忆的倾向,基于留一法模型的变化。
留一法模型 (Leave-One-Out Model)在训练集中排除某个样本后重新训练的模型,用于评估该样本对模型行为的影响。
扩散模型 (Diffusion Models)一类生成模型,通过逐步去噪过程从噪声中生成数据,训练时学习反转扩散过程。
Frobenius 范数 (Frobenius Norm)矩阵所有元素平方和的平方根,常用于衡量矩阵的大小或距离。
infoAI 公众号二维码

📱 每天一份 AI 前沿日报

关注公众号,每天 09:00 推送 · 不错过任何重磅