记忆智能体:融合非参数与参数化记忆的深度研究代理框架 Memory Intelligence Agent
提出 Manager-Planner-Executor 三体架构的双记忆深度研究代理,通过交替强化学习与测试时学习解决长上下文记忆噪声与检索成本问题。
前置知识
Deep Research Agent (DRA, 深度研究代理)
一类结合大语言模型推理能力与外部工具(如搜索、检索)的智能体系统,采用多轮交错式的思考-搜索-推理循环完成复杂开放式任务。与直接问答不同,DRA 在面对需要外部事实、多跳推理或多模态信息的问题时会主动调用搜索引擎、检索器等工具获取信息后再做归纳。
本文的核心研究对象就是 DRA 的记忆机制。理解 DRA 的工具调用循环(ReAct 范式)和多轮推理范式,是理解 MIA 为何要拆分 Planner/Executor、设计反思-重规划机制的前提。
GRPO (Group Relative Policy Optimization)
一种基于群体相对优势的强化学习优化算法,与 PPO 类似但无需训练独立的价值函数。它对每个 prompt 采样一组(group)候选输出,根据组内相对奖励计算优势(advantage),并对策略网络进行裁剪式更新与 KL 散度惩罚,常用于大模型的 RLHF/RLAIF 微调。
MIA 在 Executor 训练(公式 1)和 Planner 训练(公式 3)阶段都基于 GRPO 设计目标函数。理解裁剪项 clip(1-ε, 1+ε) 和优势项 Â 的含义,才能读懂为什么 Planner/Executor 要交替训练而非联合训练。
ReAct (Reason + Act, 推理-行动)
一种让 LLM 在每一步先生成自然语言思考(think),再决定调用工具(act),观察工具结果后再次进入思考循环的范式。其核心是让思考与工具调用在同一序列中交错出现,实现可解释的动态工具使用。
MIA 的 Executor 完全遵循 ReAct 模式执行 Planner 生成的步骤,且论文明确提到奖励函数 r2 鼓励标准化的工具调用。理解 ReAct 是看懂 Table 1 滚动流程的前提。
非参数记忆 vs. 参数化记忆 (Non-parametric vs. Parametric Memory)
非参数记忆指外挂的、可增删改查的显式知识库(如向量数据库、记忆 buffer),检索时显式调用;参数化记忆指隐式存储在模型权重中的知识,通过前向计算体现出来,不占用额外存储空间。两者在 RAG 系统中往往是互补的:参数化记忆压缩表达,非参数记忆保存具体实例。
MIA 的核心创新正是双向转换这两种记忆:Memory Manager 保存压缩后的工作流(非参数),Planner 通过 RL 把频繁出现的轨迹模式内化到模型权重中(参数化)。读懂这种双向闭环才能理解本文为何强调 'memory evolution'。
测试时学习 (Test-Time Learning, TTL)
指模型在推理/部署阶段,根据未见过的新数据继续在线更新参数或记忆的能力,区别于传统的离线训练。代表工作如 TTT、TTT 层。TTL 强调模型在生产环境中持续适应分布变化、不依赖预收集数据或多 epoch 滚动。
MIA 的关键机制之一就是让 Planner 在测试时(即面对新批次数据时)通过 GRPO 在线更新参数,无需中断推理过程。这是 3.4 节的核心,也是相对于传统 CoT prompting 工作的本质差异。
研究动机
现有的 Deep Research Agent 普遍依赖长上下文记忆机制,即把所有历史搜索经验塞进 prompt 里供模型参考。这种做法暴露出四个致命问题:第一,长上下文会稀释注意力,让模型难以聚焦当前问题;第二,无关或弱相关的旧内容会被错误激活,给推理引入噪声;第三,持续增长的上下文带来急剧膨胀的存储开销;第四,对超大记忆库的检索代价高昂,时延难以接受。论文进一步指出,长上下文记忆主要捕获的是知识/事实型记忆(如用户属性、文档片段),而 DRA 真正需要的是过程型记忆(如搜索轨迹、失败尝试、成功策略)。已有的 CoT + Few-shot prompting 方案虽然部分缓解了上述问题,但 Planner 没有任务特定训练、检索仅靠相似度而不考虑质量与频率、Executor 也不懂得解读规划指令,本质上是「一个不合格的 Planner 从臃肿记忆里检索、用不完整的 prompt 引导一个没训练好的 Executor 去做研究」。
本文的目标是本文提出的总体目标是构建一个名为 MIA(Memory Intelligence Agent)的统一框架,让 DRA 既拥有过程型经验的显式存储,又具备长期自我进化的能力。具体而言,希望达成四个子目标:(1) 通过 Manager-Planner-Executor 三体架构,将历史记忆、参数化规划和动态执行解耦,破解存储与推理之间的张力;(2) 让 Planner 与 Executor 通过交替强化学习相互适配,缩小规划-执行之间的语义鸿沟;(3) 让 Planner 在测试时持续进化,无需打断推理流程就能内部化新经验;(4) 引入反思机制和无监督判别机制,使代理在缺乏标准答案的开放场景下也能自我纠错与持续进化。最终目标是让即使是 7B 规模的 Executor 也能在多模态与文本深度研究任务上达到或超越大型闭源 LLM。
与已有工作不同的是,本文的独特切入角度是把脑启发记忆机制与 RL 训练结合成一个闭环:现有方法要么只做长上下文检索(RAG、Mem0、A-Mem)、要么只做记忆抽象(ReasoningBank、ExpeL、Memento),要么把记忆冻结在 LLM 之外(Agentic Memory)。MIA 首次提出让非参数记忆与参数化记忆之间通过 RL 训练形成双向转换闭环——成功的轨迹既压缩成可检索的 workflow(非参数),又通过 GRPO 内化到 Planner 的权重里(参数化)。同时,本文引入「模拟学术会议同行评审」的 AC-Reviewer 多视角无监督评估机制,把 LLM-as-a-Judge 的单点偏差问题分解到 Reasoning/Source/Validity 三个正交维度,让无监督 TTL 也能稳定运行。这两点在现有 DRA 工作中均未被系统化地提出过。
核心方法
MIA 用「大脑启发」的视角把 DRA 的记忆分成两类:非参数的海马体式情景记忆(保存压缩后的工作流样本)和参数化的新皮层式知识(Planner 模型的权重)。整体架构分为三个角色——Memory Manager 负责管理非参数记忆库,Planner 负责根据检索到的 workflow 用 few-shot CoT 生成当前问题的搜索计划,Executor 负责按照计划在 ReAct 循环里调用 search/web_image_to_image_search 等工具并产出最终答案。推理时三者按 Memory Retrieval → Collaborative Reasoning (Plan-Execute-Reflect-Replan) → Experience Consolidation 三阶段循环运行:检索阶段按语义相似度、价值奖励、频率奖励三维评分找到高质量正负样本;协作推理阶段 Planner 给计划、Executor 执行、反馈触发 Planner 反思-重规划;经验整合阶段由 LLM Judger 判定轨迹正误,提取最短成功轨迹和随机失败轨迹压缩成 workflow 入库,同时用当前批次的轨迹对 Planner 进行 GRPO 在线更新(即测试时学习)。整套设计的核心是让非参数记忆成为 Planner 的「上下文参考」,让参数化记忆通过 RL 不断吸收上下文中的高频模式,两者通过交替 RL 和 TTL 形成相互强化的闭环。
MIA 与已有方法最本质的区别有三处:第一,明确把「过程型记忆」(搜索轨迹、工作流)作为研究对象,而非简单的检索文档,提出用 Qwen3-32B 把冗长轨迹压缩为抽象工作流摘要,入库时再按语义相似度做知识替换,从源头解决记忆臃肿问题。第二,采用两阶段交替 RL——先冻结 Planner 训练 Executor,让 Executor 学会理解规划;再冻结 Executor 训练 Planner,让 Planner 学会吸收记忆并反思——这种解耦使得奖励信号在每一方都是相对稳定的,避免了端到端 RL 中策略耦合带来的高方差。第三,引入测试时学习 (TTL):Planner 在面对每个测试批次时同时做 G 条计划 rollout,由 Router 从中选出最优解输出,同时用所有 rollout 的奖励更新 Planner 参数,真正实现边推理边进化。这与 Memento 等冻结 LLM 微调解码器的工作形成对照,MIA 让 Planner 本身持续学习。
方法步骤详情
MIA 的工作流程可按推理时序拆解为四个阶段。Stage 1 (Memory Retrieval):Memory Manager 先把视觉输入转成文本 caption,与历史问题一同送入 sup-simcse-bert-base 编码器计算余弦相似度,结合问题级权重 αq=0.8、caption 级权重 αc=0.2,再叠加 value reward = si/(ui+1) 和 frequency reward = 1/(ui+1.1) 三维评分 (λs=0.7, λv=λf=0.3),挑出 top-k 条高质量正负 workflow 作为 Planner 的 few-shot 上下文。Stage 2 (Collaborative Reasoning):Planner 用 [Relevant Memories] + 当前问题生成原子化、可执行的 step-by-step 计划;Executor 在 ReAct 循环中按 Think→Act→Observe→Think 推进,每步调用 search 或 web_image_to_image_search 工具,得到中间结果后再次交回 Planner;Planner 据此判断是否触发 Reflect-Replan(最多触发一次,以节省时间)。Stage 3 (Experience Consolidation):LLM Judger (Qwen3-32B) 用 grade prompt 判定最终答案对错,成功轨迹中选最短的、失败轨迹中随机采样一条,由 Memory Manager 调用 Qwen3-32B 压缩为结构化 workflow 摘要,按语义相似度阈值做替换式入库,同时更新 value 与 frequency 计数。Stage 4 (Parametric Update):TTL 模块在此并行执行,对当前批次用 G=4 个 rollout 经 Router 选出最优输出,剩余轨迹一起计算 GRPO advantage (Âi = (Ri - μR)/(σR + ε)),再按公式 (3) 更新 Planner 参数,并把最短成功轨迹 + 随机失败轨迹作为 (p*succ, p*fail) 对存入 Meta Plan Memory。训练时则对应两阶段:阶段一冻结 Planner (Qwen3-8B) 对 Executor (Qwen2.5-VL-7B) 做 8 epoch GRPO 训练 (lr=1e-6, batch=128, KL=0.0),奖励 r_ME = 0.7·r1 + 0.2·r2 + 0.1·r3;阶段二冻结 Executor,对 Planner 做 4 epoch GRPO,奖励 r_MP = 0.7·r1(final) + 0.2·r1(intermediate) + 0.05·r2 + 0.05·r3,并在 prompt 中加入 memory context。
技术新颖性
MIA 的技术新颖性可以总结为四点。第一,定义了一种新的过程型记忆结构:每条记忆单元存储 (question, image caption, workflow, judgment label, usage count, success count) 五元组,按相似度+价值+频率三维评分检索,相对传统 RAG/Mem0 仅靠向量相似度检索有质的提升。第二,提出两阶段交替 GRPO 训练范式,把 Planner-Executor 这一联合优化问题解耦为两个相对独立的子优化,使得 Planner 和 Executor 的奖励曲线更稳定、收敛更快(论文 Figure 7 显示 Executor reward 单调上升,Planner reward 虽慢但稳定)。第三,提出测试时双向记忆进化:非参数层用最短成功 + 随机失败的 contrastive pair 强化样本质量,参数化层用 GRPO 在线更新 Planner 权重,二者并行执行且共享同一批 rollout 数据,避免传统离线 RL 必须预收集数据的瓶颈。第四,引入模拟学术会议的 Reviewer×3 + Area Chair 1 的多视角同行评审机制用于无监督 TTL,三个 Reviewer 分别评估 Reasoning Logical Consistency、Information Sourcing Credibility、Result Validity,最终由 AC 综合三者权重 0.5/0.3/0.2 输出最终判定,并用结构的 JSON 报告提供 evidence quote 审计轨迹,从机制上消除 LLM-as-a-Judge 的「hallucinated objectivity」偏差。
实验结果
核心实验共分四组,每组都用 Qwen2.5-VL-7B 作为 Executor 并在 11 个 benchmark 上对比。Table 3 的多模态结果显示,MIA 在 FVQA-test、InfoSeek、SimpleVQA、LiveVQA、MMSearch、In-house 1、In-house 2 七个数据集分别取得 69.6、65.5、64.9、43.1、62.6、31.8、37.7 的 accuracy,平均 53.6,相对此前最强记忆方法 Memento 提升 5.5(其中 LiveVQA +6.4、In-house 1 +9.1 提升最显著),并超过 GPT-5.4 与 GPT-4o,逼近 Gemini-3-Flash。Table 4 的文本结果显示 MIA 在 SimpleQA/2Wiki/HotpotQA/GAIA 取得 47.7/71.8/63.5/31.1,平均 53.5,比 Memento 提升 7.5(GAIA +8.8 尤为亮眼)。Figure 8 的闭源 Executor 泛化实验中,MIA 把 GPT-5.4 在 LiveVQA 推高 8.9 (60.0→69.0)、HotpotQA 推高 6.4 (75.5→81.9);对 Gemini-3-Flash +3.1/+2.6,对 Claude-Sonnet-4.6 +1.8/+1.7,呈现「基础模型越弱、改进越大」的稳健趋势。Figure 9 的工具调用散点揭示:纯长上下文方法 (RAG/Mem0/A-Mem) 由于工具调用次数过少而表现差;纯元指导方法 (ReasoningBank/ExpeL) 居中;Memento 和 MIA 加入显式 Planner 后显著拉开差距,MIA 处于散点最右上角(即同时具备高准确率与高频成功工具调用)。Table 5/6 的消融实验以 Base 模型为起点,逐步加入 Only Memory、Only Plan、Memory for Planner、+Reflect、Trained Planner、+TTL 组件,多模态平均准确率依次为 41.5 → 41.1 → 44.3 → 45.4 → 45.8 → 48.2 → 51.5(粗略累加),可见 TTL 与交替 RL 训练 Planner 是最大两次增益,Memory for Planner 模块相比 Only Memory 提升 3.5(多模态平均)。Table 7 的无监督自进化结果更有说服力:Unsupervised MIA epoch-1 在 FVQA/LiveVQA/2Wiki/HotpotQA 取得 65.1/40.1/71.6/61.7,epoch-2 提升到 66.4/41.4/73.4/63.1,epoch-3 再到 67.1/41.8/74.7/63.2,证实了在缺乏标准答案的情况下模型仍能通过 TTL 持续进步。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| FVQA-test (Multimodal Fact VQA) | Accuracy | 69.6 | Memento: 66.3; MMSearch-R1: 58.0; GPT-5.4: 50.8 | +3.3 over Memento, +19 over GPT-5.4 |
| InfoSeek (Multimodal Info Seeking) | Accuracy | 65.5 | Memento: 57.3; MMSearch-R1: 49.0; Gemini-3-Flash: 69.0 | +8.2 over Memento, 接近 Gemini-3-Flash |
| SimpleVQA (Multimodal Short Fact QA) | Accuracy | 64.9 | Memento: 61.9; Deepeyes2: 59.4; Gemini-3-Flash: 73.7 | +3.0 over Memento |
| LiveVQA (Multimodal Multi-hop VQA) | Accuracy | 43.1 | Memento: 36.7; Gemini-2.5-Pro: 27.7; GPT-4o: 26.9 | +6.4 over Memento, 翻倍 GPT-4o |
| MMSearch (Multimodal Search) | Accuracy | 62.6 | Memento: 61.4; MMSearch-R1: 43.9 | +1.2 over Memento |
| In-house 1 (Scientific Domain VQA) | Accuracy | 31.8 | Memento: 22.7; Gemini-3-Flash: 52.5 | +9.1 over Memento (最大单点提升) |
| In-house 2 (Multi-hop Dynamic News VQA) | Accuracy | 37.7 | Memento: 30.7; Gemini-3-Flash: 25.5 | +7.0 over Memento, 比 Gemini-3-Flash 高 12.2 |
| SimpleQA (Text Factual QA) | Accuracy | 47.7 | Memento: 42.4; ExpeL: 43.0 | +5.3 over Memento |
| 2WikiMultiHopQA (Text Multi-hop) | Accuracy | 71.8 | Memento: 64.2; GPT-5.4: 50.8 | +7.6 over Memento, 提升最显著 |
| HotpotQA (Text Multi-hop) | Accuracy | 63.5 | Memento: 55.2; GPT-5.4: 75.0 (closed) | +8.3 over Memento |
| GAIA-Text (Hard Text Reasoning) | Accuracy | 31.1 | Memento: 22.3; Gemini-3-Flash: ~30 (closed) | +8.8 over Memento, 最大文本端提升 |
局限与改进
作者承认的限制与可观察到的局限主要有以下几点:(1) 依赖 Qwen3-32B 作为 Memory Manager、LLM Judger、Router 等「冻结服务」,需要为每个角色单独维护一份部署,无形中增加了推理时的算力成本与系统复杂度,论文并未报告服务侧 GPU 占用。(2) 反思-重规划机制 (Reflect-Replan) 被硬编码为「最多触发一次」,虽然节省时间但面对极复杂的多跳问题时可能不足,作者未给出在不同难度下触发次数的消融。(3) GRPO 训练需要 8 GPU 训练 Executor + 4 GPU 训练 Planner 的集群,普通研究者难以复现训练流程;只有 inference 阶段可以靠 vLLM 单卡做。(4) Memor Manager 压缩 workflow 时基于一个用 MMSearch-R1 风格构造的 prompt (Figure 4),对低资源语言或非英文 workflow 的抽象质量未经验证。(5) 工具集仅限于 text search + image search 两种,远比 WebWatcher、Deepeyes2 简单,因此虽然得出「简单的工具也可以靠记忆取胜」的结论,但这种比较的公平性存疑——更复杂工具需要更多 Planner 决策,反过来可能放大 Planner 错误。(6) 无监督 TTL 的真正「无标注」设定中,引入 Qwen3-32B 作为 Reviewer×3 + AC,本质上是把监督信号外包给一个更大模型,在用户场景下未必可以接受这种计算开销,且 Qwen3-32B 的判断偏差会通过权重 0.5/0.3/0.2 在 AC 阶段被放大或掩盖。
独立分析的弱点
独立分析下 MIA 的弱点可以从工程与学术两个层面拆解。第一,Planner 的 RL 训练对超参数极度敏感:训练分析 (Figure 7) 显示 Planner 的 reward 曲线比 Executor 平缓得多,说明间接奖励信号噪声很大;论文承认 Planner 的响应长度也存在较大波动。建议改进方向:可尝试为 Planner 单独设计 intermediate reward(比如 plan 的 plan length、tool selection diversity),或者改用 process reward model (PRM) 给每个 plan step 打分,而非依赖最终答案的稀疏奖励。第二,TTL 的「边推理边训练」在生产环境部署时存在并发风险:每个 batch 要并行做 G=4 次 rollout + 1 次更新,若一次更新失败或回滚不当,会污染后续所有 batch 的参数。建议增加周期性 checkpoint 与回滚机制,并把 update 与 serving 拆分到不同进程。第三,Reviewer×3 + AC 的同行评审机制让无监督判定变得更鲁棒,但同时把判定成本从 1 个 LLM 调用放大到 4 个,且不同维度需要分别调试 prompt,迁移到新领域时调优成本陡增。建议引入可学习的维度权重,或者用 GroupWise bias 一次性评估。第四,Memory Manager 的工作流压缩目前调用 Qwen3-32B 单点完成,缺少对压缩质量的自动验证;如果压缩后丢失关键步骤,会让检索回来的负样本失效。建议加入 workflow summary 的 NLI 一致性校验或与原轨迹做 ROUGE/BLEU 对比,低于阈值则不替换入库。第五,工具集仅支持 search/web_image_to_image_search,未涉及代码执行、表格解析、跨语言检索等;论文刻意做了「轻量工具」实验,但深度研究场景中常需要多模态多工具协同。第六,论文未报告推理时延与服务成本,每条样本需要 G 次 rollout + 1 次 Router 调用 + 1 次 Executor 完整 ReAct 循环才能输出,难以直接用于实时性要求高的产品。
未来方向
作者在结尾提到计划将 MIA 扩展到「更复杂更动态的环境」。基于本文成果,可延伸的具体方向包括:(1) 把 Planner 拆成 hierarchical 形式:高层做任务分解,低层每步预测具体工具参数;同时引入 tool-aware 的中间奖励,缓解 Planner 的稀疏奖励问题。(2) 把 Memory Manager 的相似度检索升级为稠密语义图检索,例如让每条 workflow 用 code-as-actions (like Voyager) 表示,使检索结果可直接被 Executor 编译执行,从而把检索从「找相似」升级为「找可用 skill」。(3) 当前图像搜索工具仅返回 top-3,可以引入更细粒度的 visual evidence grounding(子图像定位+属性抽取),减少 caption 信息损失;这对 In-house 2 这种动态新闻类任务特别关键。(4) 把 Reviewer×3 + AC 的同行评审结构推广到多智能体协作,用 MIA 同时调度多个 Executor 解决子任务,再用 Reviewer 投票判定一致性。(5) 把 Planner TTL 的参数化更新与 Slow Thinking 模型结合:用 o1-style 的 chain-of-search 轨迹作为 Planner 的微调数据源,让 Planner 本身具备深层推理能力,进一步压缩「直觉式」与「思考式」之间的鸿沟。(6) 探索 Memory Manager 的去重/遗忘机制:长期运行下记忆库可能仍缓慢膨胀,可以引入基于熵或方差的主动遗忘策略,让 system 真正 lifelong evolvable。(7) 把无监督 TTL 推广到跨语言/跨模态迁移场景,例如用英文训练的 Planner 零样本迁移到中文多模态任务,测试记忆机制的模态无关性。
复现评估
MIA 的开源状况非常友好,论文明确给出 Code (https://github.com/ECNU-SII/MIA)、Model (https://huggingface.co/LightningCreeper/MIA)、Dataset (https://huggingface.co/datasets/LightningCreeper/MIA) 三个仓库。复现所需的硬件门槛处于中高水平:训练 Executor 需要 8 张 GPU(论文未指明型号,应为 A100 80G 或等效算力),训练 Planner 需要 4 张 GPU,TTL 测试每个批次的 G=4 rollout 在单卡 vLLM 上即可。训练数据总量适中(FVQA-train 4,856 + MATPO 6,175 条),所有 benchmark 都是公开可下载的,附录 E 详细列出了每个数据集的样例数与来源。复现难度主要集中在:(a) GRPO 的两阶段交替训练需要分别实施 8 epoch + 4 epoch,超参数 (lr=1e-6, KL=0.0, G=8 rollouts) 都很标准但 Executor 训练对 prompt template (Figure 4) 高度敏感;(b) Memory Manager 需要部署一个独立的 Qwen3-32B 长上下文服务,对显存与推理时延都提出要求;(c) Reviewer×3 + AC 的 prompt 模板在 Appendix G 中给出,但 prompt 的微小变化会显著影响无监督 TTL 的稳定性,作者未提供 prompt tuning 经验。综合来看,一个具备 8×A100 算力的研究组可在 1-2 周内复现主要 Table 3/4 的结果,但完全复现 TTL 与无监督自进化曲线可能需要更细致的 prompt 工程。
论文图表