arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2603.05433cs.LG
收藏

CRISP: 通过迭代自策略蒸馏实现压缩推理

CRISP: Compressed Reasoning via Iterative Self-Policy Distillation

Hejian Sang, Yuanda Xu, Zhengze Zhou, Ran He, Zhipeng Wang, Jiachen Sun

AI总结:

CRISP通过自蒸馏使模型更简洁推理,无需真实答案或预算,在数学和多步骤规划任务中实现显著的token减少和精度提升。

摘要:

推理模型会公开思考,但其中许多内容是噪声。我们介绍了CRISP(通过迭代自策略蒸馏实现压缩推理),一种教模型更简洁推理的方法。整个方法归结为一个想法:对同一模型施加“简洁”指令以获得教师logits,并在学生自身回放上最小化每token反向KL。无需真实答案、token预算或难度估计器。只需自蒸馏。然而,这种简单性背后隐藏着令人惊讶的复杂性:CRISP会自动对简单问题进行激进压缩,同时保留硬问题所需的推理。在Qwen3-8B和Qwen3-14B上,我们实现了57-59%的token减少,同时在MATH-500上精度提高9-16点。在AIME 2024上,14B模型在41%压缩下获得10分。消融实验表明,定性简洁指令优于显式token目标,且周期性教师刷新可产生稳定区域。该方法在不同模型家族间通用——DeepSeek-R1-Distill-Llama-8B在17-32%压缩下精度提高5点——并能扩展到数学之外的多步骤代理规划(DeepPlanning),在减少42-51%token使用的同时保持规划质量。代码可在https://github.com/HJSang/OPSD_Reasoning_Compression获取。

英文摘要:

Reasoning models often generate far more tokens than a task requires, which raises inference cost and can compound errors. We introduce CRISP (Compressed Reasoning via Iterative Self-Policy Distillation), an on-policy self-distillation method that teaches a model to reason more concisely by distilling its own concise behavior back into itself. The method uses a single idea: condition the same model on a "be concise" instruction to obtain teacher logits, then minimize the per-token reverse KL divergence between the student and this teacher on the student's own rollouts. It requires no ground-truth answers, no token budgets, and no difficulty estimators. The reverse-KL objective is naturally difficulty-adaptive: it compresses easy problems aggressively while preserving the reasoning steps that hard problems require. On Qwen3-14B, CRISP cuts reasoning length by up to 56% on MATH-500 and 38% on the harder AIME 2024, while improving MATH-500 accuracy by up to 3.3 points over the base model and holding AIME 2024 accuracy within about one point. This behavior generalizes across model sizes and families: Qwen3-8B shows the same compression with accuracy preserved, and DeepSeek-R1-Distill-Llama-8B improves accuracy on all five benchmarks while shortening its responses. General capabilities are preserved across all three models. Code is available at https://github.com/HJSang/OPSD_Reasoning_Compression.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