用新算法让大模型突破推理瓶颈,解决更难的数学题。
Unlocking Reasoning Capabilities in LLMs via Reinforcement Learning Exploration
- 改用前向KL正则化,鼓励模型跳出原有思维范围探索
- 在AIME2024/2025测试中,模型表现超越基础版本并攻克难题
- 适合研究强化学习与大模型推理能力的学者参考
基于可验证奖励的强化学习(RLVR)近期提升了大语言模型(LLMs)的推理能力,尤其在数学问题求解方面。然而,随着采样预算增加,RLVR训练模型相对于预训练基线的优势常减弱甚至消失,暴露了对基模型搜索空间的强依赖。我们归因于广泛使用的反向KL散度正则化器的模式聚焦行为,使策略困于基模型支持域内,限制了广泛探索。为此,提出RAPO(奖励感知策略优化)算法,通过(i)用前向KL惩罚替代反向KL以实现分布外探索,(ii)重加权参考策略以支持自适应分布内探索。我们在8K SimpleRL-Zero数据集上,无需监督微调地训练Qwen2.5-3B和7B模型,并在AIME2024和AIME2025上评估。结果表明,RAPO持续提升解题性能,显著突破基线性能上限,解决了此前无法处理的问题,推动了复杂推理任务下RLVR的边界。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards (RLVR) has recently enhanced the reasoning capabilities of large language models (LLMs), particularly for mathematical problem solving. However, a fundamental limitation remains: as the sampling budget increases, the advantage of RLVR-trained models over their pretrained bases often diminishes or even vanishes, revealing a strong dependence on the base model's restricted search space. We attribute this phenomenon to the widespread use of the reverse Kullback-Leibler (KL) divergence regularizer, whose mode-seeking behavior keeps the policy trapped inside the base model's support region and hampers wider exploration. To address this issue, we propose RAPO (Rewards-Aware Policy Optimization), an algorithm to promote broader yet focused exploration. Our method (i) utilizes the forward KL penalty to replace the reverse KL penalty for out-of-distribution exploration, and (ii) reweights the reference policy to facilitate adaptive in-distribution exploration. We train Qwen2.5-3B and 7B models with RAPO on the 8K SimpleRL-Zero dataset, without supervised fine-tuning, and evaluate them on AIME2024 and AIME2025. Results show that RAPO consistently improves problem-solving performance. Notably, RAPO enables models to surpass the base model's performance ceiling and solves previously intractable problems, advancing the frontier of RLVR for challenging reasoning tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。