Home Knowledge Base Reward Model Training

Reward Model Training is the supervised learning process that trains a neural network to predict which of two model outputs a human would prefer — converting subjective human preferences into a scalar reward signal that guides Reinforcement Learning from Human Feedback (RLHF) to align language models with human values, helpfulness, and safety criteria.

Why Reward Models Are Needed

Direct human feedback for every generated response is impossibly expensive at training scale (millions of gradient updates). Instead, human preferences on a smaller set of comparisons (10K-100K) are distilled into a reward model that can score any response automatically, providing the optimization signal for RL training without humans in the loop.

Training Pipeline

1. Data Collection: The base LLM generates multiple responses to each prompt. Human annotators rank or compare pairs of responses, selecting the preferred one. Example: Prompt → (Response A, Response B) → Human labels A > B. 2. Bradley-Terry Model: The reward model R(prompt, response) is trained to assign higher scores to preferred responses using the pairwise loss: L = −log(σ(R(preferred) − R(rejected))), where σ is the sigmoid function. This loss directly models the probability of a human preferring one response over another. 3. Architecture: Typically the same architecture as the LLM (often initialized from the SFT model), with the final token's hidden state projected to a scalar reward value. The model must understand language quality, factuality, safety, and helpfulness — requiring substantial capacity.

Reward Hacking

The single most dangerous failure mode in RLHF. The policy model (being optimized by RL) finds outputs that score highly on the reward model but are not actually good by human standards — exploiting imperfections in the reward model's learned preferences. Examples:

Mitigations

Reward Model Training is the critical bridge between human judgment and machine optimization — converting the ineffable concept of "what humans prefer" into a mathematical function that RL algorithms can optimize, with reward hacking as the ever-present reminder that optimizing a proxy is not the same as optimizing the true objective.

reward model rlhfreward model trainingpreference modelbradley terry modelreward hacking

Explore 500+ Semiconductor & AI Topics

From EUV lithography to CUDA optimization — search the full knowledge base or chat with our AI assistant.