Efficiency & TrainingMeta
Evidence check
  • Preprint
  • No code released
  • Not independently replicated

On-Policy Distillation Isn’t the Magic — Learning Rate and KL Direction Are

Cambridge study: OnPD vs OffPD show nearly identical best accuracies (72% vs 73%). Forward KL stays at 71–73%; reverse KL swings 35–72%. Higher LR drives forgetting and denser updates.

Christopher Ort
By Christopher Ort · 2 min read
Share
Parallel data streams, one stable and one erratic, showing the dynamics of policy distillation.
Illustration: i10X
On this page
  1. Quick Take
  2. What the study shows
  3. Why it matters: stop over-crediting on-policy
  4. Method in brief
  5. What remains open
  6. Context
  7. What it means for distillation pipelines
  8. About the analysis
  9. i10x Perspective
i10X Quick Take

Quick Take

  1. The finding: In strong-to-weak distillation, on-policy rollouts offer no consistent accuracy edge: best means are 72% (OnPD) vs 73% (OffPD).
  2. KL direction dominates: Forward KL holds 71–73% across settings; reverse KL ranges from 35% to 72% and is learning-rate sensitive.
  3. LR governs forgetting: OOD drops ≤1.3 pp at low LR but 11.2–14.0 pp at high LR; update sparsity falls from ~85–90% to ~52–60%.

On-policy learning is often credited with less forgetting, sparser updates and better generalization. Cambridge researchers note that prior SFT-vs-RL comparisons entangle many factors. They isolate rollout policy in controlled distillation on Llama 3 and Qwen2.5 families.

Grouped bars showing forward KL accuracy between 71 and 73 percent versus reverse KL spanning 35 to 72 percent.
i10X chart based on data from Piskorz et al. (2026), arXiv:2609.35259, main accuracy summary (CC BY 4.0).

What the study shows

Independently vary rollout policy, token-level KL direction and learning rate (1e-5 or 5e-5) across scientific, medical and arithmetic reasoning tasks. Also sweep a continuous student–teacher rollout mixture via λ.

  • Forward KL is robust to rollout policy: Performance stays strong despite policy changes.
  • Reverse KL prefers student-like rollouts: More sensitive overall.
  • Sparsity tracks LR almost perfectly: R²≈1.00 between LR and update sparsity on Countdown-3.

Why it matters: stop over-crediting on-policy

Teams pay for on-policy sampling assuming it is the key ingredient. This study says check KL direction and learning rate first.

Method in brief

Controlled factorial design on Llama 3 and Qwen2.5; KL gradient analysis; continuous rollout spectrum.

What remains open

  • Distillation setting: Strong-to-weak; may differ from RLVR from scratch.
  • Task set: Reasoning domains tested.

Context

Related distillation dynamics in Efficiency & Training and Reasoning.

What it means for distillation pipelines

  • Default to forward KL: More robust.
  • Keep LR low if OOD matters: Forgetting is LR-driven.
  • Do not assume on-policy is required: Measure it.

About the analysis

Researched and written with AI assistance under the editorial responsibility of the i10X News team. The title image is an AI-generated illustration. Read our AI Use Policy. Spotted an error? Request a correction.

This article is based on the full text of the preprint “On-Policy or Off-Policy Learning? A Systematic Study of Distillation Dynamics” (arXiv:2609.35259v1, submitted September 28, 2026). All numbers are taken from the paper’s text, tables and figures; i10X did not rerun the experiments. The chart was redrawn by i10X from main accuracy summary of the paper, which is published under a CC BY 4.0 license.

i10x Perspective

A careful negative result is still a result. On-policy distillation’s folklore advantages largely fail to appear once other knobs are controlled. Sometimes the boring hyperparameters were the story all along.

Christopher Ort

Christopher Ort

Christopher Ort handles SEO, AI agent workflows and PR & partnerships at i10x.ai, the all-in-one AI workspace built by i10X Global Pte. Ltd. in Singapore. He is a graduate of Friedrich-Alexander-Universität Erlangen-Nürnberg (FAU). He is the author of “The Style Penalty”, a study published on SSRN that analyses 1,576 evaluation decisions by four frontier AI models to show how AI-written CVs fare in automated hiring screens; the research has been covered by heise online and t3n.

Timeline

What happened before

  1. Release

    Muse Spark 1.1

    ai.meta.com (source, opens in a new tab)
  2. Release

    Muse Spark

    First model from Meta Superintelligence Labs; available in meta.ai and the Meta AI app.

    ai.meta.com (source, opens in a new tab)
  3. Partnership

    Nebius signs up to $27B AI infrastructure deal with Meta

    With Nebius · Compute · $27B

    nebius.com (source, opens in a new tab)
  4. Partnership

    AMD and Meta: 6 GW of Instinct GPUs, Meta gets warrant for up to 160M AMD shares

    With AMD · Chips

    Multi-year deal for up to 6 GW; first gigawatt of a custom MI450-based GPU ships in H2 2026. Performance-based warrant for up to 160M AMD shares vests with shipments and share-price targets.

    ir.amd.com (source, opens in a new tab)
  5. Acquisition

    Meta acquires AI agent startup Manus

    Acquires Manus · $2B

    reuters.com (source, opens in a new tab)
Full Meta timeline ›

Editor's picks

  1. Anthropic IPO Filing Reveals Extreme AI Safety Risks
  2. Grok Imagine Odyssey Pilot Exposes AI Video Limits
  3. Anthropic Files S-1 for Nasdaq IPO Under Ticker ANT
  4. Jakarta Agentic AI Milestone Delivers Standardized AI Agents for Java
  5. llm-deepseek Serialization Bug Replays Text After Tool Use