arXiv:2606.12370cs.LGcs.CL2026-06被引 4

通过拒绝采样优化多标记预测,显著提升强化学习训练速度。

Breaking Entropy Bounds: Accelerating RL Training via MTP with Rejection Sampling

论文配图:Breaking Entropy Bounds: Accelerating RL Training via MTP with Rejection Sampling
图 1 · 摘自论文原文
  • 用拒绝采样替代贪婪采样,缓解模型熵波动对生成质量的影响。
  • 提出端到端的电视距离损失,使接受率提升10%,最高达95%。
  • 预训练阶段完成模型优化,避免在线更新,适合大规模强化学习部署。

强化学习(RL)已成为现代大语言模型的关键组件,但推理阶段仍是训练流程的主要瓶颈。尽管多标记预测(MTP)可通过推测解码自然加速推理,但许多研究发现,RL训练过程中MTP接受率显著下降,导致加速效果有限。为此,我们提出Bebop,系统研究了大语言模型后训练中MTP的应用,并提供可落地的集成方案。首先,我们揭示MTP接受率受模型熵波动的根本限制,其与熵值呈明显的负线性关系。其次,相比贪婪草案采样,概率性拒绝采样能有效缓解熵扰动的影响;同时发现传统交叉熵或KL损失在该场景下次优,因此提出一种端到端的电视距离(TV)损失,直接优化多步拒绝采样接受率,实现约10%的接受率提升,最高可达95%,并在数学推理、代码生成和智能体任务中带来最高25%的推理吞吐增益。第三,我们测试多种在线MTP训练策略,发现预训练阶段结合端到端TV损失与拒绝采样,可在整个RL过程中保持稳定接受率和加速比,无需高成本的在线更新。大量实验验证:本方法在Qwen3.5、Qwen3.6、Qwen3.7模型的异步强化学习训练中,实现了最高1.8倍的整体加速。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has become a key component in modern large language models, yet the rollout stage remains the key bottleneck in RL training pipelines. Although Multi-Token Prediction (MTP) offers a natural solution to accelerate rollouts through speculative decoding, many studies have observed that MTP acceptance rates degrade significantly during RL training, leading to limited speedup performance. To address this bottleneck, we present Bebop, a systematic study of MTP in LLM post-training, and offer practical recipes to integrate MTP into large-scale RL pipelines. First, we reveal that the MTP acceptance rate is fundamentally bounded by the fluctuation of model entropy, which demonstrates a clear negative linear relationship with the rise of entropy in the RL stage. Second, we show that probabilistic rejection sampling largely alleviates the disturbance introduced by entropy in RL compared to greedy draft sampling. We further identify that the conventional MTP training objectives (cross-entropy or KL) are suboptimal in such settings, and therefore we propose a novel end-to-end TV loss that directly optimizes multi-step rejection sampling acceptance rate, yielding ~10% acceptance rate improvements, achieving up to 95% acceptance rates and up to 25% extra inference throughput gains across mathematical reasoning, code generation, and agentic tasks. Third, we test various online MTP training strategies during RL and show that pre-RL MTP training with e2e TV loss and rejection sampling achieves a consistent acceptance rate and speedup throughout the entire RL, eliminating the need for costly online MTP updating. We provide extensive experiments and analysis that validate our findings. Experimental results show our method achieves up to 1.8x end-to-end acceleration in async RL training of Qwen3.5, Qwen3.6, and Qwen3.7 models.

强化学习生成加速拒绝采样多标记预测

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