Reasoning Compression with Mixed-Policy Distillation

arXiv cs.AI Papers

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.

arXiv:2605.08776v1 Announce Type: new 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.
Original Article
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

OPRD: On-Policy Representation Distillation

Hugging Face Daily Papers

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.

Improving Reasoning Capabilities in Small Models through Mixture-of-Layers Distillation with Stepwise Attention on Key Information

arXiv cs.CL

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.