现代大语言模型(LLM)通过预训练和上下文学习(ICL)实现了令人印象深刻的性能。然而,一旦部署,它们基本上保持静态,以固定的知识截止日期运行,这使得它们无法获得长期能力或适应新信息。
虽然用新数据更新模型似乎是显而易见的解决方案,但持续的参更新往往会引发灾难性遗忘,这是一个根本性问题,即学习新任务会严重降低先前学习任务的性能。或者,严格依赖ICL会将知识保留限制在即时上下文窗口内:一旦上下文会话结束,所获得的短期信息就会被遗忘。
受人类大脑在休息时如何处理和稳定记忆的启发,我们在持续学习中引入了一个睡眠框架:一种学习范式,使大语言模型(LLM)具有连续的周期性生命周期,使它们能够持续学习,将脆弱的短期上下文巩固为稳定的长期参数。通过这个框架,我们证明了在主动交互和离线记忆巩固之间交替,允许LLM持续获取新事实,保留长达10M令牌的长上下文知识,并提高复杂推理能力。
重新思考机器学习:主动(清醒)与睡眠阶段
传统机器学习依赖于训练和推理之间的严格分离。在持续学习框架中,这种区分变得人为;智能代理必须不断学习、适应和处理信息。
在睡眠范式下,LLM在两个主要状态下运行:
- 主动(清醒)阶段:模型主动接收、处理输入并与之交互,将临时信息存储在快速、短期记忆组件中。
- 睡眠阶段:模型暂停处理新输入,并将其内部计算转向记忆巩固和自我改进。
虽然“睡眠”、“做梦”和“自我改进”等术语借自神经科学,但我们的框架完全在固定的模型架构和人类定义的训练规则内运行。在离线睡眠阶段,模型解锁预分配的多专家(MoE)记忆块,并使用自生成的练习示例更新其数值权重。这种自动化的后训练循环帮助已部署的模型吸收新事实并提高其推理准确性,同时保持其核心代码库和设计不变。
睡眠过程的关键组件
睡眠范式通过两个阶段解决灾难性遗忘和持续适应:记忆巩固和做梦。
1. 记忆巩固
在人类大脑中,非快速眼动(NREM)慢波睡眠通过抽象和蒸馏将脆弱的记忆巩固为长期结构。我们提出的睡眠框架通过两个关键技术机制反映了这一自然现象:
- 参数扩展与连续记忆层次: 为了防止灾难性遗忘,模型在睡眠期间动态扩展其容量。使用多专家(MoE)和Hope架构,新解锁的低秩专家被集成到稳定的低频记忆块中,以存储传入知识而不破坏旧参数。
- 知识播种(向上蒸馏): 存储在高频层(例如,上下文记忆)中的脆弱短期记忆被转移到新扩展的长期参数中。这是通过广义知识蒸馏(GKD)结合基于强化学习(RL)的模仿学习实现的:
- 同策略蒸馏: 扩展的学生模型从在自采样序列上生成的教师分布中学习令牌特定的反馈。
- 学习模仿(LTI): 当完成教师生成文本的前缀时,学生模型根据语义和绝对相似性(Levenshtein距离)获得奖励。
- 突触修剪: 一旦记忆巩固到低频模块完成,高频临时参数被重置,为未来的主动会话释放容量。

2. 做梦以自我改进
在快速眼动睡眠期间,人类大脑主动模拟场景,巩固情感/语义连接,并探索新的神经通路。与此对应,在“做梦阶段”,睡眠框架实现了一个自主的自我改进循环:
- 随机专家选择: 在合成数据生成期间,MoE路由器随机激活非标准专家。这迫使模型混合不同领域的知识并揭示潜在模式。
- 策划排练: 模型生成合成训练样本(“梦”),并使用基于梯度的数据选择过滤它们,以优先考虑最有可能提升整体模型能力的样本。
- RL细化: 模型对评分最高的梦样本进行监督微调,并评估生成是否改善了下游任务的性能。
理论基础
除了实证结果,我们的睡眠范式基于一个理论,该理论阐明了为什么它的两个阶段(记忆巩固和做梦)可以支持持续学习,而不仅仅是牺牲旧能力换取新知识。
对于记忆巩固,分析表明知识播种可以理解为一个结构化的低秩压缩问题。模型不被要求将其短期痕迹的每一个细节复制到参数中。相反,它识别出一组紧凑的方向,这些方向最能同时保留模型自身的同策略行为和由学习目标所奖励的语义质量。
这一视角解释了两种信号的互补作用:同策略蒸馏使巩固与模型在推理时实际产生的轨迹对齐,减少了仅基于教师生成分布训练时可能出现的失配;同时,语义奖励改变了值得保留的内容,优先考虑保留含义和任务相关行为的方向,而不仅仅是匹配词元概率。参数扩展则提供了写入这些选定记忆所需的能力,而不会覆盖先前已巩固的知识。
对于做梦,该理论将随机专家激活和基于奖励的样本选择描述为改善学习几何的机制。注入不同的MoE拓宽了在合成生成过程中探索的梯度方向集合,使模型接触到比仅由其常规清醒时数据所诱导的更丰富的更新集合。基于梯度的过滤随后强调携带最有用的学习信号的合成经验,加强信息丰富的方向,同时保留底层梯度空间的结构。
这些步骤共同产生了一个更平衡的优化景观:学习较少被少数方向主导,梯度条件更好,模型在精炼其能力时能够取得更稳定的进展。在这一观点下,做梦不仅仅是自生成的数据增强;它是一个有意的过程,用于多样化、选择和巩固模型借以自我改进的更新。

对于持续翻译,在未见过的语言(如满语和Kalamang语)中,睡眠框架在多阶段评估中保持了接近零的性能下降,而ICL、Cartridges和监督微调(SFT)等标准基线在顺序学习中遭受了严重的灾难性遗忘。
在BABILong基准测试的长上下文理解中,我们的睡眠方法展示了高达10M词元的近乎完美的扩展,通过利用睡眠时巩固将长上下文信息直接压缩到参数记忆中,优于GPT-4和Titans等强基线。
对于在SQuAD和持续预训练(CPT)上评估的事实知识整合,我们的睡眠方法在单任务上达到48.9%,在CPT评估上达到46.2%,优于包括基础模型、SFT和自适应语言模型(SEAL)在内的基线方法,成功整合了新的事实信息而不降低现有模型表示。
最后,在包括AIME 24/25和HMMT 25在内的数学推理基准测试中,睡眠增强的基础模型分别达到了79.2%、69.0%和46.1%的准确率。这优于SFT、组相对策略优化(GRPO)和同策略自蒸馏(OPSD)等基线,表明将记忆巩固与模仿学习相结合可驱动卓越的推理性能。
结论
我们的睡眠范式为自主、持续学习的AI系统提供了一条路径。通过超越静态预训练和脆弱的上下文窗口,在主动学习和离线休息之间交替,使LLM能够保留短期记忆,将新技能抽象为长期参数结构,并在没有灾难性遗忘的情况下持续自我改进。
致谢
本研究 由Ali Behrouz(Google和康奈尔大学)、Farnoosh Hashemi(康奈尔大学)、Adel Javanmard(Google和南加州大学)和Vahab Mirrokni进行。我们感谢Yuan Deng和Corinna Cortes审阅本工作并提供宝贵建议。我们还感谢Mark Simborg、Kimberly Schwede和Tim Herrmann在撰写本博客文章方面的帮助。
