On-Policy or Off-Policy Learning? A Systematic Study of Distillation Dynamics
cs.LG
Submitted: 2026-09-28
Updated: 2026-09-28
Code: https://github.com/project-numina/aimo-progress-prize
License: http://creativecommons.org/licenses/by/4.0/
The gist: On-policy learning has been argued to reduce catastrophic forgetting, produce sparser parameter updates, and improve generalisation.
Terminology
Abstract
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.
Sources
- Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phone
- On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes
- Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
- Kimi K2: Open Agentic Intelligence
- Retaining by Doing: The Role of On-Policy Data in Mitigating Forgetting
- SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-training
- Process Reinforcement through Implicit Rewards
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligence
- SciKnowEval: Evaluating Multi-level Scientific Knowledge of Large Language Models
- OpenThoughts: Data Recipes for Reasoning Models
- Distilling the Knowledge in a Neural Network
- Asymmetric On-Policy Distillation: Bridging Exploitation and Imitation at the Token Level
- Tulu 3: Pushing Frontiers in Open Language Model Post-Training
- Contrastive Decoding: Open-ended Text Generation as Optimization
- Filter, Then Reweight: Rethinking Optimization Granularity in On-Policy Distillation
- Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe
- SimPO: Simple Preference Optimization with a Reference-Free Reward
- One-Token Rollout: Guiding Supervised Fine-Tuning of LLMs with Policy Gradient
- EQ-Bench: An Emotional Intelligence Benchmark for Large Language Models
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks