arXiv:2605.21699cs.LGcs.CL2026-05被引 1

解决跨分词器知识蒸馏中关键词丢失与匹配僵化问题

X-Token: Projection-Guided Cross-Tokenizer Knowledge Distillation

论文配图:X-Token: Projection-Guided Cross-Tokenizer Knowledge Distillation
图 1 · 摘自论文原文
  • 用稀疏投影矩阵动态对齐不同分词器的词表,避免错误过滤关键token
  • 在GSM8k上提升至12.89分,相比旧方法提高10.33分
  • 支持多教师蒸馏,适合需要高效迁移小模型的场景

跨分词器知识蒸馏使学生模型能从词汇表不兼容的教师模型学习。现有方法基于隐藏状态或输出概率,后者作为即插即用方案更受欢迎。但基于概率的方法要么仅使用正确词的概率,遗漏教师分布中的完整‘暗知识’,要么处理全输出分布,依赖严格的词元划分和非原则性的启发式排序。我们发现全分布、基于概率的方法存在两个关键缺陷:(i) 非常见词失效,即关键词落入未匹配子集(如在数字拆分的Qwen监督下,Llama的1100个多位数被抑制),导致GSM8k分数从12.89降至2.56;(ii) 匹配过于保守,严格一对一匹配排除了表面形式不同的近似等价词。针对这些问题,我们提出X-Token,包含两种互补损失:P-KL消除划分,通过稀疏投影矩阵W(由分词器级字符串规则初始化)将学生分布与教师对齐,解决非常见词失效;H-KL保留混合形式,放宽匹配条件,让每个学生词元与其在W下的最优教师映射对齐。两者共享W,可自然扩展至多教师。实验表明,在Llama-3.2-1B上,使用Qwen3-4B教师时,X-Token比当前SOTA GOLD高出+3.82分;使用Phi-4-Mini教师时高出+0.5分。双教师设置(Phi-4-mini + Llama-3B)比单教师蒸馏再提升+1.3分。

原文摘要 · Abstract (English)

Cross-tokenizer knowledge distillation allows a student model to learn from teachers with incompatible vocabularies. Prior work operates on hidden states or logits; the latter is preferred as a drop-in replacement requiring no auxiliary components. Logit-based methods either use only the correct-token probability, missing the full 'dark knowledge' in the teacher's distribution, or operate on the full output distribution, relying on strict token partitioning and/or unprincipled heuristic ranking. We identify two key shortcomings of full-distribution, logit-based methods: (i) an uncommon-token failure, where critical tokens fall into the unmatched subset (e.g., Llama's 1100 multi-digit numerals under digit-splitting Qwen supervision) and are suppressed during training, reducing GSM8k from 12.89 to 2.56 compared to same-tokenizer KD from a weaker teacher; and (ii) over-conservative matching, where strict 1-to-1 matching excludes near-equivalent tokens across surface forms. These failures require distinct remedies: eliminating the partition when critical tokens are misaligned, and refining it when alignment is reliable. We propose X-Token, an approach with two complementary loss formulations targeting these issues. P-KL removes partitioning and aligns the student's distribution with the teacher's via a sparse projection matrix W (initialized from tokenizer-level string rules) to address the uncommon-token failure. H-KL retains the hybrid form while relaxing matching to align each student token with its top-ranked teacher mapping under W. Both objectives share W and extend naturally to multiple teachers. Empirically, on Llama-3.2-1B, X-Token outperforms the current state of the art GOLD by +3.82 average points with a Qwen3-4B teacher and by +0.5 with a Phi-4-Mini teacher. Further, a two-teacher setup (Phi-4-mini + Llama-3B) improves over single-teacher distillation by +1.3 points.

知识蒸馏跨分词器小模型优化多教师

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