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

research.google

224 points by acossta 3 days ago


https://arxiv.org/pdf/2606.03979

gopalv - 3 days ago

> On-policy distillation: The expanded student model learns token-specific feedback from teacher distributions generated on self-sampled sequences.

The weird thing is to see this based off the KV-Cache in the diagrams.

Completely different from the Databricks RAFT [1].

My team built "Dreamtime" earlier this year [2], which used a different mechanism to distill out learnings across multiple sessions.

This didn't pull from the token layer, but from user nudges on unsatisfactory responses.

This works like a sawtooth wave where we accumulate context, check for dissonances (a user gets to say "It depends" and fix it), compress it back, retire the old one.

The audit requirements are the real problem for customers, where you need to be able to ask "When did this change? and who changed it?" which was hard to answer from the lowest layer of the model unless you can retain lineage to the original sessions.

A model we train can't serve as an accomplice to a user who wants to hide some financial crimes by nudging the agent repeatedly.

[1] - https://arxiv.org/abs/2403.10131

imimayj1337 - 3 days ago

I know we just had the carbon announcement which was distinctly confusing, but comparatively I think that this is an exceptionally well-written article. I do believe that Google has some exceptional scientific minds as illustrated by their ability to take highly technical concepts and explain them in an understandable way for the layperson.

As for the technical nature of this, I'm not sure if I've totally understood correctly despite how well it's written, but are they essentially holding back sort of unweighted parameters built in to the model architecture which they can then populate with new context during 'sleep' and then using the techniques outlined in the 'dream' section to reinforce what gets stored in those MoE weights? That to me seems like a solution which only Google is best placed to solve due to their fully integrated AI stack and historic experience with hardware optimisation, but which would be rather hard to do if you are a more resource and compute constrained organisation

deepwoods - 13 hours ago

I wonder how well this would translate with a much larger model. They ran the experiments with comparatively tiny models. You would presumably need to churn a lot more tokens to move the needle of a multi-trillion-parameter model's weights. It might just be that it's not that hard to improve Llama-3B and Llama 3-8B because they're not very good to begin with. They compare it to other methods - in-context learning and TTT - but I'd be curious to see it compared to a "control group" of RLHF.

As an aside, I get that this is a preprint, but it's very odd to see a piece of writing with obvious typos and grammar issues (e.g. page 13: "We attributes this results to," "In this setting (see Table 4), Sleep achieves a 80% success rate, higher then the other methods.") in an era with LLMs. It sort of locks in that this was written by human beings, and I'm all for that, but it jumps out at me more than it used to.

macleginn - 2 days ago

The actual paper: https://arxiv.org/pdf/2606.03979

Shacharp - 7 hours ago

I love that they tested it on Qwen 3. I’m curious if they’ve experimented with putting the new parameters onto system RAM or even an SSD. Big bummer they didn’t publish that part.

donbox - 5 hours ago

AFAIK Muse (from Meta) dreams as well... every night... to consolidate its knowledge graph.

macleginn - 2 days ago

Reminds me of one of my favourite ML papers, DreamCoder: https://dl.acm.org/doi/pdf/10.1145/3453483.3454080

shevis - 12 hours ago

> Parameter expansion then provides the capacity needed to write these selected memories without overwriting previously consolidated knowledge.

This seems like the limiting factor here, depending on efficiency

mdp2021 - 2 days ago

Good, this seems one of the crucial missing pieces towards AGI. Add critical thinking during cleanup, and we get very close to it.

amelius - 8 hours ago

Who can afford to sleep in this race to the bottom?

jaydayvis - 4 hours ago

But my dear friends, they were already dreaming

westurner - 9 hours ago

More on memory consolidation from neurosci and neurology:

- "New study shows highly creative people’s brains work differently from others'" (2022) https://news.ycombinator.com/item?id=41921738 ; and does this correlate for example like REM/non-REM and ADHD? How much does this topology vary amongst the population?

- "Discovery of Memory "Glue" Explains Lifelong Recall: KIBRA Protein, PKMzeta" (2024) https://news.ycombinator.com/item?id=41921738

- "Brain found to store three copies of every memory" https://news.ycombinator.com/item?id=41352124 but actually more than three: {in the hippocampus {EBN, LBN,}, prefrontal cortex, and engrams} but functional activations drift even just days apart given representational drift.

- "Memories stick around even after half the synapses are gone" (2026) https://news.ycombinator.com/item?id=49404593

zawaideh - 3 days ago

[dead]

xxajsg - 14 hours ago

[flagged]

blinkbat - 13 hours ago

Does this shit actually work lol