CLOct 13, 2024

RMB: Comprehensively Benchmarking Reward Models in LLM Alignment

arXiv:2410.09893v257 citationsh-index: 40Has CodeICLR
AI Analysis

This work addresses the challenge of effectively benchmarking RMs to improve LLM alignment, though it is incremental as it builds on existing evaluation methods by expanding data and metrics.

The authors tackled the problem of evaluating reward models (RMs) for aligning large language models (LLMs) by proposing RMB, a comprehensive benchmark covering over 49 real-world scenarios, which shows a positive correlation with downstream alignment performance and reveals generalization defects in state-of-the-art RMs.

Reward models (RMs) guide the alignment of large language models (LLMs), steering them toward behaviors preferred by humans. Evaluating RMs is the key to better aligning LLMs. However, the current evaluation of RMs may not directly correspond to their alignment performance due to the limited distribution of evaluation data and evaluation methods that are not closely related to alignment objectives. To address these limitations, we propose RMB, a comprehensive RM benchmark that covers over 49 real-world scenarios and includes both pairwise and Best-of-N (BoN) evaluations to better reflect the effectiveness of RMs in guiding alignment optimization. We demonstrate a positive correlation between our benchmark and the downstream alignment task performance. Based on our benchmark, we conduct extensive analysis on the state-of-the-art RMs, revealing their generalization defects that were not discovered by previous benchmarks, and highlighting the potential of generative RMs. Furthermore, we delve into open questions in reward models, specifically examining the effectiveness of majority voting for the evaluation of reward models and analyzing the impact factors of generative RMs, including the influence of evaluation criteria and instructing methods. Our evaluation code and datasets are available at https://github.com/Zhou-Zoey/RMB-Reward-Model-Benchmark.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes