提出异步强化学习框架,显著提升大模型训练效率且保持精度。
AsyncOPD: How Stale Can On-Policy Distillation Be?

- 采用异步管道分离数据生成与模型更新,缓解推理耗时瓶颈。
- 实验表明异步训练吞吐量提升1.6至3.8倍,精度损失可忽略。
- 针对过时数据问题设计新优化策略,适合大规模语言模型训练场景。
基于策略的蒸馏(OPD)通过教师反馈指导学生模型自举推演,正成为大语言模型后训练的重要方法。然而,如同强化学习,其面临策略内系统瓶颈,推演过程占据训练时间主导。异步训练管道可通过解耦推演生成与学习更新缓解此瓶颈,但引入过时策略数据。现有研究虽探讨异步强化学习中的过时数据问题,但在OPD中的影响仍不明确。本文首次系统研究异步OPD中的过时性问题,聚焦实际场景:教师反馈使用局部KL损失,全词汇表教师输出因存储或传输成本过高而需有限缓存。我们发现,KL方向决定过时数据敏感度:教师加权前向KL对过时推演更鲁棒,学生加权反向KL则易受干扰。针对后者,我们验证了用于稳定异步强化学习的方法在OPD中无效;相比之下,一种简单替代方案——在学习时刻用当前学生重算反向KL信号——表现更优。此外,分析有限教师评分缓存导致稀疏与采样反向KL估计器的偏差-方差权衡,由此启发多样本蒙特卡洛(MC)方法,既保持估计一致性又降低单样本方差。最后,我们发布开源异步OPD训练框架AsyncOPD,基于上述估计器选择。实验显示,相比严格同步训练,AsyncOPD将训练吞吐量提升1.6至3.8倍,同时达到相当的准确率。
原文摘要 · Abstract (English)
On-policy distillation (OPD) trains a student on its own rollouts guided by teacher feedback and is becoming increasingly important for large language model (LLM) post-training. Like reinforcement learning (RL), however, OPD faces an on-policy systems bottleneck, as rollouts can dominate training time for reasoning workloads. Asynchronous training pipelines can alleviate this bottleneck by decoupling rollout generation from learner updates, but doing so introduces stale-policy data. While prior work has studied stale data in asynchronous RL, its effects in OPD remain underexplored. We present the first systematic study of staleness in asynchronous OPD, focusing on a practical setting where teacher feedback is implemented through local KL losses and full-vocabulary teacher logits are too expensive to store or transfer, necessitating finite teacher-score caches. We first show that KL direction changes the stale-data problem: teacher-weighted forward KL is more robust to stale rollouts, whereas student-weighted reverse KL is vulnerable. Second, for this vulnerable reverse-KL case, we study whether methods designed to stabilize asynchronous RL can mitigate OPD staleness. In our experiments, they do not improve over a simpler OPD-specific surrogate: recomputing the reverse-KL signal under the current student at learner time. Third, we analyze how finite teacher-score caches create a bias-variance tradeoff for sparse and sampled reverse-KL OPD estimators. This motivates multi-sample Monte Carlo (MC), which preserves MC correctability while reducing one-sample variance. Finally, we present and open-source AsyncOPD, a fully asynchronous OPD training pipeline built from these estimator choices. Experiments show that AsyncOPD improves training throughput by $1.6\times$ to $3.8\times$ over strict synchronous training while reaching comparable accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。