跳到正文
arXiv · cs.LG· Shashank Sule, Xinyue Yu, Boonam Shin, Yuxuan Liu, Abolfazl Hashemi, Hayden Schaeffer·· 3 小时前

扩散模型记忆化新指标:输入损失曲率可作为无采样代理

Loss Curvature as a Sampling-Free Proxy for Memorization in Diffusion Models

AI 导读

研究者提出用输入损失曲率(扩散模型训练损失 Hessian 的 Frobenius 范数)量化扩散模型的记忆化现象。训练与验证样本的平均曲率可识别模型何时开始复述训练数据,但不足以衡量单个样本被记忆的概率。为此作者引入扩散模型的反事实记忆化定义,并证明其受留一模型输入曲率控制,建立了样本局部损失景观与生成概率的理论联系。

来源:arXiv · cs.LG · arxiv.org