arXiv:2608.07068cs.AI2026-08

通过记忆状态对齐,让长时序智能体在稀疏奖励下仍能高效学习。

MemOPD: On-Policy Distillation through Memory State Alignment for Long-Horizon Agents

论文配图:MemOPD: On-Policy Distillation through Memory State Alignment for Long-Horizon Agents
图 1 · 摘自论文原文
  • 用记忆状态重建技术确保教师评估与学生动作处于同一真实状态
  • 在相同任务上比传统方法提升7.0%的F1值,整体性能提升416.2%
  • 适合需要长时序决策的强化学习场景,如对话系统、机器人规划

长时序智能体在交互中累积不断增长的上下文,影响性能与稳定性。紧凑记忆通过压缩和重写模型调用间的历史信息缓解此问题。通常学习保留内容依赖于最终任务奖励的近端策略优化(PPO),但稀疏奖励难以指导每次记忆更新。为此提出在线策略蒸馏(OPD),为学生轨迹提供密集教师监督。为保证监督有效性,教师必须在动作生成的相同状态下评估该动作。然而,记忆压缩过程中的上下文重写会破坏这种状态对齐。当采样响应被保留并重新编码以供后续调用时,将交互扁平化为持久历史可能导致教师在学生从未访问过的状态下评分。动作虽源于本策略,但状态不一致。因此我们提出记忆对齐的在线策略蒸馏(MemOPD)。MemOPD记录每次模型调用的输入与采样输出,恢复原始标记位置和因果可见性,并打包重构的调用以实现高效教师评分。教师在采样动作位置提供全词汇监督,同时PPO维持最终任务目标。实验验证了多个上下文更新下的状态对齐,结果显示其比持久历史教师评分提升7.0% F1。总体而言,MemOPD-3B相比PPO最高提升416.2%,打包机制使训练期间演员计算速度提升达1.63倍。代码已公开:https://github.com/TPssp/MemOPD。

原文摘要 · Abstract (English)

Long-horizon agents accumulate growing contexts during interaction, impairing performance and stability. Compact memory mitigates this problem by compressing and rewriting the history retained between model invocations. Learning what to retain typically relies on proximal policy optimization (PPO) with final task rewards, but sparse rewards provide little guidance for individual memory updates. This limitation motivates on-policy distillation (OPD), which supplies dense teacher supervision on student rollouts. For such supervision to be valid, the teacher must evaluate each sampled action under the same state in which it was generated. However, the context rewriting performed during memory compression can break this alignment. When sampled responses are retained and re-encoded for later invocations, flattening the interaction into a persistent history may cause the teacher to score the action under a state that the student never visited during rollout. The action therefore remains on-policy by provenance, but not necessarily by state. We therefore propose Memory-Aligned On-Policy Distillation (MemOPD). MemOPD records the inputs and sampled outputs of each model invocation, restores its original token positions and causal visibility, and packs the reconstructed invocations for efficient teacher scoring. The teacher provides full-vocabulary supervision at the sampled action positions, while PPO preserves the final task objective. Experiments verify state alignment across several context updates and show that it improves F1 by 7.0% over persistent-history teacher scoring in a matched control. Overall, MemOPD-3B improves F1 over PPO by up to 416.2%, while packing yields up to a 1.63x speedup in actor computation during training. The code for this work is publicly available at: https://github.com/TPssp/MemOPD.

强化学习长时序决策策略蒸馏记忆对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。