arXiv:2602.03452cs.LGcs.AI2026-02

通过正负样本配对提升语言模型数学推理能力

Beyond Variance: Prompt-Efficient RLVR via Rare-Event Amplification and Bidirectional Pairing

  • 用难但可解和易但脆弱的提示配对,生成双向学习信号
  • 在Qwen2.5-Math-7B上,AIME2025 Pass@8提升至22.2,AMC23 Pass@64达97.0
  • 仅需单对提示每步更新,效果媲美1209个提示的大规模训练

强化学习结合可验证奖励(RLVR)在训练大语言模型进行确定性结果推理任务时表现优异。已有研究显示,少量提示即可有效训练,但提示选择常依赖训练准确率方差,导致优化方向不稳定且泛化能力弱。本文从机制层面重新审视提示选择,提出有效小批量应同时具备可靠的正向锚点与来自罕见失败的明确负向信号。为此,我们提出正-负配对策略:每次更新采样一个难但可解的提示 $q^{+}$(低成功率)和一个易但脆弱的提示 $q^{-}$(高成功率但不完美),基于多次推演的实测成功率区分。进一步引入加权GRPO,对成对结果重加权,使用组归一化优势,将 $q^{+}$ 上的稀有成功放大为强烈正向引导,同时将 $q^{-}$ 上的罕见失败转为强负惩罚。这种双向信号为成功与失败提供丰富反馈,提升样本效率且不抑制探索。在 Qwen2.5-Math-7B 上,每步仅用一对提示的训练,显著优于基于方差选择的GRPO基线:AIME 2025 Pass@8 从16.8升至22.2,AMC23 Pass@64 从94.0升至97.0,性能接近使用1209个提示的大规模训练。在 Qwen2.5-Math-7B-Instruct 上也取得类似提升。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) is effective for training large language models on deterministic outcome reasoning tasks. Prior work shows RLVR works with few prompts, but prompt selection is often based only on training-accuracy variance, leading to unstable optimization directions and weaker transfer. We revisit prompt selection from a mechanism-level view and argue that an effective minibatch should provide both (i) a reliable positive anchor and (ii) explicit negative learning signals from rare failures. Based on this principle, we propose \emph{positive--negative pairing}: at each update, we sample a hard-but-solvable $q^{+}$ and an easy-but-brittle prompt $q^{-}$(high success rate but not perfect), characterized by low and high empirical success rates under multiple rollouts. We further introduce Weighted GRPO, which reweights binary outcomes at the pair level and uses group-normalized advantages to amplify rare successes on $q^{+}$ into sharp positive guidance while turning rare failures on $q^{-}$ into strong negative penalties. This bidirectional signal provides informative learning feedback for both successes and failures, improving sample efficiency without suppressing exploration. On Qwen2.5-Math-7B, a single paired minibatch per update consistently outperforms a GRPO baseline that selects two prompts via commonly used variance-based selection heuristics: AIME~2025 Pass@8 improves from 16.8 to 22.2, and AMC23 Pass@64 from 94.0 to 97.0, while remaining competitive with large-scale RLVR trained from a pool of 1209 training prompts. Similar gains are observed on Qwen2.5-Math-7B-Instruct.

强化学习提示工程数学推理样本效率

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