arXiv:2605.07783cs.CL2026-05被引 1

用链式蒸馏高效初始化不同大小的小模型,省去重复调用大模型。

Chain-based Distillation for Effective Initialization of Variable-Sized Small Language Models

论文配图:Chain-based Distillation for Effective Initialization of Variable-Sized Small Language Models
图 1 · 摘自论文原文
  • 构建中间模型链,分步迁移大模型知识,避免反复调用教师模型。
  • 138M小模型无需预训练,在10B词数据上表现优于从零训练的模型。
  • 支持跨架构、跨词表迁移,适合多场景小模型快速初始化。

大型语言模型性能强但部署成本高。从小模型(SLMs)从头训练计算开销大,传统知识蒸馏需为不同目标尺寸反复访问大模型教师,可扩展性差。为此,我们提出链式蒸馏(CBD),一种可扩展的变尺寸语言模型高效初始化范式。通过逐步蒸馏构建稀疏有限的中间模型序列(称作锚点),形成从源大模型逐步传递知识的蒸馏链。为支持异构设置,引入桥接蒸馏实现跨架构和跨词表的知识迁移。不同尺寸模型通过相邻锚点间参数插值初始化,消除对大教师模型的重复推理。实验表明,该方法显著提升效率与下游性能:一个138M参数的小模型(无恢复预训练)在100亿词语料上的特定任务中表现优于从零训练的模型。CBD在异构设置下也表现出多样性,可用于初始化不同架构与词表的模型。

原文摘要 · Abstract (English)

Large language models (LLMs) achieve strong performance but remain costly to deploy in resource-constrained settings. Training small language models (SLMs) from scratch is computationally expensive, while conventional knowledge distillation requires repeated access to large teachers for different target sizes, leading to poor scalability. To solve these problems, we propose \textbf{Chain-based Distillation (CBD)}, a scalable paradigm for efficiently initializing variable-sized language models. A sparse and limited sequence of intermediate models (called anchors) is constructed via stepwise distillation, forming a distillation chain that progressively transfers knowledge from the source LLMs. To support heterogeneous settings, we introduce \emph{bridge distillation} for cross-architecture and cross-vocabulary transfer. Models of variable sizes are initialized via parameter interpolation between adjacent anchors, eliminating repeated large teacher inference. Experiments show that the proposed method substantially improves efficiency and downstream performance. A 138M-parameter SLM without recovery pre-training, outperforms scratch-trained models on a 10B-token corpus on the specific task. CBD also demonstrates versatility in heterogeneous settings for initialize models with different architectures and vocabularies.

小模型知识蒸馏链式方法高效初始化

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