arXiv:2605.08441cs.LGcs.AI2026-05被引 1

DUET通过动态分配生成长度和次数,用更少算力提升强化学习训练效率与质量。

DUET: Optimize Token-Budget Allocation for Reinforcement Learning with Verifiable Rewards

论文配图:DUET: Optimize Token-Budget Allocation for Reinforcement Learning with Verifiable Rewards
图 1 · 摘自论文原文
  • 双维度优化:按提示重要性分配滚动次数,用重要性重加权的终止规则控制生成长度。
  • 仅用50%算力仍优于全预算基线,训练速度提升2.51倍,数学与编码任务表现更好。
  • 适用于各类大模型,算力越紧效果越优,适合追求高效高质量训练的研究者。

基于可验证奖励的强化学习(RLVR)每轮训练生成数十万令牌,其中采样生成占主导计算开销。整体令牌预算可通过两个主要维度控制:(i) 决定哪些提示分配采样,(ii) 决定每次采样的长度。以往工作通常仅控制其中一个维度。本文提出双控令牌分配方法DUET,其在GRPO基础上构建轻量预采样代理以判断提示信息量,决定分配给每个提示的采样数量;并引入标记门控中止规则与重要性重加权机制,决定何时终止采样。在Qwen3-1.7B模型上训练数学数据集MATH时,DUET优于全预算的GRPO及其他三种预算感知基线方法。该优势在数学、编程等多任务上均有效,科学问答领域性能与最优基线持平,且实现1.62倍的墙钟加速。尤为显著的是,仅使用50%令牌预算的DUET仍超越所有基线在全预算下的表现,相较全预算GRPO提速2.51倍。在其他主干LLM(如Qwen3-4B、Llama-3.2-3B-Instruct)上也验证了高表现。值得注意的是,随着预算收紧,DUET与最强基线的差距反而扩大,打破高效方法随算力减少而性能下降的常见规律。结果表明,该预算控制策略不仅加速训练,更能提升学习信号质量。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) generates hundreds of thousands of tokens per training step, with rollout generation dominating the computational cost. The overall token budget can be controlled along two main dimensions: (i) deciding which prompts to allocate rollouts to, and (ii) deciding how long each rollout should be. Prior work has generally controlled only one of these dimensions at a time. We show that jointly tuning both decisions under a shared compute budget improves both reasoning quality and wall-clock training time. We instantiate this view as \textbf{DU}al-controlled tok\textbf{E}n alloca\textbf{T}ion (DUET), a computationally efficient layer over GRPO that uses a lightweight pre-rollout surrogate of prompt informativeness to set how many rollouts each prompt receives, and a marker-gated abort rule with importance reweighting to set when to stop them. On Qwen3-1.7B trained on MATH, DUET outperforms full-budget GRPO and the other three budget-aware baseline methods. DUET's advantage further generalizes to other benchmarks across math and coding, and is on par with the best baseline on the scientific Q\&A domain, while also achieving a $1.62\times$ wall-clock speedup. More notably, using only 50\% of the token budget, DUET still outperforms all baseline methods at their full budget, achieving an even higher $2.51\times$ speedup over full-budget GRPO. We verify the high performance of DUET on other backbone LLMs, including Qwen3-4B and Llama-3.2-3B-Instruct. Notably, the gap between DUET and the strongest baseline \emph{widens} as the budget tightens, contrary to the usual pattern in which efficient methods trade off quality as compute decreases. More broadly, these results suggest that DUET budget-aware control strategies are valuable not only for accelerating training, but also for improving the quality of the learning signal.

强化学习令牌优化高效训练大模型

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