Reward Models · Chapter 09
The generative reward model: the verdict as text
1. From score to judgment
A discriminative reward model maps an answer to a number. A generative reward model first says what it thinks of the answer, then gives the number.
Chapters 01 to 08 all share one design. A Transformer reads the prompt and response, and a small head on the last hidden state outputs one or more scores. Whatever checking the model does happens silently inside a single forward pass.
A generative reward model (GRM) drops the numeric head. It is a language model asked to evaluate: it reads the prompt and response, generates a critique \(c\), and ends with a verdict \(r\), such as "Yes", "No" or "8/10":
The critique is written in the same tokens the model uses for everything else. The model can therefore check arithmetic, restate the question or compare against a reference before it commits to a verdict.
2. Definition
A discriminative reward model is a function
A generative reward model is a language model \(p_\theta\) that defines a joint distribution over a critique \(c\) and a verdict \(r\), both token sequences, given a prompt and response:
The factorisation is just the order of generation: first the critique, then the verdict conditioned on it. The quantity we actually want is the verdict probability with the critique summed out:
The sum runs over every possible critique, so it cannot be computed exactly. The model may reach a verdict along many lines of reasoning; Equation 3 averages over them. Section 9 estimates it by sampling. The special case with an empty critique, \(c=\varnothing\), is a model that answers "Yes" or "No" immediately.
Here the focus is the output format: the judgment is produced as generated tokens, whether by a prompted model or one fine-tuned on critiques. Chapter 10 turns to reward models explicitly trained to reason at length before scoring, where the quality of that reasoning is itself optimised.
3. Discriminative vs generative
Discriminative RM
\(r_\theta(x,y)\) from a linear head on \(h_T\). One forward pass.
answer → score
Generative RM
\(p_\theta(c,r\mid x,y)\) from the language-model head. Many decoding steps.
answer → reason about answer → score
| Aspect | Discriminative | Generative |
|---|---|---|
| Output layer | New numeric head | Existing next-token head |
| Compute per judgment | One forward pass | Prefill plus decoding of the critique |
| Explanation | None | Readable critique |
| Extra compute at test time | Cannot help | Longer or more critiques |
| Reuse of pre-training | Through the backbone only | Backbone and output head, same task format |
| Training data | Labels or preferences | Labels, preferences or critiques |
The last two rows are the main argument for the generative form. Verification becomes another text-generation task, the kind of task the model was pre-trained on, and its accuracy can be raised after training by spending more tokens.
4. Checking by writing it out
Question: what is \(17\times 8\)? Candidate answer: \(17\times8=126\).
A discriminative RM must judge in one pass. It might output
A generative RM writes the check first:
and assigns \(P(\text{correct})=0.01\). The two numbers are similar. The difference is that the generative model's verdict rests on an intermediate result, 136, that it computed in the open.
The benefit grows with the difficulty of the check. Take the equation \(3(x+2)=15\), and a candidate solution whose first line is \(3x+2=15\). A classifier must spot the error from the surface of the text. A generative RM can redo the step:
Each line of the critique is an easy step that conditions the next. The verdict token is predicted after the evidence is on the page, not before.
5. Scores from token probabilities
RL and best-of-\(N\) need a number, not a word. The cleanest way to get one is to read it from the model's next-token distribution. Append a question \(I\) such as "Is the answer correct? Answer Yes or No." and take
where \(\ell\) are the logits at the verdict position. The right-hand form renormalises over the two allowed answers. If \(\ell_{\text{Yes}}=2.1\) and \(\ell_{\text{No}}=-0.4\), the score is \(\sigma(2.5)\approx0.924\). This is the approach of Generative Verifiers (GenRM): verification posed as next-token prediction, with the probability of "Yes" as the reward.
For a rating scale the same idea gives an expected score. If the model puts probability 0.2, 0.5 and 0.3 on the tokens "7", "8" and "9", then
Parsing the generated word gives a hard 0 or 1 (or a single integer rating). The token probability keeps the model's confidence: 0.924 and 0.51 are both "Yes" as text but very different as rewards. Continuous scores also break ties when ranking many candidates.
6. Training
Suppose the training set holds tuples \((x,y,c^\ast,r^\ast)\): a prompt, a response, a good critique and the correct verdict. Concatenate the target \(z=[c^\ast,r^\ast]\) and train with the ordinary language-model loss:
Unlike a scalar head, the model learns to generate evaluation text. With no critiques available, the same loss on a one-token target trains a direct verifier:
This is binary cross-entropy in disguise, but computed through the existing next-token head instead of a new one.
Where do good critiques \(c^\ast\) come from? Three common sources:
- Human-written critiques. Highest quality, most expensive.
- A stronger model's critiques, distilled into a smaller judge.
- Self-generated and filtered. Sample several critiques from the model itself and keep only those whose verdict matches the known label \(r^\ast\). This is rejection sampling: it needs only verdict labels, which are cheap when answers can be checked.
Because Equations 7 and 8 are the same loss as ordinary fine-tuning, judge data can be mixed with other instruction data in one training run.
7. LLM-as-a-judge
The simplest generative RM needs no reward-model training at all. Take a strong instruction-tuned model and prompt it to evaluate. This is LLM-as-a-judge, widely used both for evaluation benchmarks and as a source of AI feedback for training. Two prompt formats dominate.
Single-answer grading
"Rate this response from 1 to 10." One response per call; the score is absolute.
pointwise, like Chapter 02
Pairwise comparison
"Which response, A or B, is better?" Two responses per call; the verdict is relative.
pairwise, like Chapter 03
Pairwise judgments are usually more consistent, because comparing is easier than calibrating an absolute scale. Single-answer grades are cheaper for \(N\) candidates (\(N\) calls instead of up to \(\binom N2\)) and can be compared across prompts. The pairwise verdict probability plays the role of the Bradley–Terry probability of Chapter 03:
A third variant, reference-guided grading, supplies a correct solution in the prompt. It helps most on maths and code, where judges are otherwise easily misled by a confident but wrong answer.
8. Judge biases
A prompted judge inherits the habits of the model behind it. Three biases are well documented:
- Position bias. In pairwise prompts, the judge favours whichever response appears first (or second), independent of content.
- Verbosity bias. Longer, more detailed-looking answers are preferred even when the extra length adds nothing.
- Self-preference. A judge tends to rate text written by itself, or by its own model family, more highly.
Position bias has a simple fix: ask both orders and average. Suppose the judge prefers A with probability 0.70 when A is shown first, and with probability 0.40 when A is shown second. Then
and half the gap, \(\tfrac12(0.70-0.40)=0.15\), estimates how much the first slot alone is worth. When the two orders disagree on the winner, as here, many pipelines record a tie rather than trust either call.
Verbosity bias is reduced by instructing the judge to ignore length, by comparing responses of similar length, or by regressing length out of the scores. Self-preference is reduced by using a judge from a different model family than the policy being trained or evaluated. None of these removes the bias completely.
9. Averaging over critiques
A single sampled critique can go wrong: an arithmetic slip in the check itself, or a line of reasoning that misses the real error. Equation 3 says the verdict we want is an average over critiques. Monte-Carlo estimation gives it directly. Sample \(K\) critiques independently and average the verdict probabilities:
\(\hat p_K\) is an unbiased estimate of \(p_\theta(r\mid x,y)\), and its variance falls as \(1/K\):
Both follow from the critiques being independent draws from \(p_\theta(c\mid x,y)\).
This is a form of inference-time compute: the same trained judge gives more reliable verdicts if we pay for more samples. The standard error shrinks as \(1/\sqrt K\), so going from 1 to 4 critiques halves it, and from 4 to 16 halves it again. A discriminative RM has no such dial; its one forward pass is all it can do.
Two caveats. First, averaging removes noise between critiques, not bias shared by all of them: if the judge systematically misses a kind of error, every critique misses it. Second, a common alternative is majority vote over the parsed verdicts. It is robust to an occasional overconfident critique but discards the confidence information of Section 5.
10. Pipeline and cost
The cost difference is large. Both models first read \((x,y)\), a parallel prefill. A discriminative RM stops there. A generative RM then decodes token by token. With critiques of length \(L_c\), it generates about
With \(L_c=300\) and \(K=8\) that is about 2,400 sequential decoding steps for one judgment. In RL, where every rollout of every batch must be scored, this can exceed the cost of generating the rollouts themselves. The trade-off is simple: better verification potential, much more inference cost.
11. Weaknesses
- Inference cost. Decoding critiques is orders of magnitude slower than one forward pass (Equation 12).
- Persuasive but wrong critiques. A fluent critique can contain its own errors, and readers, human or machine, tend to trust it.
- Unfaithful verdicts. The verdict token need not follow from the critique; the model may decide first and rationalise afterwards.
- Inherited biases. Position, verbosity and self-preference (Section 8) carry over from the underlying model.
- Prompt sensitivity. Small changes to the evaluation instruction or rating scale can shift scores noticeably.
- Exploitable through text. The policy writes the text the judge reads; under RL it can learn phrasings that sway the judge, including claims about its own correctness.
12. Laboratory: averaging sampled verdicts
A judge evaluates one answer. Its marginal verdict \(p(\text{correct}\mid x,y)\) is set by the first slider. Each sampled critique leads to its own verdict probability, scattered around that value with the spread set by the second slider. Choose \(K\), sample, and compare the individual verdicts with their average. The repeat button shows how the spread of the average shrinks as \(K\) grows.
Monte-Carlo verdicts
\(\hat p_K=\frac1K\sum_k p(r\mid x,y,c_k)\)Sampled verdicts
Try \(p=0.7\) and \(s=0.3\) with \(K=1\): single verdicts often fall below 0.5 and flip the decision the judge makes on average. Then repeat with \(K=16\). The spread of the average drops by a factor of four, and the share of averages on the wrong side of 0.5 falls sharply.
13. Worked example, step by step
A judge evaluates a solution that is in fact correct. Five critiques are sampled, and after each one the probability of "Yes" is read off. Combine them and compare with relying on a single critique.
- Verdict probabilities: \(p_1,\ldots,p_5=0.9,\ 0.3,\ 0.8,\ 0.8,\ 0.7\). Critique 2 made a slip in its own check.
- Average: \(\hat p_5=\tfrac15(0.9+0.3+0.8+0.8+0.7)=\tfrac{3.5}{5}=0.70\).
- Majority vote at threshold 0.5: four of five say "Yes", so the vote is \(4/5=0.8\) and the answer is accepted.
- Spread: deviations from 0.70 are \(0.2,-0.4,0.1,0.1,0\); squares sum to \(0.22\); sample variance \(0.22/4=0.055\); standard deviation \(\approx0.235\).
- Standard error of the average: \(0.235/\sqrt5\approx0.105\). The average is about two standard errors above 0.5.
- With \(K=1\), the outcome depends on which critique was drawn. Drawing critique 2 gives 0.3 and rejects a correct answer.
- Cost: with critiques of about 250 tokens, \(K=5\) means about \(5\times251=1{,}255\) generated tokens for one judgment.
14. What comes next
In this chapter the critique was a means to a verdict: prompted, or imitated from examples. The next chapter takes the reasoning seriously as the object of training. Critique-then-score reward models are trained to think at length before they score, so that more reasoning reliably buys better judgments.
Discriminative RM: answer → score. Generative RM: answer → reason about answer → score, with the score read from token probabilities.
See also
- Zheng et al., 2023, judging LLM-as-a-judge with MT-Bench and Chatbot Arena: judge formats and position, verbosity and self-enhancement biases.
- Zhang et al., 2024, Generative Verifiers: reward modelling as next-token prediction (GenRM).
- Wang et al., 2023, large language models are not fair evaluators: position bias in pairwise judging.
- Kim et al., 2023, Prometheus: an open model fine-tuned for fine-grained, rubric-based evaluation.
- Bai et al., 2022, Constitutional AI: AI-generated feedback in place of human labels.
- Lee et al., 2023, RLAIF: reinforcement learning from AI feedback compared with human feedback.
Previous
← Multi-Objective Reward Model
Next chapter
Critique-then-score reasoning RM
Reward models trained to reason at length before they score.