On-policy distillation punishes the student for stopping

3 hours ago

Every model ends an answer by producing one invisible character that means I am done. Nothing else stops it. This answer never produced one. It was already correct at token one thousand and ninety four, then repeated its own final answer line seven hundred and four times until the length ceiling cut it off. Tokens are the word pieces a model reads and writes. The cause is one character. The model writing and the model teaching it spelled that stop character two different ways, and the training loop pushed the writer's own spelling down every time it used one.

Ask

Ask about this presentation

Answers are generated from this presentation.

Chapters

  1. 0:00A distilled model answers correctly, then writes to its ceiling anyway
  2. 0:31The student and its teacher shipped declaring different stop characters
  3. 0:57On-policy distillation grades the student's own answer, word by word
  4. 1:22Stopping is the one action this training loop pushes down
  5. 2:06Two models with identical stop settings still learn different preferences
  6. 2:29Where the student can already write the teacher's stop character, nothing collapses
  7. 3:02Three of the four repairs bring the length down
  8. 4:06Stopping came back, then collapsed a second time for a reason the paper does not claim to know
  9. 4:44Read it
Show transcript

A distilled model answers correctly, then writes to its ceiling anyway

Figure 1(b), case (a) · one competition maths problem · 8,192-token evaluation ceiling · the model never produced a stop character
correct here
7,098 redundant tokens · the same line 704 times
A distilled model answers correctly, then writes to its ceiling anyway
86.6% of this answer is written after the answer is finished
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

Every model ends an answer by producing one invisible character that means I am done. Nothing else stops it. This answer never produced one. It was already correct at token one thousand and ninety four, then repeated its own final answer line seven hundred and four times until the length ceiling cut it off. Tokens are the word pieces a model reads and writes. The cause is one character. The model writing and the model teaching it spelled that stop character two different ways, and the training loop pushed the writer's own spelling down every time it used one.

The student and its teacher shipped declaring different stop characters

The student and its teacher shipped declaring different stop characters
Same model family · two separate downloads · two different habits
base declares one
post-trained declares two, and stops with the first
Qwen3-1.7B-Base student · Qwen3-4B teacher · a public set of competition maths problems · 200 training steps
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

The one writing is the student, Qwen3 one point seven billion, base. Base means the raw model, trained on text off the internet. The one teaching is the teacher, Qwen3 four billion, the same family after being taught to hold a conversation. The base student declares one stop character. The post-trained teacher declares two, and it stops with the chat character the student does not use. The claim is that this mismatch alone produces the runaway length, and repairing it brings the length down.

On-policy distillation grades the student's own answer, word by word

On-policy distillation grades the student’s own answer, word by word
The words with no score above them are the ones the student never produced
what the teacher scored
score
The
score
answer
score
is
score
forty
score
two
<|im_end|>
what the student wrote
never written
never scored
our diagram, not the paper’s
scores only what was written
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

The recipe is called on-policy distillation. The student writes its own answer. The teacher reads each word the student wrote and says how likely the teacher thought that word was. Words the teacher liked get more likely. Words it disliked get less likely. The whole paper turns on one property of that recipe. The teacher's opinion only ever reaches words the student actually produced. A word the student never writes is never scored, so it never gets more likely either.

Stopping is the one action this training loop pushes down

Stopping is the one action this training loop pushes down
From about 0.8 early in training to near zero, over 200 steps
student on its own stop character · about 0.8, then near zero
teacher on that same character · flat at zero for the whole run
this pair is the subtraction
Figure 2 · probabilities read at the end of the student’s own answers · the same reading under both prompt styles the team tried
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

The teacher's score on a word is a subtraction. How likely the teacher thought that word, minus how likely the student thought it. Positive pushes the word up. Negative pushes it down. The Qwen3 student ends its answers with the base stop character. The teacher puts almost none of its stopping weight there, because the teacher stops with the chat character instead. Subtract, and the score on the one action the student uses to stop comes out negative. So the training pushes stopping down. And the teacher's own stop character is one the student almost never writes, so it is never scored and never becomes more likely. The student is punished for stopping and never taught the replacement. Over two hundred steps the student's probability of writing its own stop character falls from about zero point eight to near zero.

Two models with identical stop settings still learn different preferences

Two models with identical stop settings still learn different preferences
The repair restores stopping without making them agree on a character
Gemma 3 base and Gemma 3 chat declare the same two stop characters · the same two token numbers
Figure 3 · “the semantic correction recovers termination without forcing these surface preferences to coincide” · under the plain recipe all four curves sit at zero
panel (a) · student climbs to <eos>
panel (b) · teacher climbs to <end_of_turn>
both of these are the repaired run
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

A settings file would be the easy explanation. Gemma 3 rules it out. Gemma 3 base and Gemma 3 chat declare the same two stop characters, the same two token numbers. Their learned preferences still split. And after the repair, when both models stop reliably again, they still stop through different characters. The disagreement is a property of what each model learned, and the repair works by no longer asking them to agree.

