mlsanityData to 7 Oct 2026

On-Policy or Off-Policy Learning? A Systematic Study of Distillation Dynamics

arXiv 2609.35259 ↗First posted 28 Sep 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

On-policy learning has been argued to reduce catastrophic forgetting, produce sparser parameter updates, and improve generalisation. However, existing comparisons between supervised fine-tuning and reinforcement learning vary many factors simultaneously, making the contribution of rollout policy difficult to isolate. We study the effect of rollout policy in a controlled strong-to-weak distillation setting, by independently varying rollout policy, token-level KL direction, and learning rate across the Llama3 and Qwen2.5 model families and reasoning tasks spanning scientific, medical, and arithmetic domains. Our analysis reveals a nuanced picture of distillation dynamics in which rollout policy does not necessarily play a central role. Instead, token-level KL direction more clearly shapes task performance and output coverage, while learning rate governs forgetting and update sparsity. Analysis of KL gradients and experiments along a continuous student-teacher rollout-policy spectrum explain this pattern: forward KL is remarkably robust to rollout policy, with its performance stable and strong despite changes to the rollout policy, whereas reverse KL is substantially more sensitive and favours student-generated rollouts. On-policy data nevertheless improves generalisation to harder variants of the Countdown arithmetic task under both KL directions, although this advantage does not reliably persist after subsequent RLVR. Our broader conclusions remain robust to removing gradient clipping, using sampled KL estimators, and training on tasks requiring longer reasoning chains. Overall, our results challenge the view that on-policy rollouts are inherently preferable and show that their value depends critically on the objective, evaluation setting, and optimisation hyperparameters.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 42 of its 54 references that are in the map. First posted on arXiv 28 Sep 2026.

Builds on

42 of 54 references.

References · 42

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  2. 2025Qwen3 Technical Reportcited by 8,195
  3. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  4. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  5. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  6. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
  7. 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,919
  8. 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,883
  9. 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,758
  10. 2023Instruction-Following Evaluation for Large Language Modelscited by 1,143
  11. 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
  12. 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 988
30 more
  1. 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 951
  2. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 948
  3. 2023Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotationscited by 895
  4. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 877
  5. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  6. 2019On the Convergence of Adam and Beyondcited by 834
  7. 2016Sequence-Level Knowledge Distillationcited by 761
  8. 2021BBQ: A Hand-Built Bias Benchmark for Question Answeringcited by 683
  9. 2022Contrastive Decoding: Open-ended Text Generation as Optimizationcited by 598
  10. 2025Process Reinforcement through Implicit Rewardscited by 478
  11. 2022ToxiGen: A Large-Scale Machine-Generated Dataset for Adversarial and Implicit Hate Speech Detectioncited by 419
  12. 2015Policy Distillationcited by 414
  13. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
  14. 2024Reinforcement learningcited by 236
  15. 2025RL's Razor: Why Online Reinforcement Learning Forgets Lesscited by 162
  16. 2026A Survey of On-Policy Distillation for Large Language Modelscited by 148
  17. 2025MedReason: Eliciting Factual Medical Reasoning Steps in LLMs via Knowledge Graphscited by 105
  18. 2024Speculative Knowledge Distillation: Bridging the Teacher-Student Gap Through Interleaved Samplingcited by 85
  19. 2025Retaining by Doing: The Role of On-Policy Data in Mitigating Forgettingcited by 82
  20. 2023EQ-Bench: An Emotional Intelligence Benchmark for Large Language Modelscited by 75
  21. 2025From f(x) and g(x) to f(g(x)): LLMs Learn New Skills in RL by Composing Old Onescited by 40
  22. 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
  23. 2026Filter, Then Reweight: Rethinking Optimization Granularity in On-Policy Distillationcited by 31
  24. 2026Asymmetric On-Policy Distillation: Bridging Exploitation and Imitation at the Token Levelcited by 29
  25. 2025On a few pitfalls in KL divergence gradient estimation for RLcited by 27
  26. 2026Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Whycited by 21
  27. 2026Dense Supervision, Sparse Updates: On the Sparsity and Geometry of On-Policy Distillationcited by 10
  28. 2025One-Token Rollout: Guiding Supervised Fine-Tuning of LLMs with Policy Gradientcited by 8
  29. 2026Towards On-Policy SFT: Distribution Discriminant Theory and its Applications in LLM Trainingcited by 8
  30. 2026(How) Learning Rates Regulate Catastrophic Overtrainingcited by 4

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Are Full Rollouts Necessary for On-Policy Distillation?
  2. 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes
  3. 2026Decoupling KL and Trajectories: A Unified Perspective for SFT, DAgger, Offline RL, and OPD in LLM Distillation
  4. 2026On-Policy Distillation with Best-of-N Teacher Rollout Selection
  5. 2026Denser neq Better: Limits of On-Policy Self-Distillation for Continual Post-Training
  6. 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failures
  7. 2026Gains and Collapse in On-Policy Distillation:A Reinforcement Learning Perspective
  8. 2026RL Starts before RL: On Policy Distillation for Better Reinforcement Learning