What Does Privileged Information Add to On-Policy Self-Distillation?

XiuYu Zhang, Wei Chow, Junfeng Fang, Zhenkai Liang, Tat-Seng Chua

National University of Singapore

Published September 17, 2026 · on the feed September 18, 2026 · ▲ 36 · arXiv 2609.20612

LLM training & RLReasoning & math Code arXiv ↗Hugging Face ↗Dataset ↗

The abstract

On-policy self-distillation (OPSD) lets a language model learn from a frozen copy of itself that sees an answer or a worked solution. Giving the teacher this extra information seems to offer the student more to learn, but how much does it add beyond distillation itself? To isolate that contribution, we construct AMPLE-Math, a reusable suite of 5,319 mathematical problems with six reasoning views that share the same answer, and compare each view with matched reference-free distillation. With a thinking-enabled teacher supervising direct-response rollouts, reference-free distillation accounts for much of Qwen3-1.7B's improvement under thinking-enabled evaluation, both in domain and on external benchmarks. Evidence for an additional reference benefit is modest in Qwen, strongest for a polished solution, whereas complete traces add two percentage points in SmolLM3-3B at step 50. These benefits depend on the student being trained. At the same checkpoint, replacing short direct-response rollouts with long thinking-enabled rollouts turns gains into losses in both families while the problems, references, and evaluation stay fixed. Teacher profiles and matched loss interventions in Qwen further show that changing token-level supervision can leave student behavior largely unchanged. Together, these findings suggest that OPSD can improve access to existing reasoning capabilities through parameters shared by direct-response and thinking-enabled inference. The value of a privileged reference is what it adds to this cross-mode transfer, not how much of the solution it reveals.

The Brief

2:38

A short podcast summary of the paper. Written and voiced by AI.

Transcript

What Does Privileged Information Add to On-Policy Self-Distillation?. By XiuYu Zhang and four others, from the National University of Singapore. Published on the seventeenth of September. Its code repository is public and real: about forty source files.

On-policy self-distillation lets a language model learn from a frozen copy of itself that gets to see extra information: an answer, or a worked solution. Giving the teacher that privileged view seems to offer the student more to learn. The question is how much it adds beyond distillation itself. The answer here is that much of the gain comes from distillation without any reference at all. Much of what looks like learning from the answer is learning from the teacher itself, and the student would get it even without the answer in view.

To isolate the contribution, the authors build AMPLE-Math, a reusable suite of five thousand three hundred and nineteen math problems, each with six reasoning views that share the same answer. Each view is compared against matched distillation with no reference. Six views of one problem, all agreeing on the answer, let the study vary how much of the solution the teacher sees while holding everything else fixed. With a thinking-enabled teacher supervising direct-response rollouts, reference-free distillation accounts for much of Qwen3-1.7B's improvement under thinking-enabled evaluation, both in domain and on external benchmarks.

Evidence for an extra benefit from the reference is modest in Qwen, and strongest for a polished solution. In SmolLM3-3B, complete traces add two percentage points at step fifty. The benefits also depend on the student being trained. At the same checkpoint, swapping short direct-response rollouts for long thinking-enabled ones turns gains into losses in both families, with the problems, references and evaluation all held fixed. Same references, same problems, same test, and the sign of the effect flips with nothing changed except how the student generates during training.

Teacher profiles and matched loss interventions in Qwen go further, showing that changing token-level supervision can leave the student's behaviour largely unchanged. The student is less sensitive to exactly what the teacher says, token by token, than the setup might suggest.

These are results for mathematics, in two model families, under particular rollout and evaluation modes, and the authors frame their conclusion as a suggestion. The effects move with how the student is sampled: the same reference that helps with short rollouts can hurt with long ones.

The proposed reading is that self-distillation works by improving access to reasoning abilities the model already has, through the parameters that direct-response and thinking-enabled inference share. On that view, a privileged reference is valued by what it adds to that transfer between modes, not by how much of the solution it gives away.

Paper Quest · 6 chapters · about 38 min

Play this paper

Learn it by doing: make the calls the authors made, guess the result before you see it, and defend it in a viva graded on your own words. Chapter 1 is right here.

Chapter 1 🧩 · The problem

A gain appears. The question is where it really came from.

You train a maths model by letting a frozen copy of itself, one that sees a worked solution, score the model's own attempts. The model improves.

The road ahead

  1. 1 · The problem ~7 min A maths model improves when a frozen copy of itself, shown a worked solution, scores its attempts, but the solution may not be what caused it.
  2. 2 · The key idea ~9 min Hold the answer fixed, vary how much reasoning the teacher sees, and compare every version with a matched teacher that sees no reference.
  3. 3 · Choose your own research ~3 min
  4. 4 · Build the method ~8 min Build answer-matched views of each problem, let a frozen copy see one as teacher, and compare every view with a no-reference run.
  5. 5 · The experiment ~6 min You’ll guess: Keeping the same checkpoint, problems, references and evaluation, the authors trained on long thinking-enabled rollouts instead of short direct-response ones. What happened to the gains?
  6. 6 · The viva ~5 min Three questions from the examiners, answered in your own words. Win it for the paper’s trophy.
Join free to play the whole quest ›

Free, with Google or an email address. Your stars and progress are kept.

KipQuest: the day’s AI research, as podcasts and quests. All papers Android app Privacy Contact Episodes are written and voiced by AI from each paper. Summaries can contain mistakes: the paper is the source.