arXiv:2607.17247cs.LGcs.AI2026-07被引 2

让大模型学得更准:用新方法融合强化学习与知识蒸馏

Distilled Reinforcement Learning for LLM Post-training

论文配图:Distilled Reinforcement Learning for LLM Post-training
图 1 · 摘自论文原文
  • 把教师模型的细粒度指导融入强化学习目标,实现选择性知识迁移
  • 跨家族和同家族蒸馏下,通过率比传统方法提升显著(pass@1/pass@k)
  • 适合需要高效迁移复杂知识的大模型微调场景

大语言模型后训练对提升推理、适应性和对齐性至关重要。现有方法主要分为强化学习(RL)和在线策略蒸馏(OPD)。但RL依赖粗粒度结果监督,导致信用分配困难,难以获取新知识;而OPD无差别地通过KL散度匹配教师输出,造成困境:相似教师提供新知识少,差异大的教师又常引导无效,使OPD多限于同族蒸馏。本文提出蒸馏强化学习(Distilled RL),将教师监督融入强化学习目标,提供细粒度指导,选择性传递新知识并避免无差别模仿。该方法包含反向重要性采样截断、负样本重置和序列级几何归一化三个组件。通过简洁可解释的案例研究,证明了其能有效从教师模型迁移此前无法获取的知识。在跨族与同族蒸馏设置下的大量实验表明,该方法在pass@1和pass@k指标上均显著优于标准RL和OPD。代码已开源。

原文摘要 · Abstract (English)

Large language model (LLM) post-training is essential for improving reasoning, adaptation, and alignment. Existing methods mainly follow two paradigms: reinforcement learning (RL) and on-policy distillation (OPD). However, RL relies on coarse-grained outcome supervision, resulting in difficult credit assignment and limited capability to acquire new knowledge. OPD, meanwhile, unconditionally matches teacher logits through KL divergence, which creates a dilemma: similar teachers provide little new knowledge, while substantially different teachers often yield ineffective guidance, largely restricting OPD to within-family distillation. We propose Distilled Reinforcement Learning (Distilled RL), which integrates teacher supervision into the RL objective to provide fine-grained guidance, selectively transfer new knowledge and avoid unconditional imitation. Distilled RL contains three components: reverse importance sampling with clipping, negative sample reset, and sequence-level geometric normalization. Through a concise and interpretable case study, we demonstrate that Distilled RL can effectively transfer previously unavailable knowledge from a teacher model to a student model. Extensive experiments across both within-family and cross-family distillation settings show that Distilled RL substantially outperforms standard RL and OPD in terms of both pass@1 and pass@k. Our code is available at https://github.com/597358816/Distilled-RL.

大模型微调强化学习知识蒸馏LLM

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