让RNN具备随序列增长的记忆能力,提升长文本理解表现。
Memory Caching: RNNs with Growing Memory
- 通过缓存隐藏状态快照,使RNN记忆容量随序列长度动态增长。
- 在上下文回忆任务中,性能接近Transformer,优于现有RNN模型。
- 适用于需要长程依赖建模的场景,如长文本生成与理解。
Transformers已成为序列建模的主流架构,主要因其记忆容量随上下文长度增长。然而其二次复杂度限制了实际应用,促使研究者探索更高效的循环替代方案。尽管循环架构在多个领域表现良好,但在依赖回忆的任务中仍落后于Transformer,常归因于其固定大小的记忆。本文提出记忆缓存(Memory Caching, MC),一种简单有效的方法:通过缓存隐状态的检查点来增强循环模型。该方法使RNN的有效记忆容量随序列长度增长,灵活平衡固定记忆($O(L)$)与增长记忆($O(L^2)$)之间的复杂度。我们提出了四种MC变体,包括门控聚合和稀疏选择机制,探讨其对线性与深层记忆模块的影响。实验表明,MC显著提升了循环模型在语言建模与长上下文理解任务中的表现。上下文回忆任务结果显示,虽Transformer精度最优,但MC变体表现极具竞争力,缩小了与Transformer的差距,并优于当前最先进的循环模型。
原文摘要 · Abstract (English)
Transformers have been established as the de-facto backbones for most recent advances in sequence modeling, mainly due to their growing memory capacity that scales with the context length. While plausible for retrieval tasks, it causes quadratic complexity and so has motivated recent studies to explore viable subquadratic recurrent alternatives. Despite showing promising preliminary results in diverse domains, such recurrent architectures underperform Transformers in recall-intensive tasks, often attributed to their fixed-size memory. In this paper, we introduce Memory Caching (MC), a simple yet effective technique that enhances recurrent models by caching checkpoints of their memory states (a.k.a. hidden states). Memory Caching allows the effective memory capacity of RNNs to grow with sequence length, offering a flexible trade-off that interpolates between the fixed memory (i.e., $O(L)$ complexity) of RNNs and the growing memory (i.e., $O(L^2)$ complexity) of Transformers. We propose four variants of MC, including gated aggregation and sparse selective mechanisms, and discuss their implications on both linear and deep memory modules. Our experimental results on language modeling, and long-context understanding tasks show that MC enhances the performance of recurrent models, supporting its effectiveness. The results of in-context recall tasks indicate that while Transformers achieve the best accuracy, our MC variants show competitive performance, close the gap with Transformers, and performs better than state-of-the-art recurrent models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。