arXiv:2607.25357cs.LGcs.AI2026-07被引 5

Raven通过稀疏记忆路由实现高效长序列召回。

Raven: High-Recall Sequence Modeling with Sparse Memory Routing

论文配图:Raven: High-Recall Sequence Modeling with Sparse Memory Routing
图 1 · 摘自论文原文
  • 用可学习的输入依赖路由选择性更新记忆槽,避免全量状态更新干扰。
  • 在长序列召回任务中超越SWA与SSM,在16倍超长上下文仍有效。
  • 适合需要高召回率的长文本建模场景,如文档理解、代码生成。

线性时间序列模型在长上下文召回中面临内存写入方式的权衡。基于状态的线性模型(如状态空间模型SSMs和线性Transformer)采用密集写入,每次新词到来时更新整个状态,导致干扰,难以恢复特定历史词。滑动窗口注意力(SWA)则稀疏写入,仅存储固定窗口内的显式词表示,但一旦相关词被挤出,召回率即下降。我们提出Raven,一种线性时间序列模型,维护固定数量的记忆槽,每步仅对选定子集进行衰减与更新,由学习到的、输入相关的路由决定。该机制缓解了SWA的位置依赖覆盖与硬淘汰问题,同时减少SSMs中密集状态更新带来的干扰,显著提升长程内容保留能力。在多个依赖召回的基准测试中,Raven表现优于或媲美现有线性时间基线,在长上下文下显著超越SWA与SSMs。其性能在扩展至训练长度16倍的上下文时依然保持,且在混合架构中同样表现出色。

原文摘要 · Abstract (English)

Long-context recall in linear-time sequence models highlights a tradeoff in how they write to memory. State-based linear models, such as state-space models (SSMs) and linear Transformers, write densely, updating the entire state for each newly arrived token, which leads to interference and makes specific past tokens hard to recover. Sliding-window attention (SWA) exhibits the opposite behavior: it writes sparsely by storing explicit token representations, but only within a fixed window, so recall drops once the relevant token is evicted. Interpolating between these models, we introduce Raven, a linear-time sequence model that maintains a fixed set of memory slots and, at each step, decays and updates only a selected subset via learned, input-dependent routing. This lets Raven mitigate SWA's position-based overwriting and hard eviction while reducing interference from dense state updates in SSMs, thereby preserving long-range content much more effectively. Across recall-intensive benchmarks, Raven is competitive with or outperforms prior linear-time baselines, achieving strong long-context recall where both SWA and SSMs sharply degrade. It remains effective when extrapolating to context lengths as large as 16x its training length, with similar gains in hybrid architectures.

序列建模长序列稀疏路由记忆机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。