Hraness
Theme
Appearance

saved

The sleep framework: A new way for LLMs to retain knowledge in continual learning

by Ali Behrouz and Vahab MirrokniGoogle Researchpublished

Hraness wrote this summary from a saved copy of the source. Quotations are taken word for word from the source.

gist

Google Research's sleep framework gives LLMs offline phases that move fragile in-context memory into new long-term parameters. The aim is learning without forgetting. Consolidation distills short-term memory into newly unlocked low-rank experts. A dreaming stage then rehearses on curated synthetic data from randomly routed experts. The authors report gains in continual learning, 10M-token BABILong recall, knowledge incorporation and math, with modest margins on math.

ideas

  • Learning gets a wake and sleep cycle. During the active phase the model stores new information in fast short-term memory; during sleep it stops taking input and spends its compute on consolidation and self-improvement.
  • Consolidation grows capacity instead of overwriting. Low-rank MoE experts are unlocked in slower memory levels and seeded from short-term memory with on-policy distillation plus RL-based imitation learning. The temporary high-frequency parameters are then reset.
  • Dreaming is curated self-training. Randomly routed experts generate synthetic samples, gradient-based selection keeps the most useful ones, and the model fine-tunes on those that improve downstream performance.
  • The reported gains are broad but uneven. Sleep tops class-incremental baselines (97 vs 93 for Hope on CLINC with Llama3-8B) and keeps both new languages in sequential translation. On Qwen3-8B math it scores 79.2 / 69.0 / 46.1 on AIME-24/25 and HMMT-25, against 76.6 / 67.4 / 45.1 for the best baseline.
  • The substance is the arXiv paper. Language Models Need Sleep (Behrouz, Hashemi, Javanmard, Mirrokni) has been on arXiv since June 2026, and on OpenReview since September 2025. All results are the authors' own.

quotes

“Traditional machine learning relies on a strict separation between training and inference.”

Ali Behrouz and Vahab Mirrokni, Google Research.

“The model is not asked to copy every detail of its short-term traces into its parameters.”

Google Research, on knowledge seeding.

“our framework operates entirely within a fixed model architecture and human-defined training rules.”

Google Research, on the neuroscience framing.