CompassOPD:跨家族模型蒸馏新方法
CompassOPD: Cross-Family On-Policy Distillation via Within-Family Likelihood Shifts
研究人员提出CompassOPD解决跨家族模型蒸馏效果差的问题,比标准方法提升推理准确率5.5点。
CompassOPD是一种跨家族模型蒸馏方法,通过消除模型间偏移量,专注于家族内部似然转移。该方法在三个学生家族和多个教师家族的实验中,平均推理准确率提升高达5.50个百分点。对于MoE教师模型,通过减少专家激活直接构建参考点,无需单独参考检查点即可实现3.43百分点的性能提升。
CompassOPD: Cross-Family On-Policy Distillation via Within-Family Likelihood Shifts
On-policy distillation (OPD) provides dense token-level supervision on student-generated trajectories. Although OPD performs strongly when teacher and student belong to the same model family, we find that its effectiveness degrades in cross-family settings even after tokenizer alignment, with substantially stronger external teachers offering little additional improvement. To understand this disconnect, we decompose the cross-family OPD signal into two components: an offset between a low-capability teacher-family reference and the student, and the within-family log-likelihood shift from that reference to the strong teacher. Standard OPD transfers both components together, allowing the offset to dominate the update direction and obscure the changes associated with teacher capability improvements. We propose CompassOPD, which removes this offset and transfers the within-family shift, while a frozen student reference anchors updates to the student's initial policy. Thus, both teacher-side and student-side changes are measured within their respective model families. Experiments across three student families and multiple teacher families show that CompassOPD consistently outperforms standard cross-family OPD, improving average reasoning accuracy by up to 5.50 points. For an MoE teacher, we further construct the reference directly from the teacher checkpoint by reducing expert activation, eliminating the need for a separate reference checkpoint while retaining a 3.43-point gain over OPD.