LLM2D
任务内存引擎(TME):一种带有图意识扩展的结构化记忆框架,用于多步骤LLM代理任务
Task Memory Engine (TME): A Structured Memory Framework with Graph-Aware Extensions for Multi-Step LLM Agent Tasks
作者: Ye Ye
发布日期: 4/17/2025
arXiv ID: oai:arXiv.org:2504.08525v3

摘要

arXiv:2504.08525v3 通告类型: 代替 摘要:大型语言模型(LLMs)越来越多地被用作执行多步骤任务的自主代理。然而,目前大多数框架无法维持任务状态的结构化理解,通常依赖于线性提示串联或浅层的记忆缓冲区。这导致了脆弱的表现、频繁的幻觉以及较差的长程一致性。在本文中,我们提出了任务记忆引擎(TME),这是一种轻量级且结构化的记忆模块,使用层次化的任务记忆树(TMT)跟踪任务执行。树中的每个节点对应一个任务步骤,存储相关的输入、输出、状态以及子任务关系。我们引入了一种提示合成方法,根据活动节点路径动态生成LLM提示,显著提高了执行一致性和上下文关联性。通过多步骤代理任务的实际案例研究和对比实验,我们展示了TME在最小实现开销下能够获得更好的任务完成准确性和更可解释的行为。核心TME组件的参考实现可在https://github.com/biubiutomato/TME-Agent 获取,其中包含基本示例和结构化记忆集成。虽然当前实现使用基于树的结构,但TME被设计成具有图意识,支持可重用的子步骤、收敛的任务路径和共享的依赖关系。这为未来的基于DAG的内存架构奠定了基础。