arXiv:2605.05112cs.LG2026-05

通过控制通过率提升代码生成强化学习的效率与效果。

Rollout Pass-Rate Control: Steering Binary-Reward RL Toward Its Most Informative Regime

论文配图:Rollout Pass-Rate Control: Steering Binary-Reward RL Toward Its Most Informative Regime
图 1 · 摘自论文原文
  • 用前缀重播技术引导强化学习走向50%通过率的最优区域。
  • 在SWE-bench上实现2.01倍加速,14B模型准确率从0.274升至0.295。
  • 适合追求高效代码生成和强化学习优化的研究者使用。

软件工程中的智能体强化学习(RL)耗费大量算力在状态依赖轨迹上,其分组二元奖励高度偏斜且对比性弱。本文将其建模为通过率控制问题,证明在四种标准下(奖励熵、组过滤存活率、组相对策略优化下的留一法优势能量、成功-失败对数量),当通过率接近50%时,二元奖励信号最强。提出前缀重播(Prefix Sampling, PS)方法,通过重放自生成的轨迹前缀,将偏斜组导向该最优区间:成功前缀为多数失败组提供优势起点,失败前缀则削弱多数通过组。重播状态通过原有回放路径重构,重播词元从损失中掩码,使优化仅作用于当前策略延续部分。在SWE-bench Verified上,PS在评估波动范围内达到基线高分,对Qwen3-14B和32B分别实现2.01倍和1.55倍端到端时钟速度提升;14B模型峰值准确率由0.274提升至0.295。AIME 2025在4B和8B模型上的实验验证了相同的通过率控制规律,4B消融实验表明增益来自重播机制、双向覆盖与自适应控制。

原文摘要 · Abstract (English)

Agentic reinforcement learning (RL) for software engineering spends much of its compute on stateful trajectories whose grouped binary rewards are highly skewed and weakly contrastive. We frame this as pass-rate control and show that the binary reward-side signal is strongest near a 50% rollout pass rate under four criteria: reward entropy, group-filtering survival, leave-one-out (RLOO) advantage energy under Group Relative Policy Optimization (GRPO), and success-failure pair count. We propose Prefix Sampling (PS), which replays self-generated trajectory prefixes to steer skewed groups toward this regime: successful prefixes give mostly failing groups a head start, while failing prefixes handicap mostly passing groups. Replayed states are reconstructed through the existing rollout path, and replayed tokens are masked from the loss so optimization applies only to current-policy continuations. On SWE-bench Verified, PS reaches the baseline high-score regime within evaluation variability while delivering 2.01x and 1.55x end-to-end wall-clock speedups on Qwen3-14B and Qwen3-32B; the 14B peak improves from 0.274 to 0.295. AIME 2025 experiments on 4B and 8B show the same pass-rate-control pattern, and 4B ablations attribute gains to replay, bidirectional coverage, and adaptive control.

强化学习代码生成效率优化通过率控制

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