Training | | arXiv:2607.18722

Stale but Stable: Staleness-Adaptive Trust Regions for Stabilizing Asynchronous Reinforcement Learning

陈旧但稳定:面向异步强化学习的陈旧自适应信任域

🧭 异步 RL 训练稳定性优化

Tencent Hy LLM Frontier · National University of Singapore · University of Maryland, College Park · University of Georgia · Indiana University

Junyao Yang, Yucheng Shi, Zongxia Li, Zhongzhi Li, Ruhan Wang, Xiangxin Zhou 等

用 detached log-ratio 作为陈旧度代理,自适应收缩 PPO 区间端点,稳定异步 RL 训练

📌 工作参考价值

  • 在电商/广告推荐系统的在线异步 RL 训练(如 PPO 策略梯度)中,可直接复用 detached log-ratio 作为 staleness 代理,无需额外模型即可量化策略与行为策略的偏差,据此识别高风险更新样本。
  • 采用 仅收缩 sign-selected 端点 的区间约束,在保留大部分 token 原有 PPO 行为的同时,仅对高陈旧度导致的异常外向更新进行保守截断,实现细粒度稳定控制。
  • 针对推荐模型可能采用的 MoE 架构,路由重放(routing replay) 技巧可缓解异步陈旧带来的路由不一致问题,与裁剪配合形成互补稳定器,适合大规模稀疏激活的推荐模型训练。
  • 工程上,该方法在 SGLang 多推理引擎 + Megatron 异步训练框架下验证有效,可直接迁移到类似解耦架构的推荐 Agent 在线学习系统中,提升训练吞吐下的策略稳定性。

动机:异步 RL 在提升吞吐的同时引入陈旧样本,落后策略、引擎延迟和 MoE 路由进一步放大陈旧度,PPO 的 clipping 仅是向外更新的采样代理,无法有效约束高陈旧更新。

方法:提出 陈旧自适应信任域(SAT),用 detached 采样 log-ratio 作为实用陈旧度代理,通过 基于陈旧度的高斯核缩放函数 识别 batch 内高偏差尾部,并仅对 PPO 原始区间的 sign-selected 端点 施加有效收缩。此举在普通 token 上保持 PPO 行为,而对新截获的极端外向更新施加更保守的约束,从理论上证明了 局部区间包含和逐点悲观性。

结果:在 Qwen3-30B-A3B + SGLang + Megatron 的解耦异步 RL 设置中,SAT-GSPO w/ R3 在 lag1 达到 AIME24 avg@8 35.83,lag8 达到 34.79;SAT-GSPO 在 lag1 达到 34.17。自适应裁剪与路由重放互为补充,共同稳定异步训练。