大模型入门-Reward 奖励模型训练

2.4 Reward 奖励模型训练

在强化学习对齐(RLHF)的过程中,奖励模型(Reward Model, RM)的训练是至关重要的一环。它的好坏直接决定了大模型最终的训练效果 。

2.4.1 核心思想:相对排序优于绝对打分

首先,人类需要对大模型针对同一个问题生成的多个不同答案进行优先级排序

  • 为什么不直接打分?因为不同的人对绝对分数的评判标准很难统一,但判断"哪一个回答相对更好"则容易得多 。

  • 因此,业界普遍采用相对排序来替代绝对打分,并根据这个排序结果来训练奖励模型 。

2.4.2 模型结构

  • 奖励模型的底座通常直接采用第一阶段训练出的 SFT(有监督微调)模型

  • 结构上的唯一改变,是将最后一层修改为一个神经元 。

  • 这样模型就变成了一个回归模型 ,这个唯一的神经元输出的就是针对该回答的得分。在后续的强化学习阶段中,我们将用这个模型对大模型生成的每个回答进行打分评估 。

2.4.3 核心损失函数 (Loss Function)

奖励模型训练的核心在于它的损失函数设计:

loss=−log⁡(σ(rθ(x,yc)−rθ(x,yr)))loss=-\log(\sigma(r_{\theta}(x,y_{c})-r_{\theta}(x,y_{r})))loss=−log(σ(rθ(x,yc)−rθ(x,yr)))

  • 公式中的 ccc 代表 chosen,即排名较高的、被选中的回答 。rrr 代表 reject,即排名较低的、被拒绝的回答 。

  • 这个损失函数的最终目的,就是通过优化参数,使得排名靠前的回答 (ycy_cyc)的得分相应地变得比排名靠后的回答 (yry_ryr)更高 。

2.4.4 案例解析:如何进行两两组合博弈?

为了更好地理解,我们来看一个具体的例子:

问题 (Q): 苹果是什么?

A1: 苹果是一种红色水果,可以润肺、解暑、开胃。

A2: 苹果的品种繁多,根据颜色、大小、口感和用途等不同特点,可以分为多个品种。有的苹果品种适合鲜食,口感脆甜;有的适合烹饪,如做苹果派或苹果酱;还有的适合制酒或制醋。此外,美国苹果公司是全球知名的科技公司。

A3: 苹果是水果。

实际训练过程:

  1. 人工排序: 假设人工标记员给出的打分优先级为:A2 > A1 > A3 。

  2. 两两组合: 我们将这三个回答进行两两组合,形成比较对:(A2, A1)、(A2, A3)、(A1, A3) 。

  3. 模型打分与反向传播: 将这些组合对放入模型进行打分,随后根据上述的损失函数对模型进行反向传播更新参数 。

  4. 最终结果: 经过充分训练后,模型打分的结果将会完美符合人类的偏好,即输出的得分满足:A2的分数 > A1的分数 > A3的分数 。

通过这种不断输入"不同答案优先级顺序"的方式 ,我们就能训练出一个符合人类价值观偏好的奖励模型,以此来替代人工,自动化地对大模型的回答进行质量打分 。

python 复制代码
print('hello')
相关推荐
倒头就睡的小比特3 天前
算法竞赛C++常用的STL
c++·算法
默_笙3 天前
🍙 给每个请求过安检:FastAPI 是怎么把校验写进类型注解的
python
小羊没烦恼!3 天前
初探性能优化——2个月到4小时的性能提升
java·开发语言·windows·算法·c#
qq_426003963 天前
启动playwright录制codegen生成自动化测试脚本
python·自动化
虎头金猫3 天前
4K 视频总卡在公网带宽?用 N1 + OpenList 把网盘播放链路重新理顺
运维·服务器·网络·python·容器·beautifulsoup·pandas
长沙三为智能科技3 天前
家政小程序开发从0到上线:五阶段交付流程与验收清单
python
伞伞悦读3 天前
【第38期】Python 模块与包详解:import、from、模块搜索路径、包结构和 __init__
开发语言·python
猎头南楼3 天前
知识社区推荐系统实践:新用户冷启动与长短期兴趣建模的挑战 资深推荐算法工程师
人工智能·深度学习·算法·机器学习
只睡四小时3 天前
Canvas 弹道联机实战:700 行 + 固定时间步长
python·websocket·html5·游戏开发·canvas
C语言小火车3 天前
C/C++ 为什么需要编译器?
开发语言·c++