Summary

把 working memory 管理做成可学习的策略动作(Context-as-action 路线的代表作):统一动作空间 A_task ∪ A_mem,memory 动作为 Prune&Write 算子(按 turn 级 ID 删历史记录 + 写入摘要,摘要自身可被后续再删);配套 DCPO 解决”上下文删除破坏因果 LM 单调增长假设”的训练难题(按 memory 动作点把轨迹切段、段内前缀固定、段继承轨迹级 advantage)。MemAct-RL-14B 多目标平均 59.1% 超过 16× 大的 Qwen3-235B(53.1%),token 成本 −51%。

本笔记以 v3(2026-05-07) 为准;v1(2025-10)的单目标数字与”≤3 目标训练”设定不同,引用时注意版本。

Problem & Motivation

长程 agent 的 context 管理现有方案(滑窗、摘要、外挂 memory)都是外部机制,不感知 agent 的推理状态,删或留的决策与任务优化脱节。本文的 formulation 转变:把”保留什么信息”本身并入 policy,让 RL 端到端联合优化信息保留与任务性能。

Method

  • 统一动作空间:state = 当前 working memory(交互记录序列,各带唯一 ID);memory 动作 a=(ℐ_target, c) 删指定 turn + 写入摘要/要点/反思,in-place 且可寻址(未来可再删)。
  • DCPO:删除使被删内容仍”物理残留”于后续 token 的内部表示(每个 token 的潜表示编码其全部前序),造成 train-inference mismatch,简单 attention masking 不够——必须物理重构轨迹。做法:按 K 个 memory 动作点切成 K+1 独立段,段内 context 前缀固定恢复标准因果训练;稀疏终局奖励(+1.0 成功 / −0.1 违反 20K token 或 40 步约束),全局归一 advantage 由各段继承,GRPO clip 目标。
  • 训练:SFT 冷启动 930 例(DeepSeek-V3.1 合成:8-16K 提示、>16K 强制 memory 动作,注入提示从数据中移除)→ 3,000 段;RL 10,240 轨迹(HotpotQA + Asearcher),刻意只训 ≤3 目标测泛化。Backbone Qwen2.5-7B/14B。

Key Results

  • 多目标(Table 1):MemAct-RL-14B 平均 0.591(2/4/6/8-obj: 0.660/0.591/0.570/0.543)> Qwen3-235B 0.531;token 8.2 vs 16.7×10⁴(−51%);每步平均输入 context 仅 ~3,500 tokens。
  • 单目标五 benchmark(2Wiki/HotpotQA/Bamboogle/Frames/BrowseComp):与 Search-R1 基本持平(avg 0.537 vs 0.535),但 7B 版总时长 −40%(SGLang 2,000 轨迹实测)。
  • 泛化:只训 ≤3 目标 → 8 目标仍 54.3%(Search-R1 39.3%);235B/Tongyi-DeepResearch 在 >4 目标后饱和。
  • 策略随容量分化(Fig 5):7B 高频粗粒度(~6 条/次);14B 双峰(~2 条细粒度=推理中除噪 + ~6 条粗粒度=子目标完成后清场)。
  • 消融:RL 不可或缺(SFT 0.553 → RL 0.591);固定间隔(每 5 turn)0.582,静态调度在高复杂度下删关键信息。

Evidence Ledger

