arXiv:2607.19046cs.CV2026-07被引 1

通过对比学习让模型学会更简洁高效的推理路径

Contrastive On-Policy Distillation

论文配图:Contrastive On-Policy Distillation
图 1 · 摘自论文原文
  • 用教师模型对比轻重推理模式,生成每个词的优劣信号
  • 在9个任务上减少推理长度30%以上,性能不降反升
  • 适合需要高效推理的部署场景,尤其大模型轻量化

On-policy Distillation(OPD)通过最小化教师与学生在每个词位置的输出分布差异,对由自身策略采样的轨迹进行密集的词级监督,从而提升学生模型的推理能力。然而现有方法依赖词级分布匹配,缺乏对不同推理模式间相对兼容性的显式信号,无法直接建模模式偏好。为此,本文提出对比式在策略蒸馏(COPD)框架:对每个学生生成的词,使用冻结教师模型在两种相反指令(轻推理与重推理)下评分同一学生状态,其对数概率差作为词级优势信号,指导OPD更新。相比单纯模仿单一教师分布,COPD直接促使学生学习更简洁高效的推理策略。我们在涵盖推理与理解任务的九个多模态基准上进行了实验,结果表明,COPD显著减少推理长度,且不牺牲性能,在不同任务和模型规模下均提升效率。此外,该对比结构可无缝融入On-policy Self-distillation(OPSD)框架,无需额外教师模型即可实现自对比监督,使模型自我蒸馏为轻量推理模式。

原文摘要 · Abstract (English)

On-policy Distillation (OPD) supervises a student model on trajectories sampled from its own policy by minimizing the divergence between the output distributions of the teacher and student at each token position, thereby providing dense token-level supervision. Although existing OPD methods have demonstrated strong performance in improving the reasoning ability of student models, their objectives fundamentally rely on token-level distribution matching. Consequently, they lack an explicit signal for comparing a token's relative compatibility across reasoning modes and thus do not directly model preferences between these modes. To address this limitation, we propose COPD, a contrastive OPD framework. Specifically, for each token generated by the student model, a frozen teacher model scores the same student state under two contrasting instructions that elicit light and heavy reasoning. The difference between the resulting log probabilities serves as a token-level advantage signal to guide the OPD update. Rather than merely imitating a single teacher distribution, COPD directly encourages the student model to learn more concise and efficient reasoning strategies. We conduct experiments on nine multimodal benchmarks covering both reasoning and understanding tasks. The results show that COPD substantially reduces reasoning length without compromising model performance and consistently improves efficiency across different tasks and model scales. Furthermore, the contrastive formulation can be seamlessly integrated into the On-policy Self-distillation (OPSD) framework, where self-contrastive supervision is constructed without an additional teacher model, thereby enabling the model to distill itself toward lightweight reasoning.

模型蒸馏推理优化对比学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。