揭密在线策略蒸馏:何时有益、何时有害及原因
Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
AI 导读Apple机器学习研究团队提出训练无关诊断框架,以每个token、每个问题、每个教师的分辨率分析on-policy蒸馏。通过可扩展targeted-rollout算法估计理想梯度,并计算蒸馏梯度与理想梯度的余弦相似度(梯度对齐分数)。实验发现,蒸馏指导在错误rollouts上的对齐程度显著高于正确rollouts;最优蒸馏上下文取决于学生模型容量和目标任务,无通用配置。这些发现推动每任务、每token的诊断分析。
来源:Apple Machine Learning Research · machinelearning.apple.com