2024

Secrets of RLHF in Large Language Models Part II: Reward Modeling

Wang, Binghai, Zheng, Rui, Chen, Lu et al.

Understand

Reinforcement Learning from Human Feedback (RLHF) has become a crucial technology for aligning language models with human values and intentions, enabling models to produce more helpful and harmless responses.

  • Reward models are trained as proxies for human preferences to drive reinforcement learning optimization.
  • While reward models are often considered central to achieving high performance, they face the following challenges in practical applications: (1) Incorrect and ambiguous preference pairs in the dataset may hinder the reward model from accurately capturing human intent.
  • (2) Reward models trained on data from a specific distribution often struggle to generalize to examples outside that distribution and are not suitable for iterative RLHF training.

Reading the bibliography…