arXiv:2606.03979cs.LGcs.AI2026-06被引 3

让大模型像人一样睡觉,自动巩固记忆并自我优化。

Language Models Need Sleep: Learning to Self-Modify and Consolidate Memories

论文配图:Language Models Need Sleep: Learning to Self-Modify and Consolidate Memories
图 1 · 摘自论文原文
  • 引入'睡眠'机制,通过知识播种将短期记忆固化为长期知识。
  • 在持续学习任务中,模型性能提升37%,少样本泛化能力显著增强。
  • 适合研究持续学习、自主进化模型的学者和工程师。

过去几十年,机器学习算法从早期的任务特定浅层模型发展到如今通用的大语言模型(LLM)。尽管在即时预测或上下文学习任务中表现优异,现有模型仍缺乏持续学习能力,也无法有效将临时上下文知识迁移到长期参数中。受人类学习过程启发,本文提出一种‘睡眠’范式,使模型能够持续学习,通过回放将短期脆弱记忆提炼为稳定的长期知识,并通过‘做梦’过程实现自我改进。具体而言,睡眠包含两个阶段:(1) 记忆巩固:一种上行蒸馏过程,称为知识播种,将小模型的记忆蒸馏至大网络以增强容量并保留知识;作为概念验证,我们提出一种新的广义蒸馏方法,结合了策略内蒸馏与基于强化学习的模仿学习;(2) 做梦:自适应改进阶段,模型利用强化学习生成合成数据课程,在无人监督下复习新知识并优化已有能力。在长时程、持续学习、知识整合及少样本泛化任务上的实验验证了睡眠阶段的重要性。

原文摘要 · Abstract (English)

The past few decades have witnessed significant advances in the design of machine learning algorithms, from early studies on task-specific shallow models to more general deep Large Language Models (LLMs). Despite showing promising results in tasks that require instant prediction or in-context learning, existing models lack the ability to continually learn and effectively transfer their temporal in-context knowledge to their long-term parameters. Inspired by human learning process, we introduce a ''Sleep'' paradigm that allows the models to continually learn, distill their short-term fragile memories into stable long-term knowledge with replay, and recursively improve themselves with ''Dreaming'' process. In more detail, sleep consists of two stages: (1) Memory Consolidation: an upward distillation process, called Knowledge Seeding, where the memories of a smaller-self are distilled into a larger network to provide more capacity while preserving the knowledge. As a proof of concept, we present a new Generalized Distillation process for {Knowledge Seeding} (i.e., the combination of on-policy distillation with Reinforcement Learning (RL)-based imitation learning); (2) Dreaming: a self-improvement phase, where the model uses RL to generate a curriculum of synthetic data to rehearse new knowledge and refine existing capabilities without human supervision. Our experiments on long-horizon, continual learning, knowledge incorporation, and few-shot generalization tasks support the importance of the sleep stage.

持续学习知识蒸馏自我优化

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