Concepts / Preference Data and Model Alignment

Preference Data and Model Alignment

A reward model takes a (prompt, response) pair and scores how good that single response is, such as Cappy's score between 0 and 1.

  • Programming

Why Specialized Judges Exist

When an AI model produces an answer, someone has to decide whether that answer is good. One option is to ask a general-purpose AI model to act as the judge. This is flexible, because the same model can be asked about many kinds of responses. It also has limits. A general model asked to judge might be slow or costly, and its verdicts can be hard to reproduce. Specialized AI judges address this by being models built or trained for one narrow evaluation job.

This article looks at three families of specialized judges: reward models, reference-based judges, and preference models. For each one, the useful questions are the same. What goes in? What comes out? And what does the output tell you about a response? Answering those three questions for each family is the fastest way to learn when each one is the right tool.

What a Reward Model Receives and Returns

Definition

A reward model takes a (prompt, response) pair and scores how good that single response is for that prompt.

The defining feature of a reward model is that it looks at one response on its own, in the context of the prompt. It does not need a second answer to compare against, and it does not need a model answer written in advance. The input is the prompt together with a single response. The output is a score that expresses how good that response is for that prompt.

Reward models have been used in reinforcement learning from human feedback (RLHF) for many years. This is where the scores matter most: they can guide how a model is tuned.

Promptthe question or instructionReward modelscores the pairSingle scorehow good this response isOne responsea single candidate answercontextscoredreturns
What flows into a reward model (a prompt plus one response), and the single score that comes out.

Cappy, developed by Google in 2023, is a concrete case. Given a prompt and a response, Cappy produces a score between 0 and 1 that indicates how correct the response is. It is a lightweight scorer with 360 million parameters, which is much smaller than general-purpose foundation models. Its small size is part of its appeal: a compact scorer can be run many times over cheaply when a large batch of responses needs to be checked.

Tracing One Reward Model Call

What do you think happens?

A reward model is given a prompt and one response to that prompt. What does it hand back?

  • A verdict on which of two responses is preferred

  • A single score for that one response

  • A scoring rubric for the task

  • A rewritten version of the response

Reveal answer
Answer:

A single score for that one response

The defining feature of a reward model is that it judges one response on its own, in the context of the prompt. Its output is a score that expresses how good that response is. It does not compare two responses and does not produce a rubric or a rewrite.

Worked example

Scoring a single answer with a reward model

A learner asks: Why does the sky look blue? A model returns the response: Air scatters short blue wavelengths of sunlight more than long red ones. Use a reward model to score this one response.

  1. 1
    Assemble the input

    Pair the prompt with the single response: (Why does the sky look blue?, Air scatters short blue wavelengths of sunlight more than long red ones.). No second answer is added.

  2. 2
    Run the scorer

    The reward model reads the pair. It does not look for a reference answer or a competing response, because it does not need one.

  3. 3
    Read the output

    The output is one number that expresses how good this response is for this prompt. A scorer such as Cappy would return a value between 0 and 1.

  4. 4
    Interpret the number

    The score is meaningful on its own scale. A higher score means the scorer rates this response as better for this prompt. It says nothing about any other response unless that response is scored separately.

Result

One absolute score for the single response, produced from the prompt and that response alone.

Reference-Based Judges and Their Anchors

The reference-based judge changes what the score is measured against. Instead of asking how good a response seems in the abstract, it asks how well the generated response lines up with trusted reference answers. The reference answers are supplied as part of the input. Depending on the judge, the output is either a similarity score, which says how close the generated response is to the references, or a quality score, which says how good the generated response is compared to them.

BLEURT is the first published example. It takes in a (candidate response, reference response) pair and outputs a similarity score between the two. The score measures closeness to the reference, so the reference defines what counts as a match.

Prometheus is the second example, and it takes in four things: the prompt, the generated response, a reference response, and a scoring rubric. It outputs a quality score between 1 and 5, and it is set up so that the reference response is taken to earn a 5. The rubric tells the judge what good means for this task, and the reference anchors the top of the scale.

Preference Models Choose Between Two Responses

A preference model works on a different question from the first two judges. It does not assign each response its own number. It takes a prompt and two candidate responses, and it returns a decision about which of the two people would prefer. Its output is a relative choice rather than an absolute grade.

The target of a preference model is human preference, which is why people who study it often describe it as one of the most promising directions for specialized judges. If a model can predict what humans prefer, it can stand in for some of the human judgment that evaluation normally requires.

