The Accuracy Paradox in RLHF: When Better Reward Models Don't Yield Better Language Models

Yanjun Chen, Dawei Zhu, Yirong Sun, Xinghao Chen, Wei Zhang +1 more
10/9/2024
cs.CLcs.AI

Official code: EIT-NLP/AccuracyParadox-RLHF (13 stars, MIT, last commit 2024-11-11). 1 more implementation below.

Abstract

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.

View on arXivView PDF

Code Implementations(2)

MIT

[EMNLP 2024 Main] Official implementation of the paper "The Accuracy Paradox in RLHF: When Better Reward Models Don't Yield Better Language Models". (by Yanjun Chen)

130Shell, PythonOct 8, 20241 years agoMIT
MIT

[EMNLP 2024 Main] Official implementation of the paper "The Accuracy Paradox in RLHF: When Better Reward Models Don't Yield Better Language Models".

80Shell, PythonOct 2, 20241 years agoMIT

Cite this paper

@article{chen2024accuracy,
  title  = {The Accuracy Paradox in RLHF: When Better Reward Models Don't Yield Better Language Models},
  author = {Yanjun Chen and Dawei Zhu and Yirong Sun and Xinghao Chen and Wei Zhang and Xiaoyu Shen},
  year   = {2024},
  eprint = {2410.06554},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.06554v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.06554.svg)](https://codeofpaper.com/paper/2410.06554)

Discussion