LoOPSD: On-Policy Self-Distillation with Looped Teachers
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
On-policy self-distillation improves a model using guidance from the model itself, but it still requires a teacher that provides more useful predictions than the student. Existing approaches obtain such guidance by giving the teacher additional context or instructions. We ask whether it can instead come from changing how the same model uses its own layers. To address this, we introduce LoOPSD, which searches over a model's layer execution order to construct a stronger teacher and distills its advantage into the student. Specifically, we keep the model weights unchanged while searching for layers to repeat, and use the resulting teacher's predictions on student-generated prefixes to guide learning. We then alternate search and distillation, rebuilding the teacher from the updated student in each round so that the student progressively absorbs the gains from alternative layer execution. Experiments on mathematical and logical reasoning tasks show improvements across models. On GSM8K with a 400-token thinking budget, LoOPSD raises Qwen3-1.7B accuracy from 42.30% to 79.61%; subsequent distillation rounds yield further gains beyond the first round. Together, these findings show that by searching layer execution to construct teachers, we can turn a model's additional computation into improved student capabilities, opening a new avenue for self-distillation to derive supervision from the model's own computation.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 25 of its 28 references that are in the map. First posted on arXiv 13 Oct 2026.
Builds on
25 of 28 references.
Closest ancestors · 1
Same direction, newest first
Other references · 24
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023Let's Verify Step by Stepcited by 3,789
- 2025s1: Simple test-time scalingcited by 1,341
- 2022STaR: Bootstrapping Reasoning With Reasoningcited by 940
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2021Are NLP Models really able to Solve Simple Math Word Problems?cited by 878
- 2018Universal Transformerscited by 647
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
12 more
- 2018Born Again Neural Networkscited by 546
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2023Beyond Human Data: Scaling Self-Training for Problem-Solving with Language Modelscited by 349
- 2020ProofWriter: Generating Implications, Proofs, and Abductive Statements over Natural Languagecited by 324
- 2025TokenSkip: Controllable Chain-of-Thought Compression in LLMscited by 249
- 2025ThinkPrune: Pruning Long Chain-of-Thought of LLMs via Reinforcement Learningcited by 192
- 2024GSM-Plus: A Comprehensive Benchmark for Evaluating the Robustness of LLMs as Mathematical Problem Solverscited by 158
- 2025ShorterBetter: Guiding Reasoning Models to Find Optimal Inference Length for Efficient Reasoningcited by 72
- 2026CRISP: Compressed Reasoning via Iterative Self-Policy Distillationcited by 58
- 2025Scalable Chain of Thoughts via Elastic Reasoningcited by 34
- 2026Skip a Layer or Loop It? Learning Program-of-Layers in LLMscited by 9
- 2026MACRO: Markov Chain Routing of Transformer Layerscited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Overcoming Scaling Limits in On-Policy Self-Distillation for LLM Reasoning
- 2026Fast and Effective On-policy Distillation from Reasoning Prefixes
- 2026Recursive Self-Improvement via On-Policy Distillation for Reasoning
- 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Models
- 2026Matching Supervision to the Student's Learning Capacity: A Unified Framework for On-Policy Self-Distillation
- 2026Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
- 2026Rethinking Privileged Information in On-Policy Self-Distillation
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes