arXiv:2510.01857cs.AI2025-10被引 1

从专家思维链中学习推理奖励,提升大模型的推理能力与鲁棒性。

Learning Reasoning Rewards from Expert Demonstrations with Inverse Reinforcement Learning

  • 通过逆强化学习从专家思维链中推断过程级奖励函数。
  • 在GSM8K等数据集上,推理准确率最高提升17.4点,失败定位准确率达86.1%。
  • 适合需要高可靠性推理的大模型后训练与评估场景。

教导大型语言模型在后训练阶段进行推理,通常依赖于基于结果或过程的显式奖励函数进行强化学习。然而,在许多真实场景中,获取或定义此类奖励函数十分困难,尤其对复杂任务而言,使得从专家示范中学习成为更具吸引力的替代方案。主流方法监督微调(SFT)直接模仿专家的推理轨迹,但存在离线策略学习的固有局限:性能对推理时偏离示范状态的情况敏感。为此,我们提出推理对抗逆强化学习(R-AIRL)。该方法不直接模仿专家推理,而是从专家的思维链中推断出底层的过程级奖励函数。在GSM8K、MMLU-Pro和MedReason上的实验表明,使用R-AIRL学习到的推理奖励函数可有效贯穿训练与推理全流程:(1) 作为后训练的信号,优于多数情况下的SFT;(2) 用于推理时重排序,使pass@1最高提升17.4分;(3) 用于过程级评估,推理失败定位准确率最高达86.1%。总体而言,R-AIRL连接了模仿学习与基于奖励的优化,实现了从专家思维痕迹中提取有意义的推理信号。

原文摘要 · Abstract (English)

Teaching large language models (LLMs) to reason during post-training typically relies on reinforcement learning with explicit outcome- or process-based reward functions. However, in many real-world settings, obtaining or defining such reward functions is difficult, especially for complex tasks, making learning from expert demonstrations an attractive alternative. The dominant approach, supervised fine-tuning (SFT), trains models to imitate expert reasoning traces directly, but suffers from the general limitations of off-policy learning: performance can be fragile to inference-time deviations from states explicitly covered by the demonstrations. To address this, we propose Reasoning Adversarial Inverse Reinforcement Learning (R-AIRL). Rather than imitating the expert's reasoning, R-AIRL infers the underlying process-level reward from the expert Chain-of-Thoughts. Through experiments on GSM8K, MMLU-Pro and MedReason we show that the reasoning reward function learned with R-AIRL can be effectively used throughout the training and inference pipeline: (1) to provide a training signal for post-training, outperforming SFT in most of the considered settings, (2) for inference-time reranking, improving pass@1 by up to 17.4 points, and (3) for process-level evaluation, localising reasoning failures with up to 86.1% accuracy. Overall, R-AIRL bridges imitation learning and reward-based optimisation, enabling the extraction of meaningful reasoning signals from expert thinking traces.

推理增强逆强化学习专家示范大模型训练

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