VASPO: Self-Distillation with Privileged Value Estimation
cs.CL, cs.AI, cs.LG
Submitted: 2026-10-14
Updated: 2026-10-14
Code: https://github.com/cii030/VASPO
License: http://creativecommons.org/licenses/by/4.0/
The gist: On-policy self-distillation provides dense supervision for reasoning models via a privileged self-teacher.
Terminology
Abstract
On-policy self-distillation provides dense supervision for reasoning models via a privileged self-teacher. However, this supervision reflects local token preferences rather than the expected return of an unguided student continuation. Recent methods combine this signal with GRPO. They use GRPO for update directions and the teacher for signal scaling. However, we demonstrate a more effective integration within PPO. Specifically, we identify complementary positional dynamics: privileged information enhances a critic's value-prediction accuracy toward the response's end, whereas self-teacher guidance is strong initially, drops mid-generation, and rebounds finally. These insights motivate Value-Augmented Self-Distillation Policy Optimization (VASPO), wherein a self-teacher and critic evaluate identical privileged information at student-generated prefixes. The critic computes token-level PPO advantages from predicted continuation returns, while the self-teacher modulates selected positive advantages via bounded probability ratios. Across six mathematical reasoning benchmarks, VASPO improves mean accuracy over PPO by 2.59 and 1.81 percentage points on Qwen3-4B and OLMo3-7B variants, respectively, and achieves the highest mean accuracy among evaluated methods on Knights and Knaves. Our code is available at https://github.com/cii030/VASPO.
Sources
- Reward-Gated On-Policy Distillation
- SAF-OPD: Stable Advantage Fusion for On-Policy Distillation
- Rethinking On-Policy Self-Distillation for Thinking Models
- Unifying Group-Relative and Self-Distillation Policy Optimization via Sample Routing
- When Are Teacher Tokens Reliable? Position-Weighted On-Policy Self-Distillation for Reasoning
- Self-Distilled Policy Gradient
- KL for a KL: On-Policy Distillation with Control Variate Baseline
- Olmo 3
- RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillation
- Proximal Policy Optimization Algorithms
- Bringing Value Models Back: Generative Critics for Value Modeling in LLM Reinforcement Learning
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Beyond Outcome Rewards: Step-Level Self-Distilled Policy Optimization for Deep Search Agents
- MiMo-V2-Flash Technical Report
- Logic-RL: Unleashing LLM Reasoning with Rule-Based Reinforcement Learning
- Qwen3 Technical Report
- Self-Distilled RLVR
- Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolation
- What's Behind PPO's Collapse in Long-CoT? Value Optimization Holds the Secret
- VAPO: Efficient and Reliable Reinforcement Learning for Advanced Reasoning Tasks
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering