当 EOS Token 不一致:理解在线策略蒸馏中的长度膨胀

HuggingFace Daily Papers(社区热门论文)·2026-09-17 08:00·1天前
AI 导读

研究发现在线策略蒸馏(OPD)中,基础学生模型与后训练教师模型即使声明的停止集合相同,也可能将停止概率放在不同的 EOS token 上,这种终止 token 不匹配会抑制学生模型偏好的终止动作,导致回复过长甚至耗尽生成预算。

HuggingFace Daily Papers(社区热门论文)
44AI 编辑部评分,满分 100

当 EOS Token 不一致:理解在线策略蒸馏中的长度膨胀

2026-09-17 08:00· 1天前
AI 导读

研究发现在线策略蒸馏(OPD)中,基础学生模型与后训练教师模型即使声明的停止集合相同,也可能将停止概率放在不同的 EOS token 上,这种终止 token 不匹配会抑制学生模型偏好的终止动作,导致回复过长甚至耗尽生成预算。

We study length inflation in on-policy distillation (OPD), where student responses can become excessively long and even exhaust the generation budget. We identify termination-token mismatch between base students and post-trained teachers as an important source of this behavior. Across Qwen3, Llama, and Gemma, the two models can place their stopping probability on different EOS tokens, even when their declared stopping sets are identical. This mismatch can suppress the student's preferred termination action without reliably transferring the teacher-preferred alternative.

We show that aligning the decoding stopping set alone is insufficient, while treating functionally equivalent EOS tokens as a shared semantic stopping action substantially mitigates mismatch-induced length inflation across all three model families. To further understand how termination behavior evolves over training, we study OPD across different K2-Horizon training stages. This stage-wise analysis shows that termination preferences can shift substantially during training, while also revealing a distinct length inflation late in the OPD run that persists beyond termination alignment.

Together, these results identify termination mismatch as an important, but not exhaustive, source of OPD length dynamics. We release an implementation incorporating the proposed termination-handling corrections.

来源:HuggingFace Daily Papers(社区热门论文)· arxiv.org