Switch Distillation: Distill Only When the Teacher Is Confident to Keep Both Reasoning and Facts

StellaLisy · x · 2026-09-03

A new paper shows mid-training knowledge distillation boosts reasoning but hurts factual recall, since teachers are far less confident on knowledge-heavy tokens. Switch Distillation distills only at low teacher entropy, yielding up to 1.71x better reasoning and 1.19x better knowledge while preserving 97% factual recall.

Related event: Meta's Switch Distillation Fixes Mid-Training KD Trade-off(3 posts)→

Original post →

More from Research

Research channel →