arXiv:2509.25911cs.CL2025-09被引 26

用强化学习训练大模型自主构建记忆,提升长程信息理解能力。

Mem-α: Learning Memory Construction via Reinforcement Learning

  • 通过强化学习让模型自主决定存什么、怎么存、何时更新。
  • 在30k token训练下,可处理超400k token的长序列,性能显著优于基线。
  • 适合需要长期记忆与复杂信息管理的任务场景。

大语言模型代理受限于有限的上下文窗口,需依赖外部记忆系统实现长期信息理解。现有记忆增强型代理通常依赖预设指令和工具进行记忆更新,但语言模型可能缺乏判断存储内容、结构设计及更新时机的能力,尤其当记忆系统日益复杂时,易导致记忆构建不佳和信息丢失。为此,我们提出 Mem-alpha,一种通过交互与反馈训练代理有效管理复杂记忆系统的强化学习框架。我们构建了一个涵盖多样化多轮交互模式的专用训练数据集,并配备全面评估问题,以指导有效的记忆管理。训练中,代理处理顺序信息块,学习提取并存储相关内容,进而更新记忆系统。奖励信号源自对完整交互历史的下游问答准确率,直接优化记忆构建。为验证框架有效性,我们设计包含核心、情景和语义组件的内存架构,并配置多种操作工具。实证评估显示,Mem-alpha 显著优于现有记忆增强代理基线。尽管仅在最大30k token的样本上训练,其代理仍能出色泛化至超过400k token的序列,超出训练长度13倍以上,凸显 Mem-alpha 的鲁棒性。

原文摘要 · Abstract (English)

Large language model (LLM) agents are constrained by limited context windows, necessitating external memory systems for long-term information understanding. Current memory-augmented agents typically depend on pre-defined instructions and tools for memory updates. However, language models may lack the ability to determine which information to store, how to structure it, and when to update it, especially as memory systems become more complex. This results in suboptimal memory construction and information loss. To this end, we propose Mem-alpha, a reinforcement learning framework that trains agents to effectively manage complex memory systems through interaction and feedback. We also construct a specialized training dataset spanning diverse multi-turn interaction patterns paired with comprehensive evaluation questions designed to teach effective memory management. During training, agents process sequential information chunks, learn to extract and store relevant content, then update the memory system. The reward signal derives from downstream question-answering accuracy over the full interaction history, directly optimizing for memory construction. To illustrate the effectiveness of our training framework, we design a memory architecture comprising core, episodic, and semantic components, equipped with multiple tools for memory operations. Empirical evaluation demonstrates that Mem-alpha achieves significant improvements over existing memory-augmented agent baselines. Despite being trained exclusively on instances with a maximum length of 30k tokens, our agents exhibit remarkable generalization to sequences exceeding 400k tokens, over 13x the training length, highlighting the robustness of Mem-alpha.

大模型记忆系统强化学习

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