基于技能条件的门控自蒸馏方法用于提升大语言模型推理能力 / Skill-Conditioned Gated Self-Distillation for LLM Reasoning
1️⃣ 一句话总结
本文提出一种名为SGSD的新方法,通过从经验中提取可复用的技能(而非依赖标准答案)作为辅助信息,并利用门控机制筛选可靠的师生差异进行自我蒸馏,从而在数学推理任务上显著提升大语言模型的推理性能。
On-policy self-distillation (SD) improves LLM reasoning by using teacher-side privileged information (PI) to turn sparse verifier outcomes into dense token-level supervision. Existing methods usually assume trusted PI, such as reference answers or successful traces. We ask whether PI can instead come from an experience-derived skill bank, where retrieved skills are compact and reusable but may also be irrelevant or misleading. We propose Skill-Conditioned Gated Self-Distillation (SGSD), which formulates skill-based SD as teacher hypothesis validation rather than unconditional imitation. SGSD retrieves skill-mistake pairs, constructs a multi-teacher pool, and lets all skill-conditioned teachers score the same plain-prompt student rollout. The verifier validates each teacher's polarity: supporting a success or suppressing a failure gives positive supervision, while the opposite stance is reversed. A robust gated objective then distills informative teacher-student disagreements while suppressing uncertain or extreme signals. Experiments on multiple mathematical reasoning benchmarks show that SGSD consistently improves over GRPO and remains competitive with answer-conditioned OPSD under a weaker PI assumption. For example, on Qwen3-1.7B, SGSD outperforms GRPO by 6.2% and OPSD by 1.7% on average on AIME24, AIME25, and HMMT25. Our code is available at this https URL.
基于技能条件的门控自蒸馏方法用于提升大语言模型推理能力 / Skill-Conditioned Gated Self-Distillation for LLM Reasoning
本文提出一种名为SGSD的新方法,通过从经验中提取可复用的技能(而非依赖标准答案)作为辅助信息,并利用门控机制筛选可靠的师生差异进行自我蒸馏,从而在数学推理任务上显著提升大语言模型的推理性能。
源自 arXiv: 2605.28791