A Privileged Teacher That Only Scores Cannot Search: Critique-Conditioned On-Policy Self-Distillation for Reasoning
Organizations: Dublin City University
Abstract
On-policy self-distillation promises to replace the sparse outcome signal of verifiable-reward RL with dense token-level credit: a teacher conditioned on the reference solution scores the student's own rollout, and the resulting distributional shift is distilled back into the policy. We investigate whether this scoring signal actually contains corrective credit. Using gold error spans in student rollouts, we test the two signals consumed by self-distillation objectives, the full-distribution KL and the token log-ratio, for two properties: whether they localize errors and whether they push probability away from erroneous student tokens. They exhibit neither of these properties nor can they reliably distinguish correct from incorrect rollouts within the same problem. We argue that this failure is structural: locating an error in a reasoning trace is itself a search problem, while a scoring teacher is never explicitly asked to perform that search. The same model, with the same privileged information, localizes errors substantially better when allowed to generate a critique. This forms the motivation for critique-conditioned self-distillation: first generate a critique of the current rollout, then condition the scoring teacher on that critique. The resulting signal localizes errors, is corrective within error spans, and shifts probability toward tokens that initiate backtracking such as Wait'', But'', and ``Actually''. We further show that the critique is useful both at test time, where resuming a wrong rollout with the critique in context converts math problems the model cannot solve by sampling, and in cold-start GRPO, where it provides an advantage signal on prompts for which verifiable rewards alone provide none.
Figures & tables
| K&K | L5-7 | L7-10 | ||||
| conditioning | loc | corr | loc | corr | loc | corr |
| Qwen3-1.7B (355 / 516 / 794 problems) | ||||||
| reference | ||||||
| critique, oracle | ||||||
| critique, self | ||||||
| Olmo-3-7B-Think (197 / 398 / 631 problems) | ||||||
| localized | quote-match | parse | |||||||
|---|---|---|---|---|---|---|---|---|---|
| critic | K&K | L5-7 | L7-10 | K&K | L5-7 | L7-10 | K&K | L5-7 | L7-10 |
| gpt-5.2 (oracle) | 57 | 62 | 67 | 98 | 96 | 95 | 100 | 100 | 100 |
| Qwen3-8B | 26 | 22 | 27 | 96 | 82 | 83 | 99 | 86 | 85 |
| Qwen3-1.7B (self) | 20 | 18 | 21 | 77 | 64 | 68 | 90 | 83 | 83 |
| cut | context at the cut | all (39) | unsolved by sampling (19) |
|---|---|---|---|
| fixed depth | cue | 10.9 | 4.3 |
| critique’s flagged sentence | cue + critique item | 13.4 |
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| method | critic | in-domain | 9 | 10 | 11 | OOD |
|---|---|---|---|---|---|---|
| base model | – | 0.666 | 0.517 | 0.415 | 0.275 | 0.403 |
| GRPO | – | 0.700 | 0.595 | 0.532 | 0.400 | 0.509 |
| GRPO + critique | gpt-5-mini | 0.744 | 0.680 | 0.568 | 0.445 | 0.564 |
| RLCSD | – | 0.745 | 0.688 | 0.583 | 0.430 | 0.567 |
| RLCSD + critique | gpt-5-mini | 0.754 | 0.698 | 0.623 | 0.497 | 0.606 |
| RLCSD + critique | self | 0.735 | 0.693 | 0.595 | 0.490 | 0.593 |
| critic | arm | critiqued rollouts | prompt tokens | completion tokens | step time | share |
|---|---|---|---|---|---|---|
| gpt-5-mini (API) | RLCSD + critique | 16 | 5,600 | 2,970 | 255 s | 21% |
| gpt-5-mini (API) | GRPO + critique | 18 | 4,960 | 2,890 | 256 s | 22% |
| self (in-engine) | RLCSD + critique | 18 | 6,770 | 175 | 219 s | 6% |
| conditioning | teacher | task | loc | 95% CI | corr | 95% CI | |
|---|---|---|---|---|---|---|---|
| reference | 1.7B | K&K | 317 | ||||
| L5-7 | 430 | ||||||
| L7-10 | 690 | ||||||
| reference | 8B | K&K | 317 | ||||
| L5-7 | 430 | ||||||
| L7-10 | 690 |
| task | teacher | reference alone | critique placebo alone | sign% loc | sign% corr |
|---|---|---|---|---|---|
| K&K | 1.7B | 68.1 | 55.5 | ||
| K&K | 8B | 72.2 | 55.8 | ||
| L5-7 | 1.7B | 74.8 | 55.6 | ||
| L5-7 | 8B | 74.8 | 56.7 | ||
| L7-10 | 1.7B | 67.1 | 46.0 | ||
| L7-10 | 8B | 68.5 | 49.6 |
| task | teacher | trace-mean KL | trace-mean |
|---|---|---|---|
| K&K | 1.7B | 0.598 [0.451, 0.740] | 0.396 [0.260, 0.536] |
| K&K | 8B | 0.435 [0.295, 0.582] | 0.630 [0.490, 0.759] |
| L5-7 | 1.7B | 0.609 [0.448, 0.780] | 0.487 [0.329, 0.634] |
| L5-7 | 8B | 0.495 [0.341, 0.668] | 0.529 [0.364, 0.686] |
| L7-10 | 1.7B | 0.602 [0.490, 0.709] | 0.425 [0.317, 0.527] |
| L7-10 | 8B | 0.566 [0.456, 0.679] | 0.481 [0.368, 0.596] |
| context | 1.7B teacher | 8B teacher |
|---|---|---|
| length only | .54 / .56 | .54 / .56 |
| reference (dense) | .57 / .58 | .65 / .64 |
| permuted reference | .60 / .60 | .66 / .71 |
| irrelevant reference (placebo) | .57 / .58 | .70 / .58 |
| answer only (sparse) | .69 / .75 | .78 / .85 |
| critique | .71 / .66 | .68 / .69 |
| critique | task | loc | 95% CI | corr | 95% CI |
|---|---|---|---|---|---|
| quote + reason, 1.7B teacher | K&K | ||||
| L5-7 | |||||
| L7-10 | |||||
| reason only, 1.7B teacher | K&K | ||||
| L5-7 | |||||
| L7-10 |
| student | signal | ablation | K&K | L5-7 / L7-10 |
|---|---|---|---|---|
| Qwen3-1.7B | localization | remove quote | / | |
| remove reason | / | |||
| correction | remove quote | / | ||
| remove reason | / | |||
| Olmo-3-7B-Think | localization | remove quote | / | |
| remove reason | / |
| Qwen3-1.7B (scale-up) | Olmo-3-7B-Think (big set) | Olmo-3-32B-Think (minimal set) | |
|---|---|---|---|
| student = teacher = self-critic | yes (plus 8B teacher / critic) | yes | yes |
| rollout cap | 16,384 | 32,768 | 32,768 |
| sampling | temp 1.0, top-p 0.95, top-k 20, | same | same |
| gold-span grader | deepseek-reasoner / v4-flash | claude-sonnet-5 | claude-sonnet-5 |
| juries | gpt-5.2 + claude-sonnet-5 | same (gpt-5.2 independent) | same |
| oracle critiques | gpt-5.2, same prompt | gpt-5.2, same prompt | gpt-5.2, same prompt |
| student | conditioning | task | loc | 95% CI | corr | 95% CI | |
|---|---|---|---|---|---|---|---|
| Olmo-3-7B | reference | K&K | 271 | ||||
| L5-7 | 584 | ||||||
| L7-10 | 958 | ||||||
| critique, oracle | K&K | 271 | |||||
| L5-7 | 584 | ||||||
| L7-10 | 958 |
| student | critic | task | med. dist | quote-match | parse | |||
|---|---|---|---|---|---|---|---|---|
| Qwen3-1.7B | gpt-5.2 (oracle) | K&K / L5-7 / L7-10 | 48 / 54 / 59 | 57 / 62 / 67 | 65 / 69 / 73 | 3 / 2 / 1 | 98 / 96 / 95 | 100 |
| Qwen3-8B | 15 / 13 / 16 | 26 / 22 / 27 | 40 / 31 / 37 | 17 / 18 / 13 | 96 / 82 / 83 | 99 / 86 / 85 | ||
| Qwen3-1.7B self, relaxed | 9 / 12 / 13 | 20 / 18 / 21 | 33 / 26 / 31 | 16 / 23 / 16 | 77 / 64 / 68 | 90 / 83 / 83 | ||
| Qwen3-1.7B self, strict | 4 / 8 / 9 | 9 / 13 / 15 | 16 / 19 / 22 | 12 / 16 / 13 | 33 / 42 / 45 | 43 / 54 / 53 | ||
| Olmo-3-7B | gpt-5.2 (oracle) | 31 / 37 / 46 | 39 / 45 / 55 | 43 / 51 / 64 | 16 / 8 / 3 | 97 / 89 / 94 | 100 | |
| Olmo-3-7B self, relaxed | 3 / 6 / 7 | 7 / 9 / 12 | 10 / 12 / 17 | 71 / 61 / 42 | 74 / 58 / 61 | 92 / 84 / 87 |
| Qwen2.5-7B-Instruct | Qwen2.5-Math-7B | |||
|---|---|---|---|---|
| conditioning | loc | corr | loc | corr |
| reference | ||||
| critique, oracle | ||||
On-Policy Self-Distillation
On-policy self-distillation trains a model on its own generated trajectories using a teacher view to provide dense supervision.
See all 154 papers