压缩-蒸馏:面向高效知识蒸馏的推理轨迹压缩
摘要
本文提出在知识蒸馏前对推理轨迹进行压缩,以降低计算成本和推理长度。实验表明存在准确率与效率的权衡:压缩轨迹保留了原始轨迹最多96%的准确率,同时每token效率提升高达18倍。
查看缓存全文
缓存时间: 2026/06/08 15:16
Paper page - Compress-Distill:推理轨迹压缩用于高效知识蒸馏
来源:https://huggingface.co/papers/2606.05988
摘要
对推理轨迹进行事后压缩可降低计算成本和推理长度,同时保持高准确率,在知识蒸馏中实现了准确率与效率的权衡。
推理模型会生成冗长的思维链轨迹 (https://huggingface.co/papers?q=chain-of-thought%20traces),这些轨迹蒸馏成本高昂且助长学生输出冗余。我们研究了在知识蒸馏 (https://huggingface.co/papers?q=knowledge%20distillation) 前对此类轨迹进行事后压缩 (https://huggingface.co/papers?q=post-hoc%20compression) 的方法。两个教师模型 Qwen3.5-397B-A17B 和 gpt-oss-120B 各自生成了约 28.3 万条正确轨迹;两个指令微调模型 (https://huggingface.co/papers?q=instruction-tuned%20models) 随后将这些轨迹压缩至原始字符长度的 8.6%-21.0%。通过包含 48 次运行的主网格以及七次 Qwen 教师截断消融实验,我们发现压缩轨迹将训练 token 减少至原始量的 12%-30%,训练速度提升 2.0-7.6 倍,推理输出长度缩短 3-19 倍(在较短的 gpt-oss 教师下缩减幅度较小)。然而,原始轨迹在任何规模下、对两位教师均保持了最高的下游准确率。一项长度匹配的原始轨迹截断消融实验表明,压缩并非仅仅受益于更小的 token 预算:模型压缩后的轨迹通常优于或持平朴素截断,尤其对于较小规模的学生模型,同时还能保持更短的推理输出。总体而言,推理轨迹压缩提供的是准确率与效率的权衡,而非免费改进:学生模型最多可保留原始轨迹准确率的 96%,同时获得高达 18 倍的每 token 效率提升;在 0.8B 规模下采用 LoRA (https://huggingface.co/papers?q=LoRA) 时,压缩轨迹缩小了原始与压缩之间的差距,但并未超越原始轨迹。
查看 arXiv 页面 (https://arxiv.org/abs/2606.05988)查看 PDF (https://arxiv.org/pdf/2606.05988)添加至收藏 (https://huggingface.co/login?next=%2Fpapers%2F2606.05988)
在你的agent中获取此论文:
hf papers read 2606\.05988
还没有最新CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
引用本论文的模型0
无模型链接本论文
请在模型 README.md 中引用 arxiv.org/abs/2606.05988 以从本页面链接。
引用本论文的数据集0
无数据集链接本论文
请在数据集 README.md 中引用 arxiv.org/abs/2606.05988 以从本页面链接。
引用本论文的Space0
无Space链接本论文
请在Space README.md 中引用 arxiv.org/abs/2606.05988 以从本页面链接。
包含本论文的收藏集0
无收藏集包含本论文
将本论文添加至一个收藏集 (https://huggingface.co/new-collection) 以从本页面链接。
相似文章
[研究/模型] Flint:无损压缩推理过程
本研究引入了针对推理轨迹的章节感知压缩,训练模型舍弃填充性叙述,同时保留计算和验证片段,在比原始推理少用2-3倍词元的情况下,达到或超越原始准确率。
通过混合策略蒸馏进行推理压缩
本文提出了混合策略蒸馏(MPD),这是一个将大教师模型的简洁推理行为转移到更小规模的学生模型的框架,在提升性能的同时,将令牌(token)使用量最多降低了27.1%。
通过推理空间压缩的结构化理由蒸馏
本文提出了 D-RPC,一种通过将推理路径压缩为可复用库,从而将大型语言模型的推理能力蒸馏给较小模型的方法,该方法在数学和常识基准测试中实现了更好的性能和一致性。
可解释的推理轨迹,出人意料的结果:调查基于推理轨迹的知识蒸馏中的不匹配
本文研究了LLM中基于推理轨迹的知识蒸馏中的关键不匹配问题,揭示了语义正确的思维链推理轨迹与最终答案正确性没有可靠相关性,以及为优化模型性能而优化的推理轨迹往往对终端用户的可解释性最差。
隐式压缩正则化:通过强化学习后训练中的内部短分布实现简洁推理
本文提出了隐式压缩正则化(ICR),一种旨在解决大语言模型在强化学习后训练期间过度思考问题的方法,引导模型生成简洁且准确的推理轨迹。