PPO系列4 - Reward模型训练

流程:

训练Reward模型

训练数据:

相比给每条回答进行打分,人类更容易给出两者的比较结果。这样标注出来的数据,准确性更高。

模型:

可以使用和生成模型能力差不多的模型(或者更强的模型),作为Reward模型。

不能使用太差的模型,因为Reward模型是"老师"的角色,差老师教不出好学生。

但可以使用和生成模型能力差不多的模型。因为:Reward打分任务,比生成任务,难度小很多,更容易做对。例如:我来评判一篇文章《赤兔之死》是否写得好,比自己写出这种水平的文章,要容易很多。

是否能用强化学习无限提升模型能力?答:不行的。大模型的能力主要是在预训练阶段决定下来了,SFT和强化学习,都只是让大模型能力尽量发挥出来。

生成式模型:

如上,生成是N个token的概率。

Reward模型:

如上,输入是(question+chosen)或(question+rejected),生成是1个得分。

因此,Reward模型,只需要把生成式模型的最后一层(LM Head),替换成生成1个得分的层(Score Head),即可。前面的层可复用。

损失函数:

让chosen和分数 - rejected的分数,越大越好。

Reward模型推理2次,1次计算chosen的得分,1次计算rejected的得分。

代码:可用Huggingface的trl库里的RewardTrainer来做训练。

相关推荐
I Am a robert girl18 小时前
从科学代码库到智能体训练场:ScienceIDE 的架构拆解与工程启示
架构·强化学习·代码仓库·科学计算·监督微调·智能体训练·可编程环境
Better Bench2 天前
从摸石头过河到看录像学习:强化学习与离线RL的进化故事
学习·强化学习·贝尔曼方程·离线强化学习·在线强化学习·q值
论文复现现场2 天前
Qwen3.8-27B 做 GRPO 需要几张 4090?Coding Agent 显存预算与多卡验收
强化学习·qwen·大模型微调·rtx4090·算家云·grpo
微小冷3 天前
gymnasium初步教程
强化学习·gym·仿真环境·gymnasium·动力学仿真
Web3&Basketball4 天前
用 SGLang 决策端点做毫秒级 Agent 路由
python·大模型·agent·强化学习·多模态·推理
AI你一生一世5 天前
一年前,我用 RL 造了一批非自回归决策模型
强化学习·决策系统·低延迟推理·非自回归模型·实时竞价
盘古开天16665 天前
PPO算法原理详解(上):从策略梯度到近端策略优化的演进之路
人工智能·算法·机器学习·强化学习·ppo
盼小辉丶6 天前
PyTorch强化学习实战(27)——进化策略在强化学习中的应用
人工智能·pytorch·深度学习·强化学习
火星机器人life7 天前
PPO 会探索、CBF 会保命:RADAR-BPO 把复杂环境碰撞率压到 10.7%
强化学习·机器人导航·控制屏障函数·安全滤波
维度攻城狮8 天前
从Q-learning到DQN:小车倒立摆的智能控制进阶
强化学习·dqn·drl