语言模型需要“睡眠”来巩固记忆

Long-running language agents may work better if they periodically stop to consolidate memory.

The problem is that today’s transformer agents get slower and more expensive as their context grows, because attention has to keep checking more past tokens.

The usual fix for long context is to keep more tokens nearby, but that turns every next-token prediction into a larger search through the past.

The sharper idea here is that memory is not only storage.

Sometimes the hard part is converting a messy stretch of experience into a state that can actually be used later.

So the paper’s idea is to add a sleep phase, where the model pauses, rereads recent context several times, writes the useful information into fixed-size memory layers, and then clears the short-term attention cache.

During sleep, the model runs several offline passes over recent context, writes the result into fast weights inside its state-space blocks, then clears the attention cache.

This means the model pays extra compute while sleeping, not while answering, so normal prediction can still happen with 1 forward pass.

The authors test this on cellular automata, graph lookup, and GSM-Infinite math problems, where the model must use old information that is no longer sitting in its attention cache.

The main result is that longer sleep improves performance, especially on harder cases that need deeper reasoning rather than just remembering a fact.

The big deal is that long-horizon agents may not need to carry bigger and bigger raw context forever, because they can consolidate the important parts and safely forget the raw tokens.

Link – arxiv. org/abs/2605.26099

Title: "Language Models Need Sleep"