通过新颖性信号联合学习智能体记忆与探索 Joint Agent Memory and Exploration Learning via Novelty Signals
用代码覆盖率作为新颖性奖励,联合训练潜变量记忆与探索策略
前置知识
内在新颖性奖励
当外部奖励稀疏或缺失时,智能体依靠内在信号驱动探索。新颖性奖励衡量当前状态相对已访问历史有多陌生,越陌生奖励越高,从而鼓励智能体走出已知区域、发现新行为。它不依赖人工标注,可由预测误差或确定性信号(如覆盖率)生成。
本文核心奖励信号就是新颖性,理解它才能把握记忆与探索如何形成相互依赖的学习闭环。
潜变量记忆
智能体记忆的一种形式,将漫长交互历史压缩为紧凑的向量表示(记忆token),而非保留原始文本。相比可读的显式记忆,它计算高效、可与策略联合训练,但缺乏人类可解释语义,因此难以直接提供步骤级监督信号。
JAMEL的记忆架构基于潜变量token,理解其优缺点才能明白为何需要新颖性信号来监督。
代码覆盖率
软件测试领域的指标,统计程序运行时被执行到的代码实体(语句、分支、函数、行)。在GUI域,通过插桩浏览器可报告每步触发了哪些代码路径,从而提供确定性、持久性的行为新颖性度量,是一种无需标注的监督来源。
代码覆盖率是本文新颖性奖励的具体实例化,是理解奖励公式和数据采集流程的关键。
部分可观测马尔可夫决策过程(POMDP)
将探索形式化为有限时域POMDP $P=(S,A,O,P,\Omega,\rho_0,H)$:环境有隐藏状态,智能体仅观测部分信息,需依据历史推断已穷尽的行为。JAMEL的记忆正是为应对部分可观测性、压缩历史而设计。
这是本文探索问题的形式化框架,理解它才能明白记忆为何对探索不可或缺。
拒绝微调(Rejection Fine-tuning)
一种监督微调策略,从模型自身生成的多条轨迹中,依据奖励或质量筛选保留正向样本用于训练,丢弃低质量样本。本文保留每条轨迹截至最后一个正奖励步骤的前缀,确保每个保留步骤都来自最终产生新颖性的轨迹。
本文数据采集与训练采用这一范式,理解它才能把握24k训练样本的构造逻辑。
研究动机
当前LLM智能体在开放式环境中探索能力严重不足:即使遇到意外但任务相关的信息,多数情况下也无法据此行动,这种缺陷根源于训练方式而非推理配置。有效探索依赖记忆——智能体必须知道哪些状态、界面区域或行为后果已被观察,才能判断某动作是否值得尝试。最直接的做法是保留完整交互历史,但随着轨迹变长计算代价急剧上升。潜变量记忆虽能压缩历史,却面临根本难题:缺乏可靠的步骤级监督——我们通常不知道每个记忆状态应编码什么、策略该如何使用它们。显式文本记忆可解释可修改,但潜变量记忆的压缩向量缺乏人类可读语义,标准任务示范无法为其提供清晰的学习目标。
本文的目标是本文目标是设计一套框架,让智能体的潜变量记忆与探索策略能通过新颖性驱动的交互联合训练,从而在无需人工标注的情况下,使记忆学会编码“哪些行为已尝试”,使策略学会利用记忆发现未探索行为。具体而言:设计将每步历史压缩为单一记忆token的潜变量记忆架构;构建基于拒绝微调的探索数据采集流程,在GUI领域收集2.4万训练样本;并证明该模型能泛化到未见应用,在探索深度上超越开源基线、匹敌闭源模型,同时大幅降低token消耗。
与已有工作不同的是,本文独特切入角度是抓住“记忆与探索相互依赖”这一洞察:记忆使智能体避免重复、发现新行为,而探索暴露哪些历史信息对未来决策有用,因此探索本身可为记忆提供监督。仅当智能体达到历史未覆盖的行为时才给予新颖性奖励,最大化该奖励迫使记忆条件策略编码并利用已尝试内容;随着熟悉交互被穷尽,只有更深层多步序列持续产生奖励,形成自然课程。关键的是,作者选择GUI领域的代码覆盖率作为新颖性信号——它确定、持久、无需标注,任何应用都可通过插桩报告已执行代码路径,从而为潜变量记忆提供了天然、无标注的监督来源。
核心方法
JAMEL的整体思路是“以探索监督记忆”。直觉上,若智能体总能区分“已做过”和“未做过”,它就会持续探索;而要做到这点记忆必须有用。技术路线上,JAMEL用冻结的视觉语言模型作为历史压缩器,将每一步的(观测,动作)对压成单个记忆token;这些token经线性对齐器投射到策略模型嵌入空间,作为软前缀拼接到当前观测之前。训练数据通过部署通用LLM探索应用、用代码覆盖率计算奖励来收集,再经拒绝微调筛选前缀,对策略和记忆对齐器做联合监督学习,目标是最大化给定观测与记忆时动作的似然。整个过程无需外部标注,仅靠确定性覆盖率信号即可完成记忆与策略的协同训练。
核心创新点在于用持久的新颖性信号同时监督记忆与探索策略,而非像以往工作那样将记忆管理与学习目标解耦。与传统记忆方法(依赖静态缓冲区或启发式检索规则)的本质区别是:JAMEL把记忆压缩直接嵌入探索闭环,奖励仅当动作触发历史未覆盖行为时给出,从而迫使记忆条件策略必须编码“已尝试什么”才能获奖。在GUI域,新颖性被实例化为代码覆盖率:累计覆盖分数 $C(t)=\text{cov\_lines}+\text{cov\_branches}+\text{cov\_statements}+\text{cov\_functions}$,奖励 $r_t=\mathbb{1}[C(t)>C(t-1)]$。该信号确定且持久——一旦代码路径被覆盖就永不再奖励,杜绝了在少量状态间循环骗取奖励的退化。
方法步骤详情
方法分四步。(1)记忆架构:对历史每步序列化 $x_i=\text{Format}(o_i,a_i)$,喂入冻结压缩器 $F_\phi$,取EOS最终隐状态 $h_i\in\mathbb{R}^{d_c}$ 作为该步记忆token;记忆状态 $M_t=[h_1,\dots,h_{t-1}]$。(2)奖励计算:通过V8引擎和Istanbul报告采集覆盖率,按上式累加 $C(t)$,无效动作奖励为0,覆盖率基线跨回合不重置。(3)数据采集:部署通用LLM在浏览器中探索,每步产生CoT推理和单一动作;每个会话含多个回合,每回合最多N步后重置浏览器;从每回合选取截至最后一个正奖励步 $t^*$ 的前缀,无正奖励的回合丢弃。(4)训练:对每个保留步 $t\le t^*$ 预计算记忆token,构造三元组 $(o_t,M_t,a_t)$,最大化动作似然并联合更新对齐器 $W$,最终在ScaleWoB的86个应用上收集24k样本微调出JAMEL-9B。
技术新颖性
技术新颖性体现在多方面。首先是“记忆-探索相互依赖”的洞察被首次具象为可训练框架:以往潜变量记忆(如MemoryLLM、M+、TokMem)通常将记忆管理与学习目标解耦,JAMEL则把覆盖率奖励直接绑定为记忆是否有用的监督。其次是奖励设计的巧思——代码覆盖率天然确定、持久、跨回合不重置,避免了ICM/RND等预测误差信号可能被反复欺骗的缺陷,并自动形成由浅入深的课程。第三是极简记忆表示:每步历史仅用一个EOS隐状态向量压缩,再经线性对齐器注入策略,既大幅降低长轨迹计算开销,又保留了“软前缀”这一可微分接口让记忆可被端到端训练。最后,整条流水线标注免费、可自动扩展,为“探索-执行”范式和持续学习提供了可复用模板。
实验结果
在ScaleWoB基准(96个真实Web应用,86训练/10测试,每应用50步)上,JAMEL-9B取得累计覆盖奖励20.7,显著超过开源基线MAI-UI-8B(8.4)和Mobile-Agent-v3.5(5.9),并接近闭源ReAct-text(19.9)、仅落后ReAct-vision(20.9) 0.2。奖励曲线(图2)显示,本地基线因剪枝上下文而过早平台化,而JAMEL通过把全部历史压成记忆token避免了截断,保持与云模型相近的稳步上升轨迹。Token效率方面(表2),JAMEL共消耗1,061,103 token、每步2,122.2,仅为Mobile-Agent的1/2.76、MAI-UI的1/2.81、ReAct-Vision的1/21.92。逐应用分析(图3)揭示结构化探索规律:在Vipshop、Expedia、Temu等深度平台JAMEL持续上行;Youku、Keep等媒体应用各方法都早平台化;Alibaba、Taobao出现阶跃式奖励回升,显示其跳出局部最优的能力;但Pinduoduo等密集弹窗界面仍会让压缩记忆陷入长期平台。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| ScaleWoB 10个未见应用累计探索(开源对比) | 累计覆盖奖励(50步会话) | 20.7 (JAMEL-9B) | MAI-UI-8B: 8.4 / Mobile-Agent-v3.5: 5.9 | 较MAI-UI +146%、较Mobile-Agent +251% |
| ScaleWoB 与闭源模型对比 | 累计覆盖奖励 | 20.7 | ReAct-text 19.9 / ReAct-vision 20.9 (Gemini 3.1 Flash-Lite) | 超越ReAct-text,仅落后vision 0.2 |
| ScaleWoB 总token消耗 | 10应用×50步总输入token | 1,061,103 | ReAct-Vision 23,260,296 | 降低21.92倍 |
| ScaleWoB 每步token效率 | 平均每步输入token | 2,122.2 | MAI-UI 5,960.1 / Mobile-Agent 5,863.9 | 降低约2.8倍 |
局限与改进
作者承认JAMEL在密集弹窗界面仍会被困住:Pinduoduo等应用因持续出现的模态遮罩,智能体频繁尝试与背景元素交互,而这些元素被前景窗口禁用导致功能无响应,造成长时间探索平台。我自己观察到几点局限:一是新颖性奖励完全依赖GUI域的代码覆盖率,该信号无法迁移到具身/真实物理环境,论文虽提到可类比但未给出实验验证;二是训练数据由通用LLM生成,存在“老师能力天花板”问题,受限于教师模型的探索深度;三是仅在10个测试应用上评估,泛化性的统计置信度有限;四是只与少数基线对比,未涵盖更新的记忆机制。改进方向包括引入更鲁棒的记忆表示、扩大评测集、并探索跨域新颖性信号。
独立分析的弱点
独立分析的主要弱点:第一,奖励信号域特异性强——代码覆盖率只适用于可插桩的软件,对原生App、桌面端或非GUI具身环境无法直接获取,限制框架通用性,改进方向是设计可迁移的状态指纹或哈希型新颖性度量。第二,依赖教师模型采样的拒绝微调使探索能力受限于教师上界,难以发现教师都未触及的深层路径,可引入在线强化学习打破上界。第三,每步单token的极简压缩在超长轨迹或极密界面(如Pinduoduo)下信息损失明显,易陷入平台,可考虑分层记忆或动态token分配。第四,评测规模偏小(10应用×50步),方差较大,结论稳定性需更大规模验证。第五,当前是监督学习而非真正RL,未充分发挥新颖性奖励的课程潜力,与RL结合是明显缺口。
未来方向
作者明确提出两个方向。其一是探索新颖性驱动记忆架构的scaling law:整合强化学习策略,利用新颖性奖励自带的自然课程——从浅层交互平滑推进到深层多步导航,研究更大模型容量、更大数据采集、更多探索步数如何受益于该课程。其二是“记忆条件任务执行与持续学习”:先自主探索未知环境积累结构记忆,再依此执行具体用户指令,形成“探索-执行”范式,推动智能体自我演化、减少对人工标注轨迹的依赖,快速适配长尾场景。基于成果还可延伸:将代码覆盖率新颖性推广到原生App与具身环境的状态指纹;用在线RL替代拒绝微调以突破教师上界;设计分层记忆缓解超长轨迹压缩损失;构建跨域统一新颖性基准,验证记忆-探索闭环的通用性。
复现评估
复现评估较好。代码与模型已在 https://github.com/MobileLLM/JAMEL 开源。关键组件明确:压缩器为Qwen3-VL-2B-Instruct($d_c=2048$),策略为Qwen2.5-VL-7B-Instruct($d_{LM}=3584$),合成JAMEL-9B;训练数据为ScaleWoB的86个应用24k样本,该基准本身公开(GitHub SDK)。奖励计算依赖V8与Istanbul reporter,均为公开工具。复现难点在于:需在浏览器环境(BrowserGym)对Web应用做插桩采集覆盖率,工程链路较长;评测需部署10个未见应用并跑500步;算力上微调9B模型需相当GPU资源。总体属“中等偏难”复现:理论流程清晰、依赖工具开源,但数据采集与评测的工程化和算力门槛较高,且ScaleWoB部分真实应用(如微信、抖音)的稳定性可能影响结果。
论文图表
分应用展示累计奖励轨迹:Vipshop、Alibaba、Expedia、Taobao、Pinduoduo等。揭示结构化探索规律——深度平台(如Expedia)持续上行,密集界面(如Pinduoduo)出现长期平台。
该图展示了不同应用结构对探索的影响及JAMEL的优势与失败模式,是理解泛化性与局限性的关键。