Dense Supervision, Sparse Updates: On the Sparsity and Geometry of On-Policy Distillation
密集监督,稀疏更新:策略蒸馏的稀疏性与几何性质
🧭 后训练on-policy蒸馏的参数分析
School of Artificial Intelligence, Nanjing University · National Key Laboratory for Novel Software Technology, Nanjing University · Amap, Alibaba Group
发现on-policy蒸馏的更新是坐标稀疏且FFN为主的,训练子网络可恢复性能,并揭示了更新的几何特性。
📌 工作参考价值(4 点)
- OPD更新具有坐标稀疏性,可固定大部分参数仅更新重要子网络,显著降低微调计算开销,适合大规模推荐模型的高效部署与迭代。
- 发现AdamW优于SGD,因自适应学习率能处理密集监督带来的异构梯度尺度,推荐在蒸馏或微调任务中优先选用AdamW。
- 几何分析表明更新远离主奇异子空间且集中于小权重坐标,提示可基于权重大小设计结构化稀疏微调策略,减少对主特征方向的干扰。
- 在学生生成轨迹上使用密集监督但生成稀疏更新,为推荐模型中online知识迁移提供了新范式:保持监督密度但约束参数变化量,有助于稳定多任务对齐。