Skip to content

AI Glossary · Concepts

What is Reward Model?

Reward Model

A reward model is a model trained to give a score to another model's output, usually by learning from human or AI preferences between pairs of answers, so that reinforcement learning can push the main model toward higher-scoring behavior.

Also known as: reward models, preference model

· Chain of Thought

Model TrainingAI Evaluation & Reliability

  • 2 episodes
  • 2 guest clips
  • 3 sources

People cannot rate millions of chatbot answers by hand, but they can rate a few thousand. A reward model stretches that effort. Labelers see two answers to the same question and pick the better one, a form of pairwise evaluation. A separate model is trained on those choices until it can predict which answer a person would prefer, and it outputs a single number for any new answer. During reinforcement learning, the main model generates answers, the reward model scores them, and policy optimization nudges the main model toward answers that score higher.

That is the core of RLHF, as used for InstructGPT and many later chat models. Some methods skip the separate model: DPO learns from the preference pairs directly, and reinforcement learning with verifiable rewards replaces the learned score with a check such as running unit tests or matching a math answer.

A reward model is only an approximation of what people want, and the model being trained will exploit any gap. If the reward model likes long, confident answers, training produces long, confident answers whether or not they are right. That failure is reward hacking, and it is why researchers evaluate reward models themselves with benchmarks such as RewardBench.

In episode 1 at 23:25, Galileo co-founder Atindriyo Sanyal describes the training stage “including RLHF, where you use human feedback as reward signal” and says much of the gain from GPT-3 to GPT-3.5 came from it. In episode 5 at 23:00, NVIDIA’s Vivienne Zhang points listeners to RewardBench as an open leaderboard. Galileo sponsored season one, and episode 1’s panel participants are Galileo’s co-founders.

Hear it from the guest

“Including RLHF, where you use human feedback as reward signal”
“RewardBench”

Quotes lightly edited to remove filler words.

Sources

From the conversation