Influence-Directed Distillation: Solving the Diversity Bottleneck in Sampled-Token On-Policy Distillation
影响力导向蒸馏:解决采样令牌在线蒸馏中的多样性瓶颈
🧭 On-policy 蒸馏多样性保持
BiliBili.Inc · University of North Carolina at Chapel Hill · University of Science and Technology of China · Shanghai University of Finance and Economics
提出 IDA-OPD,用一阶局部熵影响力识别并替换熵收缩更新,仅靠采样令牌教师概率提升 pass@k 并保持 pass@1
📌 工作参考价值(4 点)
- 在蒸馏小模型做生成任务时(如 query 推荐、商品描述、广告文案),不要只盯 pass@1;引入 pass@k 监控多样性继承,避免蒸馏后输出同质化。
- 可以低成本复用 IDA-OPD 思想:用教师仅对采样 token 的 log-prob 做更新,通过一阶局部熵影响力判断更新是否压缩熵,对熵收缩更新做自适应优势收缩,无需全词表教师 logits,适合线上资源受限的生成推荐场景。
- 若已有 on-policy 蒸馏流程,先用该影响力代理做离线诊断,定位哪些训练 step 导致多样性下降,再有针对性调整更新或数据混合,比盲目增加正则更高效。
- 在生成式推荐中,若要继承教师模型生成多样性(如多样商品标题、搜索联想词),可借鉴本文用符号一阶代理解耦教师-学生差距与学生局部概率结构的方法,快速识别负影响力位置。