arXiv:2605.12070cs.LGcs.AI2026-05被引 3

异步强化学习中丢失旧日志导致修正失效,提出修复方法提升训练效率

Missing Old Logits in Asynchronous Agentic RL: Semantic Mismatch and Repair Methods for Off-Policy Correction

论文配图:Missing Old Logits in Asynchronous Agentic RL: Semantic Mismatch and Repair Methods for Off-Policy Correction
图 1 · 摘自论文原文
  • 分离采样与优化时丢失旧日志,使偏差与过时纠正混淆
  • 新方法通过快照、专用模型或中断同步获取旧日志,恢复正确修正
  • 采用改进PPO-EWMA,在不增加系统开销下显著提升训练速度与性能

异步强化学习通过解耦采样生成与策略优化,提升大语言模型智能体的推理吞吐量,但其在基于PPO的离线策略修正中引入关键缺陷。在异构训练系统中,重要性比率应分解为两个语义不同的部分:训练-推理差异项,用于对齐同一行为策略版本下的训练侧与推理侧分布;政策过时项,用于约束从历史策略到当前策略的更新。然而,实际异步流水线因延迟更新和部分回溯,常丢失所需的旧训练侧日志(old logits)。该缺失问题将差异修复与过时纠正纠缠,破坏了解耦修正的预期语义,导致剪裁与掩码阈值产生不良交互。为此,我们研究了精确与近似修正路径:提出三种精确获取旧日志的方法——基于快照的版本追踪、专用旧日志模型、通过部分回溯中断同步,并比较其系统权衡。从近似修正角度,我们聚焦于在无法低成本恢复精确旧日志时,通过更合适的近似策略保留解耦修正优势,避免额外系统开销。基于此分析,采用改进的PPO-EWMA方法,在保持低系统开销前提下实现训练速度与优化性能的显著提升。

原文摘要 · Abstract (English)

Asynchronous reinforcement learning improves rollout throughput for large language model agents by decoupling sample generation from policy optimization, but it also introduces a critical failure mode for PPO-style off-policy correction. In heterogeneous training systems, the total importance ratio should ideally be decomposed into two semantically distinct factors: a \emph{training--inference discrepancy term} that aligns inference-side and training-side distributions at the same behavior-policy version, and a \emph{policy-staleness term} that constrains the update from the historical policy to the current policy. We show that practical asynchronous pipelines with delayed updates and partial rollouts often lose the required historical training-side logits, or old logits. This missing-old-logit problem entangles discrepancy repair with staleness correction, breaks the intended semantics of decoupled correction, and makes clipping and masking thresholds interact undesirably. To address this issue, we study both exact and approximate correction routes. We propose three exact old-logit acquisition strategies: snapshot-based version tracking, a dedicated old-logit model, and synchronization via partial rollout interruption, and compare their system trade-offs. From the perspective of approximate correction, we focus on preserving the benefits of decoupled correction through a more appropriate approximate policy when exact old logits cannot be recovered at low cost, without incurring extra system overhead. Following this analysis, we adopt a revised PPO-EWMA method, which achieves significant gains in both training speed and optimization performance.

强化学习大模型异步训练

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