数学推理的自奖励校正Self-rewarding correction for mathematical reasoning
我们研究了一种自奖励推理的大语言模型(LLMs),该模型能够在推理过程中同时生成逐步推理步骤并评估其输出的正确性,而无需外部反馈。这种集成方法使得单个模型能够独立引导其推理过程,为模型部署提供了计算优势。我们特别关注自我校正这一代表性任务,其中模型能够自主检测其响应中的错误、修订输出,并决定何时终止迭代优化循环。为此,我们提出了一个两阶段的算法框架,仅利用自生成数据构建自奖励推理模型。在第一阶段,我们采用顺序拒绝采样法合成包含自奖励和自我校正机制的长链思维轨迹。通过对这些精选数据进行微调,模型能够学习自奖励和自我校正的模式。在第二阶段,我们通过基于规则的信号进行强化学习,进一步增强模型评估响应准确性和优化输出的能力。在Llama-3和Qwen-2.5上的实验表明,我们的方法超越了内在的自我校正能力,并达到了与依赖外部奖励模型的系统相当的性能。