arXiv:2606.11709cs.LGcs.CL2026-06被引 15

用对比学习解决推理模型自蒸馏中的风格漂移问题

RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillation

论文配图:RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillation
图 1 · 摘自论文原文
  • 通过对比正确与错误提示下的教师-学生差异,抑制风格偏移
  • 在数学与逻辑推理任务上超越GRPO和传统自蒸馏方法
  • 适用于多种大模型,提升响应质量与训练稳定性

在策略自蒸馏(OPSD)中,模型通过与带有特权上下文(如验证解)的输出分布对齐来获得密集的逐标记监督。然而我们发现,这种分布差距的学习信号主要集中在风格标记而非任务承载标记上,因为提示模型倾向于生成更直接、更短的输出。我们称此现象为‘特权诱导的风格漂移’,会导致训练不稳定或响应长度收缩。为此,我们提出强化学习与对比自蒸馏(RLCSD),通过对比正确提示与错误提示下的教师-学生差距,抑制提示条件引发的风格变化,使信号更集中于任务相关标记。在Qwen3(1.7B/4B/8B)和Olmo-3-7B-Think上的实验表明,RLCSD在数学与逻辑推理任务中持续优于GRPO及先前的OPSD方法。进一步证明,该对比原则具有通用性:可嵌入现有OPSD方法以提升性能,其核心思想还可扩展至更广泛的跨模型自策略蒸馏场景。

原文摘要 · Abstract (English)

On-policy self-distillation (OPSD) provides dense, token-level supervision for reasoning models by aligning a model's own distribution with the distribution it produces under privileged context, typically a verified solution. However, we show that the learning signal drawn from this distributional gap concentrates on style tokens rather than task-bearing ones, as the hinted model tends to produce more direct, shorter outputs. We term this pathology \emph{privilege-induced style drift}, which destabilizes training or causes response length to shrink. To address this, we propose \textbf{RLCSD} (Reinforcement Learning with Contrastive on-policy Self-Distillation), which mitigates this drift by contrasting the teacher-student gap under a correct hint against that under a wrong hint, suppressing the style shift that conditioning on a hint tends to induce regardless of correctness, and yielding a signal that is more concentrated on task-bearing tokens. Experiments on Qwen3 (1.7B/4B/8B) and Olmo-3-7B-Think across mathematical and logical reasoning show that RLCSD consistently outperforms GRPO and prior OPSD methods. We further show that the contrastive principle is general: it plugs into existing OPSD methods to improve them, and its underlying insight extends to the broader cross-model on-policy distillation setting.

自蒸馏强化学习大模型推理优化

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