Claim IDClaimTypeSource locatorEvidence excerptStatus
C1统一动作空间 + Prune&Write(turn 级 ID 删 + 写,可寻址可再删)benchmark-settingv3 §Method”amem=(ℐtarget,c)“source-verified
C2DCPO:删除致 train-inference mismatch → 轨迹切 K+1 段、段内前缀固定、段继承轨迹 advantagecausal-mechanismv3 §DCPO”re-organized into K+1 independent segments”source-verified
C314B 多目标 0.591 > Qwen3-235B 0.531;token 8.2 vs 16.7×10⁴;每步 ~3,500 tokenscomparisonv3 Table 1”0.591 / 8.2 vs 0.531 / 16.7”source-verified
C4单目标与 Search-R1 持平(0.537 vs 0.535);7B 时长 −40%(SGLang 2,000 轨迹)numberv3 Table 1 + §efficiency”reduces total duration by 40%“source-verified(v3 已用 BrowseComp 替换 v1 的 Musique)
C5训 ≤3 目标 → 8 目标 54.3% vs Search-R1 39.3%;大模型 >4 目标饱和numberv3 §4.6”54.3% … surpassing the 39.3%“source-verified
C67B 粗粒度高频 vs 14B 双峰(~2/~6 条)numberv3 §4.6 Fig 5”bimodal pattern with peaks at fine-grained (∼2) and coarse-grained (∼6)“source-verified
C7SFT 0.553→RL 0.591;固定间隔 0.582numberv3 Table 1”static schedules often delete critical information”source-verified
C8SFT 930 例→3,000 段(提示注入后移除);RL 10,240 轨迹;奖励 +1.0/−0.1(20K token/40 步约束)benchmark-settingv3 §Setup”930 accurate examples … 3,000 training segments”source-verified
C9自认局限:稀疏奖励难归因具体 memory 动作;摘要不可恢复;failure 含 memory 幻觉benchmark-settingv3 §Limitations”difficult to accurately assign credit to specific memory actions”source-verified

Strengths & Weaknesses

Strengths

  • Context-as-action 路线最完整的实例:formulation(统一动作空间)、训练可行性(DCPO 段切分是真贡献——直面”删上下文违反因果 LM 假设”而非回避)、行为分析(策略随容量分化的双峰证据)三层齐全。
  • “刻意只训 ≤3 目标”的泛化设计干净;对照覆盖静态调度与 SFT-only,学习价值的归因链完整。

Weaknesses / 边界

  • 任务全是文本检索 QA(HotpotQA/BrowseComp 类),无 GUI/视觉 observation;对 CUA 的 context 管理(CUA-Survey §6.9.1 Context-as-action 行)是邻接可迁移证据。
  • 2606-SkillMemoryBudget 的关系需谨慎:那篇打击的是 online 外挂模块(检索/注入式),MemAct 是内化进 policy 的路线且 RL 训练属 offline 摊销——不在其打击范围,但 MemAct 未报 budget-matched vanilla(多步基线)对照的多 run 方差,其 vs Search-R1 的持平结论若按该标准审视余量不大。
  • 作者自认稀疏奖励难归因到具体 memory 动作——正是 FoldAct(2512.22733,queue 中)声称的攻击面(摘要动作制造非平稳观察分布);两篇对读后再定此路线的成色。
  • v1→v3 数字变动较大,跨版本引用有污染风险。

对领域:把 context 管理从 heuristic 变成 learnable 的最强正面证据;与 2605-MaskingRegimeMap(heuristic masking 的 regime 依赖)呼应——learned 策略正是后者留空的”能否自适应定位 regime”的候选答案。

Mind Map

mindmap
  root((MemAct))
    Problem
      外部context管理不感知推理状态
    Method
      统一动作空间PruneWrite
      DCPO轨迹段切分
      稀疏终局奖励GRPO
    Results
      14B超235B且token减半
      训3目标泛化到8
      策略随容量双峰分化

Notes

  • 入队来源:2026-07-27-WebAgent-RL-and-Context-Landscape 交叉轴主线 top-10 之一(“若只读十篇”排第 1)。
  • context-folding 家族链:本篇(memory 编辑动作)→ Context-Folding 2510.11967(分支-折叠)→ FoldAct 2512.22733(理论反驳:摘要动作制造非平稳观察分布)——后两篇在 queue,读完后应对三篇做一致性对读(FoldAct 的攻击是否命中 DCPO 的段切分方案)。