torchtune.rlhf¶
RLHF 算法(如 PPO 和 DPO)的组件和损耗。
使用 Generalized Advantage Estimation 估计 PPO 算法的优势和回报 https://arxiv.org/pdf/1506.02438.pdf |
|
计算给定分数、对数概率和参考对数概率的 PPO 奖励。 |
|
在第一个 stop 标记之后截断序列,并使用 . |
|
近端策略优化 (PPO) 损失模块。 |
|
直接偏好优化 (DPO) 损失模块:https://arxiv.org/abs/2305.18290 从论文中简单地说: |
|
统计抑制采样优化 (RSO) 或“铰链”损失模块:https://arxiv.org/abs/2309.06657。 |
|
具有无参考奖励的简单偏好优化:https://arxiv.org/abs/2405.14734。 |