arXiv:2503.20083cs.CL2025-03NeurIPS被引 41

让不同分词器的大小模型也能互相教,突破了传统知识蒸馏的限制。

Universal Cross-Tokenizer Distillation via Approximate Likelihood Matching

  • 通过近似似然匹配实现跨分词器的知识迁移,无需分词器一致
  • 在数学模型蒸馏中达到与原大模型相当的解题能力
  • 适用于模型融合、分词器转换等新场景,适合模型轻量化研究者

知识蒸馏在将大型语言模型(LLM)的知识迁移到小型学生模型方面表现卓越。然而,现有方法要求教师与学生使用相似的分词器,仅适用于少数教师-学生组合。本文提出一种原理性的跨分词器蒸馏方法,首次实现根本性不同的分词器间的有效蒸馏,并在所有其他情况下显著优于以往方法。我们在三个应用场景中验证了该方法的有效性:首先,将分词器迁移视为自蒸馏,实现了前所未有的跨分词器迁移效果,包括子词模型到字节级模型的快速迁移;相同分词器的不同模型可进行集成以提升性能。其次,将一个大型数学专用模型蒸馏到使用不同分词器的小型通用模型中,实现了具有竞争力的数学问题求解性能。第三,利用本方法训练出当前最优的嵌入预测超网络,用于无训练的分词器迁移。结果拓展了蒸馏可用的教师-学生配对范围,开启了模型间交互的新方式。

原文摘要 · Abstract (English)

Distillation has shown remarkable success in transferring knowledge from a Large Language Model (LLM) teacher to a student LLM. However, current distillation methods require similar tokenizers between the teacher and the student, restricting their applicability to only a small subset of teacher-student pairs. In this work, we develop a principled cross-tokenizer distillation method to solve this crucial deficiency. Our method is the first to enable effective distillation across fundamentally different tokenizers, while also substantially outperforming prior methods in all other cases. We verify the efficacy of our method on three distinct use cases. First, we show that viewing tokenizer transfer as self-distillation enables unprecedentedly effective transfer across tokenizers, including rapid transfer of subword models to the byte-level. Transferring different models to the same tokenizer also enables ensembling to boost performance. Secondly, we distil a large maths-specialised LLM into a small general-purpose model with a different tokenizer, achieving competitive maths problem-solving performance. Thirdly, we use our method to train state-of-the-art embedding prediction hypernetworks for training-free tokenizer transfer. Our results unlock an expanded range of teacher-student pairs for distillation, enabling new ways to adapt and enhance interaction between LLMs.

知识蒸馏跨分词器大模型压缩数学推理

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