跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

MemQ:将Q学习融入基于溯源DAG的自我进化记忆代理

文章摘要

MemQ提出了一种通过将Q学习与资格迹应用于溯源有向无环图(DAG)来增强LLM代理情景记忆的方法,考虑了记忆间的依赖关系。在六个基准测试中均取得最佳性能,尤其在多步任务上提升显著。

来源arXiv AI作者: Junwei Liao, Haoting Shi, Ruiwen Zhou, Jiaqian Wang, Shengtao Zhang, Wei Zhang, Weinan Zhang, Ying Wen, Zhiyu Li, Feiyu Xiong, Bo Tang, Muning Wen
MemQ:将Q学习融入基于溯源DAG的自我进化记忆代理
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

大型语言模型(LLM)代理在执行复杂任务时,通常依赖情景记忆来积累和检索过去的经验。然而,现有的记忆增强方法往往将每个记忆视为独立的实体,在评估检索质量时忽视了记忆之间的依赖关系——即一个记忆的创建可能依赖于之前检索到的其他记忆。为了弥补这一根本性缺陷,由Junwei Liao等12位研究者组成的团队提出了MemQ,一种将Q学习与时间差分(TD(λ))资格迹相结合的新型记忆增强框架。MemQ的核心创新在于,它通过构建一个溯源有向无环图(DAG)来记录每个新记忆创建时所检索到的记忆,并利用资格迹沿DAG反向传播信用分数。具体而言,信用权重按(γλ)^d衰减,其中d代表DAG的深度,从而用结构接近性取代了传统的时间距离度量。这一设计使得代理能够更准确地评估哪些记忆对后续任务真正有价值。

为了从理论上巩固这一方法,研究人员还形式化了一个外生上下文马尔可夫决策过程(EC-MDP),其因子化转移机制将外生的任务流与内生的记忆存储解耦。这一形式化不仅揭示了MemQ的工作原理,还为参数选择提供了原则性指导。在多步任务中,代理需要按顺序检索多个相互依赖的记忆,而MemQ通过DAG结构能够捕捉到这种依赖链,从而在信用分配上实现更精细的控制。

在涵盖操作系统交互、函数调用、代码生成、多模态推理、具身推理和专家级问答等六个不同领域的基准测试中,MemQ在泛化评估和运行时学习方面均取得了最高的成功率。值得注意的是,在那些需要深度溯源链的多步任务上,MemQ的性能提升高达5.7个百分点;而在简单的单步分类任务上,提升幅度仅为0.77个百分点,因为传统的单步更新方法在这些场景下已经足够。此外,论文还对γ和λ两个参数与EC-MDP结构的相互作用进行了深入分析,为未来的研究提供了坚实的基础。MemQ的代码计划在不久后开源,这将进一步推动LLM代理在复杂环境中的自主学习和适应性进化。

展开要点与分析

文章情报

工程师进阶

要点

  • MemQ使用TD(λ)资格迹通过溯源DAG反向传播信用,用结构距离替代时间距离。
  • 该方法将记忆检索形式化为外生上下文MDP,将任务流与记忆存储解耦。
  • 在所有六个基准测试上超越基线,多步任务提升高达+5.7个百分点。
  • 通过EC-MDP结构为参数选择提供了原则性指导。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。