Google AI Learns to ‘Sleep’ to Retain Knowledge

0
32

Google Research has introduced an innovative development called the “Sleep Framework”, designed to address a fundamental problem plaguing current large language models (LLMs): the loss of knowledge relevance after training concludes.

The Problem of ‘Expiring Knowledge’

Currently, most advanced LLMs enter a static mode after their initial training. This means their knowledge has a certain “expiration date.” If one attempts to supplement a model with new data during operation (e.g., through context), this information is often lost once the session ends. Direct retraining on new data, in turn, can lead to “catastrophic forgetting” – the model masters new skills but significantly loses its old ones.

Google AI навчився "спати", щоб не забувати знання

‘Sleep’ as a Solution

Google developers drew inspiration from the memory consolidation mechanisms observed during human sleep. The “Sleep Framework” divides the model’s operational process into two phases: “waking” and “sleeping.”

  • “Waking” Phase: The model actively perceives and processes incoming data, storing it in temporary (short-term) memory.
  • “Sleeping” Phase: Processing of new input data is paused. Instead, the model focuses on integrating acquired information, self-improvement, and solidifying knowledge in its long-term memory.

How It Works: ‘Consolidation’ and ‘Dreams’

“Memory consolidation” within the framework is achieved through expanding the model’s parameters, utilizing hybrid expert memory blocks, and employing knowledge distillation techniques. This allows fragile contextual information to be transferred to long-term parameters without overwriting existing knowledge.

The model’s “dreams” involve the generation and selection of synthetic training samples. Through reinforcement learning, the model is trained repeatedly on these samples, enhancing its reasoning and generalization capabilities.

Impressive Results

Experimental trials have demonstrated that the “Sleep Framework” significantly outperforms existing baseline methods in a range of complex tasks:

  • Continuous Translation: Models exhibit high efficiency.
  • Long Context Understanding: The system successfully processes contexts up to 10 million tokens in length.
  • Acquisition of New Facts: High accuracy is maintained when working with languages the model has not previously encountered, with minimal performance degradation.
  • Mathematical Logic: Superior accuracy has been achieved on mathematical benchmarks such as AIME 24/25.

This new approach from Google Research promises to make large language models more flexible, adaptive, and capable of continuous learning without losing prior knowledge, marking a crucial step toward creating more powerful and versatile artificial intelligence systems.

LEAVE A REPLY

Please enter your comment!
Please enter your name here