高精度奖励模型反而让语言模型表现更差,发现性能悖论。
The Accuracy Paradox in RLHF: When Better Reward Models Don't Yield Better Language Models
- 用中等准确度的奖励模型训练,语言模型效果更好
- 在问答任务中,中等精度模型比高精度模型得分高12.3%
- 适合研究对齐机制与奖励模型选择的学者参考
基于人类反馈的强化学习显著提升了自然语言处理中语言模型与人类期望的一致性。其中,训练阶段使用的奖励模型强度是关键因素。本研究通过在QA-FEEDBACK数据集上,使用Longformer为基础的奖励模型,在相关性、事实性和完整性任务中进行实验,揭示了一个出人意料的悖论:采用中等准确度奖励模型训练的语言模型,其表现优于由高度准确奖励模型指导的模型。这一发现挑战了‘奖励模型越强,模型越好’的普遍认知,为未来研究模型性能的关键驱动因素及如何选择最优奖励模型开辟了新方向。代码与更多细节见https://github.com/EIT-NLP/AccuracyParadox-RLHF。
原文摘要 · Abstract (English)
Reinforcement Learning from Human Feedback significantly enhances Natural Language Processing by aligning language models with human expectations. A critical factor in this alignment is the strength of reward models used during training. This study explores whether stronger reward models invariably lead to better language models. In this paper, through experiments on relevance, factuality, and completeness tasks using the QA-FEEDBACK dataset and reward models based on Longformer, we uncover a surprising paradox: language models trained with moderately accurate reward models outperform those guided by highly accurate ones. This challenges the widely held belief that stronger reward models always lead to better language models, and opens up new avenues for future research into the key factors driving model performance and how to choose the most suitable reward models. Code and additional details are available at https://github.com/EIT-NLP/AccuracyParadox-RLHF.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。