AI Research

Google Research Proposes Memory Caching to Close the RNN-Transformer Gap

A new paper gives recurrent models memory that grows with context, without the quadratic cost.

Liza Chan
Liza ChanAI & Emerging Tech Correspondent
June 8, 20263 min read
Share:
Abstract visualization of neural network memory checkpoints stored across a long sequence of data nodes

Researchers at Google Research have published a paper proposing a way to make recurrent neural networks competitive with Transformers on long-context tasks. The technique, called Memory Caching, appeared on arXiv on February 27 and is credited to Ali Behrouz, Vahab Mirrokni, and four co-authors.

The actual problem

For about seven years, the big models you have heard of have leaned on one design. Transformers compare every token against every other token to keep track of context. That works, and it is why recall is so good, but the compute cost grows quadratically with the length of the input. Double the context, quadruple the bill.

RNNs were supposed to be the cheaper answer. They run in linear time and squeeze the entire past into one fixed hidden state. The catch is that the fixed state keeps overwriting itself, so the longer the sequence, the more the model forgets. Anyone who has watched an RNN lose the thread of a long sentence knows the failure mode.

So you pick your poison: expensive and accurate, or cheap and forgetful.

What Memory Caching actually does

Instead of forcing the network to compress everything into a single state, the method caches checkpoints of those memory states as it works through a sequence. The effective memory then grows with context length rather than staying frozen. The authors frame it as a dial that sits between the linear cost of an RNN and the quadratic cost of attention, which is a more honest pitch than "we beat Transformers."

They built four variants, including a gated aggregation approach and a sparse selective one where the model decides which checkpoints are worth keeping. That last bit is the interesting part. A model choosing what to remember is closer to how you would actually want a long conversation handled than running the whole history through attention on every step.

Do the numbers hold up?

On language modeling and long-context understanding, the paper reports that the variants improve on standard recurrent models. On in-context recall, the framing is more careful. The authors concede Transformers still post the best accuracy, while their variants close the gap and beat other recurrent models. Read that twice. The headline is not that attention loses. It is that the cheap option got close enough to matter.

The experiments cover language modeling and long-context QA, which are reasonable testbeds, though not an exhaustive stress test. No frontier-scale model has been trained on this yet, so the open question is whether the trade-off survives at the sizes that actually ship products. Plenty of subquadratic ideas look great at small scale and quietly fall apart later.

Three other models on Hugging Face already cite the work, which is fast but tells you nothing about whether it scales.

Why anyone should care

If the approach holds at scale, it is a credible alternative to the architecture underpinning nearly every large model since 2017. That is a big if. For now it is a well-argued middle ground with promising small-scale evidence and an honest accounting of where it still trails. The paper page is open for comments, and the next real signal will be whether anyone reproduces the gains at a larger parameter count.

Tags:Google ResearchRNNTransformersmachine learninglong contextneural networksMemory CachingAI researchsequence modeling
Liza Chan

Liza Chan

AI & Emerging Tech Correspondent

Liza covers the rapidly evolving world of artificial intelligence, from breakthroughs in research labs to real-world applications reshaping industries. With a background in computer science and journalism, she translates complex technical developments into accessible insights for curious readers.

Related Articles

Stay Ahead of the AI Curve

Get the latest AI news, reviews, and deals delivered straight to your inbox. Join 100,000+ AI enthusiasts.

By subscribing, you agree to our Privacy Policy. Unsubscribe anytime.