跳到正文
Rohan Paul· @rohanpaul_ai · X·· 2026-07-09AI 评分52
AI 导读

上海大学论文提出HOLA,为线性注意力/状态空间模型增加精确记忆缓存,专门存储那些导致模型最大修正的“意外”token,读取时采用更锐利的方式提取。该方法解决了快速模型将所有信息压缩到单一固定状态导致长序列早期细节被覆盖的问题。在340M参数测试中,Wikitext困惑度从27.32降至22.92,超越同基准快速模型和全注意力Transformer++;仅用2k上下文训练即可在32k token范围内保持优异召回。

正文 · 原文

A massive idea from this Shanghai University paper

Giving AI Models Human-Style Memory

Imagine reading a huge mystery book while only being allowed to keep 1 sticky note.

Every time a new clue appears, the note must be rewritten, so older clues slowly fade or get squeezed out.

That is close to what happens inside fast AI models like linear-attention and state-space models, which save memory by compressing everything they read into 1 fixed internal state.

The problem is that this makes them very forgetful. In a "needle-in-a-haystack" test, where they need to recall a particular fact buried deep in a large document, they fail because fresh details overwrite the earlier ones.

HOLA tries to fix this by giving those models a second place to store details.

The normal state still handles the broad shape of the text, like grammar, topic, and repeated patterns.

The new memory cache stores exact pieces of information that the normal state seems likely to forget.

The clever part is how HOLA decides what deserves that cache.

It does not simply save the newest tokens, because old details can still matter later.

Instead, it saves the tokens that caused the biggest correction inside the model, which usually means the model struggled to compress them cleanly.

In plain terms, HOLA keeps the facts that surprised the model.

When the model later needs an answer, this cache is read in a sharper way, so it can pull back a specific saved fact instead of blending many memories into a vague guess.

This matters because fast models usually trade memory for speed, while full-attention Transformers keep stronger recall but become heavier as the text gets longer.

In the paper’s 340M-parameter test, HOLA lowered Wikitext perplexity from 27.32 to 22.92, beating the matched fast baseline and even beating the reported full-attention Transformer++ on that measure.

It also stayed much better at needle-in-a-haystack recall out to 32k tokens, even though it was trained on 2k-token contexts.

The bigger lesson is that a fast AI model does not need to remember everything exactly.

It needs to know which details its main memory is bad at keeping, then protect those details before they disappear.

---

– arxiv. org/abs/2607.02303

Title: "A Hippocampus for Linear Attention: An Exact Memory for What the Recurrent State Forgets"

来源:Rohan Paul · x.com