Long chain-of-thought (CoT) trajectories provide rich supervision signals for distilling reasoning from teacher to student LLMs. However, both prior work and our experiments show that trajectories from stronger teachers do not necessarily yield better students, highlighting the importance of data-student suitability in distillation. Existing methods assess suitability primarily through student likelihood, favoring trajectories that closely align with the model's current behavior but overlooking more informative ones. Addressing this, we propose Rank-Surprisal Ratio (RSR), a simple metric that captures both alignment and informativeness to assess the suitability of a reasoning trajectory. RSR is motivated by the observation that effective trajectories typically combine low absolute probability with relatively high-ranked tokens under the student model, balancing learning signal strength and behavioral alignment. Concretely, RSR is defined as the ratio of a trajectory's average token-wise rank to its average negative log-likelihood, and is straightforward to compute and interpret. Across five student models and reasoning trajectories from 11 diverse teachers, RSR strongly correlates with post-training performance (average Spearman 0.86), outperforming existing metrics. We further demonstrate its practical utility in both trajectory selection and teacher selection.
翻译:长链思维(CoT)轨迹为从教师大型语言模型向学生模型蒸馏推理能力提供了丰富的监督信号。然而,先前研究及我们的实验均表明,来自更强教师的轨迹未必能培养出更优秀的学生,这凸显了数据-学生适配性在蒸馏过程中的重要性。现有方法主要通过学生似然度评估适配性,倾向于选择与模型当前行为高度对齐的轨迹,却忽略了更具信息量的轨迹。针对这一问题,我们提出秩-惊异比(RSR),这是一个同时捕捉对齐性和信息量的简单指标,用于评估推理轨迹的适配性。RSR的提出基于以下观察:有效轨迹通常在学生模型下兼具较低绝对概率与相对较高排名的词元,从而平衡学习信号强度与行为对齐。具体而言,RSR定义为轨迹的平均词元秩与平均负对数似然之比,其计算和解释均十分直观。在五个学生模型和来自11位不同教师的推理轨迹上,RSR与训练后性能表现出强相关性(平均斯皮尔曼系数0.86),优于现有指标。我们进一步展示了其在轨迹选择和教师选择两个场景中的实际应用价值。