Reward和Policy
2026/7/23大约 3 分钟
“使用 PPO 强化学习算法,基于奖励模型来优化策略(Policy)。”
这是大语言模型经典对齐演进路线中,RLHF(基于人类反馈的强化学习)的最核心步骤(Step 3)。紧接在前面聊过的 SFT(监督微调)之后,它是为了让大模型真正拥有符合人类价值观、安全且高情商的回答能力。
为了让你彻底看懂这句话背后的物理机制,我们对里面的硬核术语进行逐一拆解:
1. 什么是 Reward Model(奖励模型)?
你可以把它理解为一个“AI 裁判”。
- 在 Step 2 中,人类专家会给大模型的多个不同回答进行打分和排序(哪个回答更安全、更有礼貌、更准确)。
- 接着,用这些排序数据训练出一个独立的模型,也就是 奖励模型(Reward Model)。它的唯一功能就是看大模型的回答,然后打分(回答得好就给正分,胡说八道或有偏见就给负分)。
2. 什么是 Policy(策略)?
这里的 Policy 物理上就是 正在接受强化学习训练的大模型本身(比如此时正在训练的 Llama-Chat)。
- 在强化学习中,“策略”决定了模型在面对一个 Prompt(提示词输入)时,应该如何“做出动作”(即预测下一个吐出什么 Token)。
3. 什么是 PPO 算法?
PPO(Proximal Policy Optimization,近端策略优化) 是一种非常经典且鲁棒的深度强化学习算法。
- 它的核心作用是:在利用“AI 裁判”(奖励模型)的打分结果来更新大模型(Policy)的参数时,施加一个“紧箍咒”(裁剪机制/KL散度约束)。
- 为什么要加紧箍咒?因为大模型非常聪明,如果没有约束,它为了在裁判那里刷出极高的分(骗取奖励),可能会学会走捷径(吐出一堆无意义但裁判模型喜欢的投机取巧的词,导致模型直接训练崩掉 / Policy Collapse)。PPO 能够确保模型在稳健、小步快跑的状态下完成参数更新。
🔄 这一步的动态物理工作流
在这句话描述的训练循环中,算力集群里正在发生这样的事情:
┌────────────────────────┐
│ 用户输入一个 Prompt │
└───────────┬────────────┘
│
▼
┌────────────────────────┐
│ 大模型 (Policy) 生成回答│
└───────────┬────────────┘
│
▼
┌────────────────────────┐
│ 裁判模型打分 (Reward) │
└───────────┬────────────┘
│
▼
┌────────────────────────────────────────────────────────┐
│ PPO 算法计算 Loss -> 反向传播 -> 微调更新大模型参数 │
│ (目标:让大模型未来吐出更多能拿高分的 Token,并限制走偏)│
└────────────────────────────────────────────────────────┘
💡 SRE / Infra 视角的性能痛点:
当你作为 Infra 收到算法团队提交的“PPO 强化学习训练”任务时,这往往是集群运维中最头疼的吞吐恶梦:
- 多模型同时吃显存:在 PPO 阶段,显存里需要同时常驻 4 个模型(正在训练的 Policy 模型、提供打分的 Reward 模型、用来做对比参照的 Reference 模型、以及计算价值的 Value 模型)。这直接导致单卡显存(HBM)瞬间见底,极易触发我们之前聊过的
XID 31 / OOM崩溃。 - Infra 解法:此时必须在启动命令中联合拉起 DeepSpeed ZeRO-Stage 3 或者是采用大厂通用的 Ray 框架,将这 4 个不同的模型打散,物理部署到不同的 GPU 卡和节点上去,依靠高速 RDMA 网络(配合
NCCL_CROSS_NIC=1)进行超高频的跨模型张量数据对齐。
