通过强化学习训练语言模型进行自我纠正Training Language Models to Self-Correct via Reinforcement Learning
自我校正是大型语言模型(LLMs)极为理想的能力,然而在现代LLMs中,它一直被发现效果不佳。现有的自我校正训练方法要么需要多个模型,要么依赖于更强大的模型或其他形式的监督。为此,我们开发了一种多轮在线强化学习(RL)方法SCoRe,通过完全自动生成的数据显著提高LLM的自我校正能力。为构建SCoRe,我们首先表明,在离线模型生成的校正轨迹的监督微调(SFT)的变体对于灌输自我校正行为是不够的。特别地,我们观察到通过SFT训练要么受到训练数据与模型自身响应之间的分布不匹配的困扰,要么隐式偏好于某种在测试时通常不有效的校正行为模式。SCoRe通过在模型自身生成的校正轨迹分布下训练,并使用适当的正则化来引导学习过程,使其学习一种在测试时有效的自我校正策略,而不是仅仅拟合给定提示的高回报响应。这种正则化规定在基础模型上运行第一阶段的RL以生成一个较不容易崩溃的策略初始化,然后使用奖励奖励来增强训练期间的自我校正。当应用于Gemini 1.0 Pro和1.5 Flash模型时,我们发现SCoRe在MATH和HumanEval基准测试中分别将基础模型的自我校正性能提高了15.6%和9.1%,达到了最先进的水平。