用霍普菲尔德能量分析生成模型持续学习中的遗忘机制。
Continual Learning in Modern Hopfield Networks with an Application to Diffusion Models

- 通过霍普菲尔德能量定义内在遗忘,量化任务切换后的记忆损失。
- 高能量异常样本比聚类样本更易遗忘,且重放对高能量样本效果更显著。
- 适用于扩散模型的持续学习优化,指导高效样本重放策略。
生成模型(包括扩散模型)作为基础模型,常通过序列微调进行适应,使得持续学习成为关键问题。然而,这类模型在持续学习中的遗忘机制仍不清晰:任务切换后,哪些已学分布特征最易丢失?应优先重放哪些样本?本文基于现代霍普菲尔德网络(MHNs)的能量函数进行分析,发现任务切换后能量上升可表征内在遗忘。在可解析的MHN设置中,我们证明高能量、异常样式的样本比聚类样例经历更大的能量增加,表明位于陡峭孤立势阱中的样本更易被遗忘。进一步分析显示,重放对高能量样本特别有效,支持基于能量选择重放样本。我们在MHN及两个扩散模型(Stable Diffusion与像素空间DDPM)上验证了上述预测:霍普菲尔德能量能追踪重建误差带来的遗忘,重放实验也揭示了与MHN分析一致的能量依赖性遗忘缓解效果。
原文摘要 · Abstract (English)
Generative models, including diffusion models, are increasingly used as foundation models and adapted through sequential fine-tuning, making continual learning an essential problem setting. However, continual learning in such generative models remains poorly understood: after a task change, what aspects of the learned distribution are most easily lost, and what replay samples should be prioritized? We address these questions through the modern Hopfield energy. Recent links between modern Hopfield networks (MHNs) and diffusion models allow analyses in MHNs to be transferred to diffusion models. We introduce intrinsic forgetting as an increase in Hopfield energy after the task change. In tractable settings in an MHN, we prove that high-energy, outlier-like samples undergo a larger energy increase than cluster-like samples, implying that samples located in sharp, isolated basins are more forgettable. We further analyze memory replay and show that replay is particularly effective for high-energy samples, enabling an energy-based selection of replay samples. We validate these predictions in experiments on MHNs and two diffusion models under continual-learning settings: Stable Diffusion and a pixel-space DDPM. In these diffusion models, Hopfield energy tracks reconstruction-based forgetting, and replay experiments reveal energy-dependent mitigation of forgetting that is consistent with the MHN analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。