简介:本文将系统解析大模型对齐中强化学习的概念定义、核心原理与适用场景。读者将理解监督微调与强化学习的本质区别,掌握基于人类反馈的强化学习工作流程,以及奖励模型、策略优化等关键环节的协作方式,从而为技术选型与模型训练策略提供清晰的理论依据。
在大语言模型(Large Language Model,LLM)的训练与优化体系中,强化学习(Reinforcement Learning,RL) 是指一种通过环境反馈信号来优化模型输出策略的训练范式。具体到大语言模型的对齐阶段,它通常以基于人类反馈的强化学习(Reinforcement Learning from Human Feedback,RLHF) 的形式出现。
从技术视角来看,强化学习解决的核心问题是:在无法穷举所有正确答案的场景下,如何让模型学会生成更符合人类偏好的输出。 与传统的监督微调(Supervised Fine-Tuning,SFT)不同,强化学习不需要为每一个输入提供标准答案,而是通过一个奖励信号来评价模型自身输出的好坏,进而引导模型调整其生成策略。
从使用视角来看,这意味着训练者不再需要逐条撰写“标准回复”,而只需对模型生成的多个候选答案进行优劣评判或打分。模型会根据这些反馈信号,逐步提升生成高质量回答的概率,降低生成低质量回答的概率。
从业务视角来看,强化学习在大模型对齐中的价值在于:它能够处理那些难以用固定标准答案描述、但人类可以轻松判断好坏的任务,例如对话的自然度、安全性、有用性、无害性等主观质量维度。
监督微调是让模型学习“给定输入,输出对应答案”的过程。它要求训练数据中包含明确的输入-输出配对,模型通过最小化预测输出与标准答案之间的差异来更新参数。
这种方式在以下场景中表现良好:
然而,当面对开放式生成任务时,监督微调面临显著瓶颈:
强化学习通过引入奖励信号,将训练目标从“匹配标准答案”转变为“最大化预期奖励”。这一转变带来了几个关键优势:
正如原始材料所指出:监督微调比较辛苦,强化学习比较省力。 这里的“辛苦”体现在需要逐条编写标准答案,“省力”则体现在只需对模型输出进行好坏评价即可。
基于人类反馈的强化学习通常包含以下关键组件:
强化学习通常在一个已经经过预训练和监督微调的基础模型上进行。预训练模型提供语言能力,监督微调模型提供基本的指令跟随能力。这个模型作为强化学习阶段的初始策略(Policy)。
奖励模型(Reward Model,RM)是RLHF的核心组件之一。它的作用是:接收一个输入和一个模型输出,输出一个标量分数,表示该输出有多好。
奖励模型的训练通常分为两步:
奖励模型本质上是一个学习了人类偏好的代理,它可以在后续训练中替代人类提供快速、低成本的反馈。
策略模型即正在被优化的语言模型本身。在强化学习过程中,策略模型根据输入生成输出,奖励模型对该输出打分,策略模型根据分数更新参数,以提高获得高分的概率。
为了防止策略模型在优化过程中偏离过远、产生不合理的输出,RLHF通常还会引入一个参考模型(Reference Model)。参考模型通常是监督微调后的模型副本,其参数在训练中保持不变。通过计算策略模型与参考模型输出分布之间的差异(如KL散度),可以约束策略模型的更新幅度,保持输出的稳定性和多样性。
RLHF的典型流程可以分为三个阶段:
使用人工编写的标准答案对预训练模型进行微调,使其具备基本的指令跟随能力。这一阶段产出的模型称为SFT模型。
奖励模型的损失函数通常基于偏好排序,例如对于两个输出 ( y_w )(更受偏好)和 ( y_l )(较少受偏好),奖励模型需要满足 ( R(y_w) > R(y_l) )。
可以用以下伪代码示意:
for each training step:input = sample_from_dataset()output = policy_model.generate(input)reward = reward_model.score(input, output)kl_penalty = compute_kl(policy_model, reference_model, input)total_reward = reward - beta * kl_penaltypolicy_model.update(total_reward)
其中 beta 是控制KL惩罚强度的超参数。
强化学习在大模型对齐中的典型应用场景包括:
需要注意的是,强化学习并非适用于所有场景。对于有明确标准答案的任务,监督微调通常更直接、更高效。强化学习的优势主要体现在主观质量维度和开放式生成任务中。
| 维度 | 监督微调 | 强化学习(RLHF) | 提示工程 |
|---|---|---|---|
| 训练信号 | 标准答案 | 奖励分数 | 无训练 |
| 标注成本 | 高 | 中 | 无 |
| 适用任务 | 有标准答案 | 主观质量优化 | 快速适配 |
| 模型更新 | 参数更新 | 参数更新 | 无参数更新 |
| 能力上限 | 受限于标注 | 可超越标注 | 受限于基座 |
强化学习与监督微调不是替代关系,而是互补关系。在实践中,通常先进行监督微调,再通过强化学习进一步优化模型的对齐质量。
在应用强化学习进行大模型对齐时,需要关注以下问题:
强化学习在大模型对齐中扮演着关键角色。它通过奖励信号而非标准答案来引导模型优化,使模型能够在开放式生成任务中学习人类偏好,提升输出的有用性、安全性和自然度。其核心流程包括监督微调、奖励模型训练和策略优化三个阶段,关键组件包括策略模型、奖励模型和参考模型。
与监督微调相比,强化学习降低了对标准答案的依赖,更适合处理主观质量维度的优化。但它也引入了奖励模型偏差、训练稳定性等新的挑战。在实际应用中,需要根据任务特点、数据条件和计算资源,合理选择监督微调与强化学习的组合策略。