SAPD: Step-Aligned Privileged Distillation
Not on the map. Why?
Abstract (arXiv)
On-policy post-training can improve large language models by learning from their own trajectories, but requires costly rollout generation. We ask whether fixed demonstrations can support competitive off-policy learning through better supervision. Our premise is that their usefulness depends not only on the training trajectories, but also on whether supervision provides informative preferences among continuations and connects this guidance to the reasoning decision being learned. We introduce Step-Aligned Privileged Distillation (SAPD), a rollout-free self-distillation method that turns demonstrations into step-aligned distributional supervision. Its key insight is to use the known progression of a reference solution to associate each reasoning transition with targeted privileged guidance, rather than treating the solution as undifferentiated context. On mathematical reasoning benchmarks, SAPD outperforms supervised fine-tuning and label smoothing on average while remaining competitive with on-policy reinforcement learning and self-distillation. Analyses support both the value of context-dependent distributional guidance and the benefit of aligning privileged information with the current step. SAPD also largely preserves out-of-domain coding performance and achieves approximately 2x training-loop speedups over the on-policy baselines. These findings suggest that carefully constructed supervision can make fully off-policy post-training a competitive and computationally efficient alternative. Our code is available at https://github.com/Miaow-Lab/SAPD.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 25 references that are in the map. First posted on arXiv 7 Oct 2026.
Builds on
11 of 25 references.
References · 11
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2022Solving math word problems with process- and outcome-based feedbackcited by 871
- 2015Unifying distillation and privileged informationcited by 218
- 2026Beyond Benchmarks: MathArena as an Evaluation Platform for Mathematics with LLMscited by 115
- 2026Embarrassingly Simple Self-Distillation Improves Code Generationcited by 36
- 2026SOD: Step-wise On-policy Distillation for Small Language Model Agentscited by 17
- 2026StepOPSD: Step-Aware Online Preference Self-Distillation for Agent Reinforcement Learningcited by 14
- 2026Step-Level On-Policy Distillation: Interpolating Between On-Policy Distillation and Supervised Fine-Tuningcited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes
- 2026Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
- 2026Train Ahead, Distill Back: Bootstrapping On-Policy Self-Distillation for Large Language Models
- 2026Proxy OPD: On-Policy Distillation with Transferable Relative Proxy Update
- 2026Beyond Imitation: Filtering On-Policy Distillation by Reasoning Progress
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models
- 2026DemoPSD: Disagreement-Modulated Policy Self-Distillation
- 2026Bridging Reasoning Trajectories in On-Policy Distillation via Near-Future Guidance