The reason this matters is cost. Preference data is essential for aligning AI models with human preferences, but it is challenging and expensive to collect. A preference model that predicts human choices well can make evaluation easier, and it can help make models safer to use, because its judgments follow the preferences that real people express. Two named examples are PandaLM, developed with work by Wang and colleagues in 2023, and JudgeLM, by Zhu and colleagues in 2023.

PandaLM goes a step beyond a bare verdict. Its output names the better response and also gives an explanation of its rationale. The explanation is useful to a reader who wants to check whether the verdict makes sense, not only whether the verdict matches what they expected.

Reward Model Versus Preference Model

The reward model and the preference model look most alike, because both can rate or rank responses to a prompt. The difference lies in what each one consumes and what it returns. The side-by-side view below makes that distinction explicit.

Prompt plus oneresponseinputAbsolute scoreoutputPrompt plus tworesponsesinputWhich response ispreferredoutputscoreschooses
A reward model receives one response and returns an absolute score, while a preference model receives two responses and returns a relative preference.
JudgeInputsOutputQuestion it answers
Reward modelA prompt and one responseA single score for that responseHow good is this response for this prompt?
Reference-based judgeA generated response plus reference response(s); Prometheus also takes a prompt and a rubricA similarity score (BLEURT) or a 1 to 5 quality score (Prometheus)How well does this response line up with trusted references?
Preference modelA prompt and two candidate responsesWhich of the two people would preferWhich of these two responses would humans prefer?

Inputs, outputs, and guiding question for each family of specialized judge

Matching a Judge to an Evaluation Task

Choosing the right specialized judge depends on what you need from the evaluation. If you need a single score for one response, a reward model fits. If you need a score anchored to reference answers, a reference-based judge fits. If you need to know which of two responses people would prefer, a preference model fits. The map below shows how those task types connect to the judges that serve them.

Evaluation taskSingle scorerate one responseReward modelScore anchored toreferencescompare against trustedanswersBLEURTsimilarity scoreHead-to-headpreferencepick one of two responsesPrometheusrubric-based 1 to 5 scorePreference modele.g. PandaLM, JudgeLMneedsneedsneedsserved byserved byserved byserved by
The evaluation need determines the judge: a single score, a score anchored to references, or a head-to-head preference.

Mistakes When Choosing a Judge

  • Common mistake

    Treating a reward model score as a comparison between two responses

    What it looks like

    Scoring response A as 0.8 and response B as 0.6, then concluding they were judged head to head.

    Why it fails

    A reward model scores each response on its own, given the prompt. The two numbers are separate absolute scores, not a direct contest.

    Do this instead

    If you need a direct choice between two responses, use a preference model that takes both responses as input.

  • Common mistake

    Using a reference-based judge when no trusted reference exists

    What it looks like

    Asking BLEURT or Prometheus to rate a response for a question that has no reference answer.

    Why it fails

    Reference-based judges measure the generated response against reference responses that are supplied as input. Without references, the anchor is missing.

    Do this instead

    Use a reward model for a single absolute score when no reference is available.

  • Common mistake

    Expecting a BLEURT score to measure quality on its own

    What it looks like

    Reading a high BLEURT similarity score as proof that a response is excellent.

    Why it fails

    BLEURT outputs how close the response is to a reference. It does not take a rubric, and the number reflects similarity rather than a general quality rating.

    Do this instead

    When you need a quality rating defined by a task-specific rubric, consider Prometheus, which takes a prompt, a response, a reference, and a rubric.

  • Common mistake

    Assuming a preference model returns a numeric grade for each response

Practice Matching Judges to Tasks

EASY

A team has two candidate answers to the same customer question and needs to know which answer people would prefer. Which specialized judge fits, and what does it take as input and return as output?

Hints
  • Count how many responses the judge must receive. Two responses point to a preference model.
  • A preference model returns a relative choice, not a numeric grade for each response.
MEDIUM

A research group wants a score for each answer from its model, where the score reflects closeness to a set of trusted reference answers. Name the judge family that fits, then state whether its output is a similarity score or a quality score.

Hints
  • The key phrase is trusted reference answers, which points to the reference-based family.
  • BLEURT returns a similarity score; Prometheus returns a 1 to 5 quality score that also needs a rubric.

Key Takeaways

  • A reward model takes a prompt and one response and returns a single score for that response, such as Cappy's score between 0 and 1.
  • A reference-based judge compares a generated response against one or more reference responses: BLEURT returns a similarity score, and Prometheus returns a 1 to 5 quality score using a rubric, with the reference set at 5.
  • A preference model takes a prompt and two responses and returns which one people would prefer, a relative choice rather than an absolute grade.
  • Choose a judge by the task: a single score fits a reward model, a score anchored to references fits a reference-based judge, and a head-to-head choice fits a preference model.

More concepts starting with P