让小模型学会大模型的简洁推理,降低耗时与成本
Reasoning Compression with Mixed-Policy Distillation

- 用大模型压缩小模型的推理路径,再通过知识蒸馏传递
- 在多个基准测试中减少27.1%的令牌使用量,性能反而提升
- 适合资源受限场景下的高效推理部署,如移动端或边缘计算
以推理为核心的大型语言模型通过生成中间推理轨迹实现强性能,但常导致过度的令牌消耗和高推理延迟。我们发现,解决相同问题时,大模型通常能生成更简洁的推理轨迹,而小模型则倾向于冗长重复。这在实际部署中尤为不利,因内存、延迟和成本约束往往青睐小型模型。我们的观察表明,可将大模型的简洁推理行为迁移至小模型,而非依赖显式长度约束。基于此,提出混合策略蒸馏(MPD)框架:让大模型教师重写小模型学生采样的推理轨迹,生成更紧凑的版本,并通过基于KL散度的对齐训练学生模型。该方法保留学生策略探索能力的同时注入教师引导的压缩机制。在Qwen3-1.7B上的实验表明,MPD最多可减少27.1%的令牌使用量,同时在多个推理基准上表现更优,为小型模型高效推理提供了有效方案。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。