arXiv:2507.02834cs.LGcs.CL2025-07NeurIPS被引 10

让AI通过自我解释生成高质量推理样本,突破复杂问题的思考瓶颈。

ExPO: Unlocking Hard Reasoning with Self-Explanation-Guided Reinforcement Learning

  • 用正确答案反推推理路径,生成符合当前模型能力的优质正样本。
  • 在MATH Level-5上超越专家示范方法,提升学习效率与最终准确率。
  • 可适配GRPO、DPO等主流强化学习框架,适合复杂推理任务研究者。

通过强化学习实现自进化在复杂推理任务中常因初始生成正样本能力不足而失败。现有基于GRPO的后训练方法依赖模型已有的正确输出,导致仅强化已有知识(分布聚焦),无法探索新解题路径。为突破此限制,模型需在当前输出分布之外进行有效探索,但需优质正样本引导。我们发现,有效的正样本应同时满足:(1) 在当前策略下具有较高可能性;(2) 能提升模型预测正确答案的概率。基于此,提出自解释策略优化(ExPO)——通过条件于真实答案生成此类样本,可无缝集成于GRPO、DPO等主流强化学习方法。实验表明,ExPO显著提升探索效率,生成的推理轨迹优于模型自身错误样本,且比专家编写的思维链更契合模型策略。在挑战性任务MATH Level-5上表现优异,显著超越基于专家示范的方法。代码已开源。

原文摘要 · Abstract (English)

Self-improvement via RL often fails on complex reasoning tasks because GRPO-style post-training methods rely on the model's initial ability to generate positive samples. Without guided exploration, these approaches merely reinforce what the model already knows (distribution-sharpening) rather than enabling the model to solve problems where it initially generates no correct solutions. To unlock reasoning ability in such settings, the model must explore new reasoning trajectories beyond its current output distribution. Such exploration requires access to sufficiently good positive samples to guide the learning. While expert demonstrations seem like a natural solution, we find that they are often ineffective in RL post-training. Instead, we identify two key properties of effective positive samples: they should (1) be likely under the current policy, and (2) increase the model's likelihood of predicting the correct answer. Based on these insights, we propose $\textbf{Self-Explanation Policy Optimization (ExPO)}$-a simple and modular framework that generates such samples by conditioning on the ground-truth answer. It can be integrated with popular RL training methods like GRPO and DPO. ExPO enables efficient exploration and guides the model to produce reasoning trajectories more aligned with its policy than expert-written CoTs, while ensuring higher quality than its own (incorrect) samples. Experiments show that ExPO improves both learning efficiency and final performance on reasoning benchmarks, surpassing expert-demonstration-based methods in challenging settings such as MATH level-5, where the model initially struggles the most. Code is available at https://github.com/HumainLab/ExPO_rl_reasoning_by_explanation .

强化学习推理增强自解释MATH数据集

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