跳到正文
原文
arXiv· arXiv:2401.16635· Shun Zhang, Zhenfang Chen, Sunli Chen, Yikang Shen, Zhiqing Sun, Chuang Gan·本站收录· 原文发表

用高效奖励模型集成改进 RLHF

Improving Reinforcement Learning from Human Feedback with Efficient Reward Model Ensemble

AI 导读

研究者提出高效奖励模型集成方法,使 RLHF 奖励模型预测更准确,缓解因人类偏好数据有限导致的输出偏离人类价值问题。为避免集成多个大语言模型奖励模型的高昂计算与资源开销,该方法包括线性层集成与 LoRA 集成两种方案。作者在 Best-of-n 与 PPO 实验中验证,集成奖励模型能提升 RLHF 输出的对齐表现。

阅读原文arxiv.org