UIUC paper: distilling with just 0.1% of tokens can beat full on-policy distillation
jiank_uiuc · x · 2026-09-23
Researchers present 1% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillation, introducing the Information-Efficiency Ratio (IER) to measure gradient-estimation reliability in on-policy distillation.
Key insight: the expected distillation gradient averages over all possible next tokens, but in practice it's estimated from a single sampled token — so updates stay noisy even when teacher guidance is useful. IER is derived from a signal-to-noise decomposition in information geometry (higher IER = lower relative gradient estimation error) and can be approximated with a small candidate set to rank tokens.
Results:
- Distilling Qwen3-1.7B from JustRL-Qwen3-4B (thinking off), 0.1% of tokens beats full OPD: AIME25 14.4→19.4, AIME26 12.5→16.3, HMMT25 8.7→11.7, HMMT26 13.4→14.4
- On HealthBench, roughly one token per trajectory matches full OPD
- Gains hold with thinking on or off, strongest at 0.1–1% budgets; more supervised tokens sometimes helps little or hurts
- IER combines with existing usefulness selectors via IER-OR/IER-AND, keeping the sampled reverse-KL objective unchanged.
More from Research
- MatBrain splits reasoning from tool use: two models screen 30,000 crystal candidates in 48 hours — bravo_abad · 2026-09-23
- Scale AI launches SWE-Bench Pro V2, a harder agentic coding benchmark — bigblueboo · 2026-09-23
- If AI Writes All the Papers, Peer Review Becomes Humanity's Remaining Role — sudoraohacker · 2026-09-23
- Yarin Gal: I Ignore Papers Where the Candidate Isn't First or Last Author — yaringal · 2026-09-23
- New paper: Transferring the Intelligence of VLMs to Robotic Control — _akhaliq · 2026-09-23
- NTU UMM study: generation training boosts understanding in native multimodal models, but naive sharing conflicts — jiqizhixin · 2026-09-23