让难提示更受重视,加速大模型推理训练收敛
Asymmetric Prompt Weighting for Reinforcement Learning with Verifiable Rewards
- 对难提示赋予更高权重,打破传统对易难提示的对称处理
- 在从零训练场景下显著提升收敛速度,尤其在低成功率阶段
- 理论证明该策略能最小化达到目标准确率所需时间
基于可验证奖励的强化学习推动了大模型后训练中推理能力的进步。现有算法如GRPO、DAPO和RLOO主要关注中等难度提示,降低极易和极难提示的梯度权重。本文提出非对称提示加权策略,对低甚至零成功率提示赋予更高权重。实验发现该策略在从零开始的强化学习(如R1-Zero)中效果显著,而在高精度微调后的训练中收益较小。理论分析表明,在固定更新预算下,该加权方式能最小化将成功率从初始水平提升至目标精度所需时间。在低成功率阶段,有效响应稀少且响应成本主导时,最优权重呈现非对称性,通过提高低成功率提示权重,加速有效时间收敛。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards has driven recent advances in LLM post-training, in particular for reasoning. Policy optimization algorithms generate a number of responses for a given prompt and then effectively weight the corresponding gradients depending on the rewards. The most popular algorithms including GRPO, DAPO, and RLOO focus on ambiguous prompts, i.e., prompts with intermediate success probability, while downgrading gradients with very easy and very hard prompts. In this paper, we consider asymmetric prompt weightings that assign higher weights to prompts with low, or even zero, empirical success probability. We find that asymmetric weighting particularly benefits from-scratch RL (as in R1-Zero), where training traverses a wide accuracy range, and less so in post-SFT RL where the model already starts at high accuracy. We also provide theory that characterizes prompt weights which minimize the time needed to raise success probability from an initial level to a target accuracy under a fixed update budget. In low-success regimes, where informative responses are rare and response cost dominates, these optimal weights become asymmetric, upweighting low success probabilities and thereby accelerating effective-time convergence.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。