Reward Models · Chapter 09

The generative reward model: the verdict as text

Abstract. Every reward model so far ends in a numeric head. A generative reward model keeps the language-model head instead: it writes an evaluation of the response and then a verdict, both as ordinary tokens. This chapter defines the model as a joint distribution over critiques and verdicts, shows how to read a score from the probability of a "Yes" token, trains it with the usual next-token loss, covers prompted LLM judges and their biases, and treats averaging over several sampled critiques as a way to spend inference-time compute on better judgments.

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":

(1) \[ (x,y)\ \longrightarrow\ \text{critique } c\ \longrightarrow\ \text{verdict } r . \]

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

(2) \[ r_\theta(x,y)\in\mathbb R . \]
Definition 9.1 · Generative reward model

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:

\[ p_\theta(c,r\mid x,y)=p_\theta(c\mid x,y)\,p_\theta(r\mid x,y,c). \]

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:

(3) \[ p_\theta(r\mid x,y)=\sum_{c}p_\theta(c\mid x,y)\,p_\theta(r\mid x,y,c). \]

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.

Remark · Scope of this chapter

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

AspectDiscriminativeGenerative
Output layerNew numeric headExisting next-token head
Compute per judgmentOne forward passPrefill plus decoding of the critique
ExplanationNoneReadable critique
Extra compute at test timeCannot helpLonger or more critiques
Reuse of pre-trainingThrough the backbone onlyBackbone and output head, same task format
Training dataLabels or preferencesLabels, 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

(4) \[ r_\theta(x,y)=0.03 . \]

A generative RM writes the check first:

Critique: 17 × 8 = 10 × 8 + 7 × 8 = 80 + 56 = 136. The answer says 126, which is wrong. Verdict: No

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:

Distribute the 3: \(3(x+2)=3x+6\).
↓
The candidate wrote \(3x+2\): the 2 was not multiplied.
↓
Correct path: \(3x+6=15\Rightarrow x=3\). The candidate's path leads to \(x=13/3\).
↓
Verdict: No, \(r\approx0\).

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

(5) \[ r(x,y)=p_\theta(\text{Yes}\mid x,y,I,c) \approx\frac{e^{\ell_{\text{Yes}}}}{e^{\ell_{\text{Yes}}}+e^{\ell_{\text{No}}}} =\sigma\big(\ell_{\text{Yes}}-\ell_{\text{No}}\big), \]

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

(6) \[ \bar s=\sum_{s}s\,p_\theta(s\mid x,y,c)=7(0.2)+8(0.5)+9(0.3)=8.1 . \]
Note · Probabilities beat parsed text

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:

(7) \[ \mathcal L_{\text{SFT}}=-\sum_{t=1}^{|z|}\log p_\theta\big(z_t\mid z_{<t},x,y\big). \]

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:

(8) \[ \mathcal L_{\text{verdict}}=-\log p_\theta\big(r^\ast\mid x,y,I\big),\qquad r^\ast\in\{\text{Yes},\text{No}\}. \]

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:

(9) \[ P(A\succ B\mid x)\approx p_\theta(\text{"A"}\mid x,\,y_A,\,y_B,\,c). \]

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

(10) \[ P(A\succ B)=\tfrac12\big(P_{AB}(A)+P_{BA}(A)\big)=\tfrac12(0.70+0.40)=0.55 , \]

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.

Note · Mitigations for the other two

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:

(11) \[ \hat p_K=\frac1K\sum_{k=1}^{K}p_\theta\big(r\mid x,y,c_k\big), \qquad c_k\sim p_\theta(c\mid x,y). \]
Proposition 9.1 · More critiques, less noise

\(\hat p_K\) is an unbiased estimate of \(p_\theta(r\mid x,y)\), and its variance falls as \(1/K\):

\[ \mathbb E\big[\hat p_K\big]=p_\theta(r\mid x,y),\qquad \operatorname{Var}\big(\hat p_K\big)=\frac{\operatorname{Var}_{c}\big[p_\theta(r\mid x,y,c)\big]}{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

1 · Give the prompt and candidate answer to the reward model, with an evaluation instruction.
↓
2 · Sample \(K\) critiques that check the answer.
↓
3 · After each critique, read the verdict probability \(p_\theta(\text{Yes}\mid x,y,c_k)\).
↓
4 · Average the \(K\) probabilities (Equation 11).
↓
5 · Use the averaged score for ranking, search or RL.

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

(12) \[ \text{generated tokens per judgment}\approx K\,(L_c+1). \]

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)\)
Marginal verdict
Spread of one critique's verdict
Critiques sampled
Sample
Average \(\hat p_K\)-
Standard error-
Majority vote-

Sampled verdicts

Press "Sample K critiques".

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.

  1. 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.
  2. Average: \(\hat p_5=\tfrac15(0.9+0.3+0.8+0.8+0.7)=\tfrac{3.5}{5}=0.70\).
  3. Majority vote at threshold 0.5: four of five say "Yes", so the vote is \(4/5=0.8\) and the answer is accepted.
  4. 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\).
  5. Standard error of the average: \(0.235/\sqrt5\approx0.105\). The average is about two standard errors above 0.5.
  6. With \(K=1\), the outcome depends on which critique was drawn. Drawing critique 2 gives 0.3 and rejects a correct answer.
  7. 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.

Key idea to remember

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.

Continue →