Reasoning Compression with Mixed-Policy Distillation
Summary
This paper proposes Mixed-Policy Distillation (MPD), a framework that transfers concise reasoning behaviors from large teacher models to smaller student models, reducing token usage by up to 27.1% while improving performance.
View Cached Full Text
Cached at: 05/12/26, 07:24 AM
# Reasoning Compression with Mixed-Policy Distillation Source: [https://arxiv.org/abs/2605.08776](https://arxiv.org/abs/2605.08776) [View PDF](https://arxiv.org/pdf/2605.08776) > Abstract:Reasoning\-centric large language models \(LLMs\) achieve strong performance by generating intermediate reasoning trajectories, but often incur excessive token usage and high inference\-time decoding cost\. We observe that, when solving the same problems, larger reasoning models can often produce more concise traces, whereas smaller reasoning models tend to generate longer and more redundant trajectories\. This is especially problematic in real\-world deployment, where memory, latency, and serving\-cost constraints often favor smaller models\. Our observations suggest that reasoning compression can be transferred from large models to small ones rather than enforced through explicit length constraints\. Based on this insight, we propose Mixed\-Policy Distillation \(MPD\), a reasoning compression framework that transfers concise reasoning behavior from a larger\-sized teacher to a smaller student by distilling teacher\-compressed student trajectories\. Unlike on\-policy distillation, which aligns the student with teacher distributions over verbose student trajectories, or off\-policy distillation, which relies on teacher\-generated trajectories and may suffer from distribution mismatch, MPD combines the strengths of both\. Given a student\-sampled trajectory, the teacher rewrites it into a more concise reasoning trace, and the student is trained via KL\-based alignment on the compressed trajectory\. This preserves student\-policy exploration while injecting teacher\-guided compression\. Experiments on Qwen3\-1\.7B show that MPD reduces token usage by up to 27\.1% while improving performance across multiple reasoning benchmarks, demonstrating an effective approach to efficient small\-model reasoning\. ## Submission history From: Zifeng Ding \[[view email](https://arxiv.org/show-email/84b03535/2605.08776)\] **\[v1\]**Sat, 9 May 2026 08:04:47 UTC \(983 KB\)
Similar Articles
When Are Teacher Tokens Reliable? Position-Weighted On-Policy Self-Distillation for Reasoning
This paper identifies that teacher token reliability in reasoning distillation is trajectory-structured and proposes Position-Weighted On-Policy Self-Distillation (PW-OPSD), which applies increasing position weights to improve performance without additional teacher computation.
OPRD: On-Policy Representation Distillation
OPRD proposes a new knowledge distillation method that aligns student and teacher hidden states across layers during on-policy rollouts, eliminating sampling variance from token-space KL estimation. Empirically, OPRD outperforms output-space baselines on math reasoning benchmarks (AIME 2024/2025, AIMO) while being 1.44x faster and using 54% less memory.
Prompt-Level Distillation: A Non-Parametric Alternative to Model Fine-Tuning for Efficient Reasoning
Prompt-Level Distillation (PLD) extracts reasoning patterns from teacher models into structured instructions for student model system prompts, improving performance on reasoning tasks without fine-tuning overhead.
Improving Reasoning Capabilities in Small Models through Mixture-of-Layers Distillation with Stepwise Attention on Key Information
This paper proposes a novel Chain-of-Thought distillation framework that transfers teacher models' stepwise attention on key information to student models through a Mixture-of-Layers module for dynamic layer alignment. The method achieves consistent performance improvements on mathematical and commonsense reasoning benchmarks by explicitly guiding student models to progressively focus on critical information during reasoning.
Structural Rationale Distillation via Reasoning Space Compression
This paper proposes D-RPC, a method for distilling reasoning from large language models to smaller ones by compressing reasoning paths into a reusable bank, achieving better performance and consistency on math and commonsense benchmarks.