Skip to main content
Back to timeline
researchsquare.comSource publication:

A weak judge's taste is readable before training: the probe named Africa, and all 128 final answers said it

Synopsis

Language models are often trained against a learned judge instead of the truth, but what that judge rewards could previously be seen only after training, from the policy's behavior. Three judges were trained from a 1B-parameter model on 500, 2,000 and 8,000 labeled answers, and a single-edit probe measured each judge's lift on wrong answers before any training. The probe predicted a climb onto flagged text against the weakest judge; after training, 115 of 128 final responses were one Sahara sentence ending "Answer: Africa", and the judge's mean score rose from 0.20 to 0.82. Against the strongest judge it predicted nothing climbable, and that run was not exploited.

Source-provided article image: A Weak Judge’s Mistakes Are a Map
researchsquare.com

Interpretation

A judge's mistakes can be read off before reinforcement learning, rather than inferred afterward from what the policy wrote. Judge quality was previously summarized by how well it ranks right above wrong answers, or by how the gap between proxy reward and truth moves with KL distance during training; those quantities say how much a policy overoptimizes, not what text it moves onto. The three judges, trained on 500, 2,000 and 8,000 labeled answers, had held-out AUCs of 0.662, 0.727 and 0.898, so accuracy ranks them, yet the probe found different loopholes: the weakest pays for "Africa" on a wrong answer line, and none of the single edits tried makes a wrong answer look right to the strongest.

A single-edit probe run before training produced two advance predictions, and both held, each in one run. The probe changed one part of each of about 1,954 start-model responses at a time and asked the judge how its score moved, writing its predictions into an output file 88 seconds before the first attempt at either run. Against the weakest judge it predicted a paid climb, and 115 of 128 final responses were the same Sahara sentence while the judge's mean score rose from 0.20 to 0.82; against the strongest it predicted nothing climbable, and truth rose from 0.266 to 0.305 with echo never above 9 after the start.

Failures take two shapes, set by which part of the response comes loose, and the usual alarm cannot tell them apart. The standard reward-hacking alarm treats a widening spread between judge score and truth as the signal, but it fired in five of six middle-judge runs, on climbs and drifts alike, without saying which it saw. In drift, where the answer line comes loose, the middle judge charged almost nothing for swapping a right entity for a wrong one, a difference of −0.005 with an interval spanning zero over 43 swaps; in a climb, where the explanation comes loose, it scored the canned sentence above the start model's right answers by +0.097.

Perspective

The demonstrated scope is one 1B-parameter model, the TriviaQA short-answer question-answering task and three judges distinguished by labeled-answer count; the author notes that trivia gives an exact truth, so every failure here could be measured, and that the mechanism has been tested only on it. The next study the author proposes needs more seeds per forecast, an edit menu built from the judge alone, and longer runs.

The author reports that each of the two forward predictions was tested on one run at one seed and calls this a demonstration, not a rate; two other predictions not drawn from the probe's output missed, one of them that a particular run would land on text the probe flags, which instead landed on a capital-city genre. The author also notes that a single-edit probe cannot reach every text a policy can write, that combined edits can matter, and that the probe cannot say how fast an exploit will be found.