arXiv:2509.23040cs.CLcs.AI2025-09被引 31

让大模型能回看历史记忆,实现更可靠的长文本推理

Look Back to Reason Forward: Revisitable Memory for Long-Context LLM Agents

  • 将记忆检索融入更新过程,支持非线性回溯推理
  • 多层级奖励机制提升记忆使用效率,显著优于现有方法
  • 适合需要复杂多跳推理的长文档问答场景

大语言模型在长文本问答中面临挑战,关键证据可能分散于数百万标记中。现有方法通过线性扫描动态更新记忆缓冲区,存在潜在证据丢失、信息被覆盖及强化学习信号稀疏等问题。为此,本文提出ReMemR1,将记忆检索机制嵌入记忆更新流程,使代理能够有选择地回调历史记忆进行非线性推理。为进一步强化训练,设计多层级奖励:结合最终答案奖励与密集的步骤级信号,引导有效记忆使用。实验表明,ReMemR1在长文本问答任务上显著超越当前最优基线,且计算开销几乎可忽略,验证了其以微小代价实现鲁棒长上下文推理的能力。

原文摘要 · Abstract (English)

Large language models face challenges in long-context question answering, where key evidence of a query may be dispersed across millions of tokens. Existing works equip large language models with a memory buffer that is dynamically updated via a linear document scan, also known as the "memorize while reading" methods. While this approach scales efficiently, it suffers from pruning of latent evidence, information loss through overwriting, and sparse reinforcement learning signals. To tackle these challenges, we present ReMemR1, which integrates the mechanism of memory retrieval into the memory update process, enabling the agent to selectively callback historical memories for non-linear reasoning. To further strengthen training, we propose a multi-level reward design, which combines final-answer rewards with dense, step-level signals that guide effective memory use. Together, these contributions mitigate information degradation, improve supervision, and support complex multi-hop reasoning. Extensive experiments demonstrate that ReMemR1 significantly outperforms state-of-the-art baselines on long-context question answering while incurring negligible computational overhead, validating its ability to trade marginal cost for robust long-context reasoning.

长文本推理记忆机制大模型

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