通过自生成问题保持模型推理多样性,显著提升复杂任务表现。
Beyond Pass@1: Self-Play with Variational Problem Synthesis Sustains RLVR
- 用模型正确解答自动生成新问题,维持策略熵不下降
- 在AIME24/25上Pass@32提升18.3%与22.8%
- 适用于不同规模模型的复杂推理任务优化
强化学习结合可验证奖励(RLVR)已成为大语言模型后训练阶段解决复杂推理任务的关键范式。然而,传统RLVR训练会因降低策略熵而削弱生成多样性,限制了Pass@k性能,而Pass@k通常代表了大模型推理能力的上限。本文从训练问题角度系统分析策略生成多样性,发现更新和扩充训练问题有助于缓解熵坍缩。基于此,提出在线自对弈变分问题生成(SvS)策略:利用模型的正确解自动生成变体问题,同时保持参考答案与原题一致。该自进化机制有效维持训练过程中的策略熵,显著提升Pass@k表现。在竞赛级AIME24与AIME25基准上,Pass@32绝对提升分别达18.3%与22.8%,代码生成任务亦有明显增益。在12个不同规模(3B至32B)的推理基准上实验均表明SvS具有强泛化性与鲁棒性。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR) has recently emerged as a key paradigm for post-training Large Language Models (LLMs), particularly for complex reasoning tasks. However, vanilla RLVR training has been shown to improve Pass@1 performance at the expense of policy entropy, leading to reduced generation diversity and limiting the Pass@k performance, which typically represents the upper bound of LLM reasoning capability. In this paper, we systematically analyze the policy's generation diversity from the perspective of training problems and find that augmenting and updating training problems helps mitigate entropy collapse during training. Based on these observations, we propose an online Self-play with Variational problem Synthesis (SvS) strategy for RLVR training, which uses the policy's correct solutions to synthesize variational problems while ensuring their reference answers remain identical to the originals. This self-improving strategy effectively maintains policy entropy during training and substantially improves Pass@k compared with standard RLVR, sustaining prolonged improvements and achieving absolute gains of 18.3% and 22.8% in Pass@32 performance on the competition-level AIME24 and AIME25 benchmarks, as well as on code generation tasks. Experiments on 12 reasoning benchmarks across varying model sizes from 3B to 32B consistently demonstrate the generalizability and robustness of SvS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。