Self-Generated Critiques Boost Reward Modeling for Language Models

Yue Yu, Zhengxing Chen, Aston Zhang, Liang Tan, Chenguang Zhu +8 more
11/25/2024
cs.CLcs.AIcs.LG

No confident code implementation found for this paper.

Abstract

Reward modeling is crucial for aligning large language models (LLMs) with human preferences, especially in reinforcement learning from human feedback (RLHF). However, current reward models mainly produce scalar scores and struggle to incorporate critiques in a natural language format. We hypothesize that predicting both critiques and the scalar reward would improve reward modeling ability. Motivated by this, we propose Critic-RM, a framework that improves reward models using self-generated critiques without extra supervision. Critic-RM employs a two-stage process: generating and filtering high-quality critiques, followed by joint fine-tuning on reward prediction and critique generation. Experiments across benchmarks show that Critic-RM improves reward modeling accuracy by 3.7%-7.3% compared to standard reward models and LLM judges, demonstrating strong performance and data efficiency. Additional studies further validate the effectiveness of generated critiques in rectifying flawed reasoning steps with 2.5%-3.2% gains in improving reasoning accuracy.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{yu2024selfgenerated,
  title  = {Self-Generated Critiques Boost Reward Modeling for Language Models},
  author = {Yue Yu and Zhengxing Chen and Aston Zhang and Liang Tan and Chenguang Zhu and Richard Yuanzhe Pang and Yundi Qian and Xuewei Wang and Suchin Gururangan and Chao Zhang and Melanie Kambadur and Dhruv Mahajan and Rui Hou},
  year   = {2024},
  eprint = {2411.16646},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.16646v3}
}

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/2411.16646.svg)](https://codeofpaper.com/paper/2411.16646)

Discussion