arXiv:2607.23731cs.LG2026-07

发现强化学习中模仿学习的局部监督存在结果混淆问题

Outcome-Confounded Local Supervision in On-Policy Distillation

论文配图:Outcome-Confounded Local Supervision in On-Policy Distillation
图 1 · 摘自论文原文
  • 通过轨迹结果分离出四种模仿状态:安全模仿、有效分歧、有害分歧和失败共识
  • 在数学推理任务中,67.8%的输出词元处于错误轨迹上的共识状态
  • 现有方法无法消除失败共识,需引入过程标签或对齐信息改进

在线策略蒸馏(OPD)让学生模型在自身生成轨迹上训练,教师提供访问前缀处的密集词元级似然。这些似然通常被局部解读:一致视为安全模仿,不一致则视为错误信号。我们发现这两种解读均受完整轨迹结果的干扰。为此提出一种结果解析诊断方法,将逐点师生差异与最终答案正确性交叉分析,区分出安全模仿、有效分歧、有害分歧和‘错误共识’四类状态。在八次种子的数学推理实验中,使用Qwen3-8B学生和Qwen3-32B教师时,错误共识占全部响应词元质量的67.84%;采用Qwen2.5-7B/32B组合时仍达67.68%。该现象在阈值、序列级别、格式及截断审计下均稳定存在。即使教师在四个独立尝试中均正确解答同一提示,学生准确率仅提升至86.91%,但错误共识仍占14.76%。进一步进行三组匹配训练探针实验,分别采用模仿、掩码或对比整条轨迹的策略,均未持续降低错误共识。结果表明,仅凭局部差异与轨迹结果无法定位失败轨迹不可挽回的起点。解决此问题需额外位置信息,如过程标签、教师从学生前缀的续写或跨滚动的词元对齐。本研究贡献在于诊断框架而非新训练方法。

原文摘要 · Abstract (English)

On-policy distillation (OPD) trains a student on its own trajectories while a teacher supplies dense token-level likelihoods at student-visited prefixes. These likelihoods are often read locally: agreement appears safe to imitate, whereas disagreement appears to identify an error. We show that both readings are confounded by the outcome of the completed trajectory. We introduce an outcome-resolved diagnostic that crosses pointwise teacher-student divergence with final-answer correctness, separating safe imitation, productive divergence, harmful divergence, and agreement-on-failure. In an eight-seed mathematical-reasoning study with a Qwen3-8B student and Qwen3-32B teacher, agreement-on-failure constitutes 67.84% of pooled response-token mass; with a Qwen2.5-7B/32B pair it remains 67.68%. The result persists across threshold, sequence-level, format, and truncation audits. Even on prompts that the Qwen3 teacher solves in all four independent attempts, student accuracy rises to 86.91% but agreement-on-failure remains 14.76%. We then run three matched training probes that use the available signals to imitate, mask, or contrast whole trajectories; none consistently reduces agreement-on-failure. The result points to a localization limitation: local divergence paired with a trajectory-level outcome does not identify where a failed trajectory became unrecoverable. Addressing this limitation requires additional positional information, such as process labels, teacher continuations from student prefixes, or token-level alignment across rollouts. Our contribution is therefore diagnostic rather than a new training method.

强化学习知识蒸馏偏差诊断语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。