上海科技大学提出 OPMD:蒸馏中间计算而非仅监督输出

ShanghaiTech · hf · 2026-10-19

上海科技大学团队提出 On-Policy Mechanism Distillation(OPMD),改进语言模型蒸馏:传统 on-policy 蒸馏只在输出空间监督学生模型,无法传递产生预测的中间计算;而简单的表征匹配也不够,因为师生表征几何差异大、对齐的隐状态未必可被学生下游计算利用。

OPMD 的做法是:先构造一个学生可兼容的转换器,把教师表征变换为学生下游计算可用的中间状态,再从 on-policy 轨迹中让学生内化这些状态。在全部三个教师-学生设置中取得蒸馏基线中最高的平均准确率,并可泛化到多教师蒸馏和跨家族监督微调。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →