历史自我重演:释放线性注意力的长上下文能力
摘要
论文提出了 NoteBank,一种免训练的稀疏情景记忆,通过归档选定文本并通过原生计算进行重放来增强线性注意力模型,在一百万 token 下将 Gated DeltaNet 在 RULER 上的长上下文检索准确率从 1.7% 提升至 70.3%,且 FLOPs 开销可忽略不计。
查看缓存全文
缓存时间: 2026/10/15 18:59
论文页面 - 历史重演:释放线性注意力中的长上下文能力
来源:https://huggingface.co/papers/2610.15715 作者:
,
,
,
,
,
,
,
,
,
,
摘要
NoteBank 通过一个稀疏情景记忆为线性注意力模型提供支持,该记忆存储选定的文本并通过原生计算进行回放,在无需额外训练或显著开销的情况下大幅提升了长上下文检索能力。
线性注意力(https://huggingface.co/papers?q=Linear%20attention)有望为长上下文语言建模提供所需的效率,但其固定大小的循环状态(https://huggingface.co/papers?q=recurrent%20state)必须同时承担持续的计算任务并保留不断增长的历史信息。这种耦合强化了一种普遍观点:状态的记忆容量决定了模型的长上下文能力。我们挑战了这一预设的上限:循环计算不必承担长期存储的全部重任。我们提出NoteBank(https://huggingface.co/papers?q=NoteBank),一种稀疏情景记忆(https://huggingface.co/papers?q=sparse%20episodic%20memory),它在未来的查询出现之前归档选定的文本,利用骨干模型的原生特征进行写入和寻址,并通过回放机制在其原生前向计算中读取检索到的证据。这一免训练、即插即用的接口无需外部嵌入模型、单独训练的检索器或独立的神经阅读器。在一百万 token 的长度下,它将Gated DeltaNet(https://huggingface.co/papers?q=Gated%20DeltaNet)在六项RULER(https://huggingface.co/papers?q=RULER)检索任务上的平均准确率从 1.7% 提升至 70.3%,且在 4K 到 1M token 范围内未观察到明显的检索性能退化。这一恢复仅增加 ≤0.025% 的理论 FLOPs,持久上下文存储约为同等宽度、深度和精度下 Transformer 完整 MHA KV 缓存的 1/2,600。实验从三个 1.3B 循环骨干模型扩展到Mamba-2(https://huggingface.co/papers?q=Mamba-2)-8B 和后训练的 Qwen3.5-9B,在LongBench(https://huggingface.co/papers?q=LongBench)v1/v2 和MemoryAgentBench(https://huggingface.co/papers?q=MemoryAgentBench)上均展示了性能增益。这些结果共同表明,稀疏证据存储与原生回放能够在保留循环计算效率的同时,显著扩展预训练线性注意力(https://huggingface.co/papers?q=linear%20attention)模型的长上下文能力。
查看 arXiv 页面(https://arxiv.org/abs/2610.15715)查看 PDF(https://arxiv.org/pdf/2610.15715)项目主页(https://nanboli.github.io/notebank-page)添加到收藏夹(https://huggingface.co/login?next=%2Fpapers%2F2610.15715)
在您的 agent 中获取此论文:
hf papers read 2610.15715
还没有最新的 CLI?curl -LsSf https://hf.co/cli/install.sh | bash
引用此论文的模型 0
没有模型关联此论文
在模型 README.md 中引用 arxiv.org/abs/2610.15715 即可在此页面上建立链接。
引用此论文的数据集 0
没有数据集关联此论文
在数据集 README.md 中引用 arxiv.org/abs/2610.15715 即可在此页面上建立链接。
引用此论文的 Space 0
没有 Space 关联此论文
在 Space README.md 中引用 arxiv.org/abs/2610.15715 即可在此页面上建立链接。
包含此论文的合集 0
没有合集包含此论文
将此论文添加到合集(https://huggingface.co/new-collection)即可在此页面上建立链接。
相似文章
更少的上下文,更高的准确性:一种用于LLM代理的双时态记忆引擎,其中精简检索的上下文胜过了完整历史
本文介绍了Engram,一个开源的用于LLM代理的双时态记忆引擎,它通过检索一个紧凑的上下文片段(约9.6k token),在LongMemEval上以混合读取路径融合稠密、词汇、图和时间信号,比完整历史基线(79k token)高出10.4个准确率点。
内存管理的长上下文注意力:可编辑请求本地内存的初步研究
本文研究了内存管理的长上下文注意力,这是一个将高效状态压缩与显式可编辑内存槽分开的研究方向。实验表明,结合快速循环/稀疏主干网络与显式内存管理的混合方法,在合成任务和长上下文基准测试中均优于纯固定状态或纯稀疏方法。
时间上下文重建驱动长上下文语言模型中的类情节顺序记忆
本文基于一部小说的时序记忆任务,研究长上下文语言模型是否表现出与人类相似的类情节顺序记忆。作者发现模型显示出相同的距离效应,并揭示了一个单独的关注头(attention head)重建时间上下文,支持时间上下文重建作为大语言模型中类情节记忆的机制。
LongAct:利用内在激活模式进行长上下文强化学习
LongAct 提出了一种显著性引导的稀疏更新策略,通过选择性更新与查询和键向量中高幅值激活相关的权重来改进 LLMs 的长上下文推理能力,在 LongBench v2 上实现了约 8% 的提升。
@omarsar0:新论文值得一读。(收藏)基本思路是将压缩循环状态与一个小型精确记忆配…
HOLA(海马体线性注意力)受海马记忆启发,通过一个有界的精确KV缓存增强线性注意力,在不牺牲效率的情况下改善长程召回和困惑度。在340M参数规模下,它在Wikitext上优于全注意力Transformer,并在高达32k token的范围内实现稳健的针召回。