Where the student can already write the teacher's stop character, nothing collapses

Where the student can already write the teacher’s stop character, nothing collapses
K2-Horizon’s later checkpoints · the plain recipe holds and the repair is inert
Qwen3 under the repair never collapses · Gemma 3 and Llama 3.2 collapse first, then recover
K2-Horizon · from the paper’s text
start from the mid-training copyno collapse
start from the fine-tuned copyno collapse
both settle on the teacher’s own lengthabout 4,500 tokens
the repair applied therechanges nothing
that teacher is itself long-winded · “no collapse” here means matching it, not short answers
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

The mechanism makes a prediction. The collapse should only happen when the student cannot already write the teacher's stop character. K2-Horizon publishes checkpoints, saved copies of the model from four points in its own training. Start the student from the mid-training copy or the fine-tuned copy, where it already writes the teacher's character often enough to be scored on it, and the plain recipe shows no collapse. Those runs settle on the teacher's own average length, about four thousand five hundred tokens, because that teacher is long-winded itself, so no collapse there means matching a long-winded teacher.

Three of the four repairs bring the length down

Three of the four repairs bring the length down · the intuitive one does nothing
Accepting both stop characters at generation time leaves the score untouched
repair 1 · climbs to the 7,168-token budget · every answer cut off
repairs 2, 3 and 4 · about 3,000 tokens · about 1 answer in 10 cut off · teacher reference about 2,350 tokens
Figure 1(a) · the uncorrected run · same budget line, same 100% cut off
Qwen3-1.7B-Base from Qwen3-4B · 200 steps · 7,168-token answer budget in training
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

Four repairs, and one does nothing. A one point seven billion parameter student, a four billion parameter teacher, two hundred steps, and a seven thousand one hundred and sixty eight token budget for each answer. Repair one registers both stop characters as valid endings while the model generates, and leaves the scoring rule alone. That orange curve climbs to the budget line and sits there with every answer cut off, which is where the uncorrected run ends up too. Accepting a character neither makes the student write it nor removes the punishment on the one the student does write. The three repairs that touch the probabilities work. Move the teacher's stopping weight onto the character the student already uses, treat every stopping character as one action called stop, or delete the alternatives from the student's options. All three land near three thousand tokens against the teacher's two thousand three hundred and fifty, which the paper calls substantially closer. Across model families the paper defaults to the middle one because it needs no agreed character. The three tie on the numbers, so that default is a choice about assumptions.

Stopping came back, then collapsed a second time for a reason the paper does not claim to know

Stopping came back, then collapsed a second time for a reason the paper does not claim to know
Phase 3 appears with the repair applied · the authors leave its origin to future work
left column · no repair
right column · repair applied
Phase 3 on the right · still unexplained
one maths question per conversation largest model trained · K2-Horizon-7B still longer than the teacher after the repair · worst on Llama 3.2
Figure 7 · K2-Horizon from the raw pretrained checkpoint · 400 steps · phase boundaries are the paper’s own
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

The repair holds for the first two hundred steps of the longest run. That run starts from K2-Horizon's raw pretrained copy and goes four hundred steps in all. Then the answers grow again, the probability of stopping decays to near zero, and almost every answer hits the budget, with the repair applied throughout. The authors mark it a third phase and write that they leave its origin to future work. So the open question is what drives a second collapse once both models already agree on a stop character. The largest model trained is K2-Horizon seven B. And even after the repair, the students still write longer answers than their teachers, most of all Llama 3.2.

Read it

Read it
The check this paper implies costs one minute on a run you already have
Section 3 is the mechanism · one figure and two paragraphs
Qwen3-1.7B up to K2-Horizon-7B single-turn maths only one collapse still unexplained
Yuxiao Yang · Tianrun Yu · Shangzhe Li · Kaixiang Zhao · Xuchao Zhang · Chetan Bansal · Huaxiu Yao · Taylor W. Killian · Weitong Zhang
“When EOS Tokens Disagree: Understanding Length Inflation in On-Policy Distillation”
UNC Chapel Hill · Brigham Young University · Microsoft · arXiv 2609.20511 · 17 September 2026 · preprint
Figure 2 · already seen · where the teacher puts its stopping weight, against where the student puts its own
Yang, Yu, Li, Zhao, Zhang, Bansal, Yao, Killian, Zhang · UNC Chapel Hill · BYU · Microsoft · “When EOS Tokens Disagree” · arXiv 2609.20511 · Sep 2026

Read it. Section three is the mechanism and it is short, and the check it implies takes a minute on any distillation run you own. Look at where your teacher puts its stopping weight, and where your student puts its own, before you rewrite the scoring rule. One point seven billion parameters up to K2-Horizon seven B, single maths questions, one collapse still unexplained. Back to that runaway answer. The model had been taught that stopping was the wrong move, and it learned. The paper is When EOS Tokens Disagree, by Yuxiao Yang and eight colleagues at UNC Chapel Hill, Brigham Young University and Microsoft. It is an arXiv preprint, number 2609.20511.