通过动态调整师生模型分布,实现高效知识迁移,让小模型更接近大模型表现。
TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Models
- 设计自适应中间分布,逐步融合师生模型输出,缓解能力差距与模式崩溃问题。
- 在不同规模模型上验证,小模型性能显著提升,尤其在指令微调和预训练中效果突出。
- 已用于打造两款顶尖小型基础模型,适合资源受限场景下的高效AI部署。
因果语言模型展现出强大能力,但其庞大体积给资源受限环境的部署带来挑战。知识蒸馏是将大模型知识迁移到小模型的常用方法,但仍面临师生模型间显著的能力差距、模式平均和模式崩溃等问题。为此,我们提出一种新型知识蒸馏方法——时序自适应插值蒸馏(TAID),通过自适应中间分布动态插值学生与教师模型的输出分布,从学生初始分布逐步过渡到教师分布。理论分析表明,该方法可有效防止模式崩溃;实验验证其能有效缓解能力差距,并平衡模式平均与模式崩溃。大规模实验显示,TAID在多种模型规模与架构下均表现出色,适用于指令微调与预训练任务。此外,我们基于TAID构建了两款当前最先进的紧凑型基础模型:用于语言任务的TAID-LLM-1.5B 和用于视觉语言任务的TAID-VLM-2B,证明了其在构建高性能、高效率模型方面的有效性,推动更普惠AI技术的发展。
原文摘要 · Abstract (English)
Causal language models have demonstrated remarkable capabilities, but their size poses significant challenges for deployment in resource-constrained environments. Knowledge distillation, a widely-used technique for transferring knowledge from a large teacher model to a small student model, presents a promising approach for model compression. A significant remaining issue lies in the major differences between teacher and student models, namely the substantial capacity gap, mode averaging, and mode collapse, which pose barriers during distillation. To address these issues, we introduce $\textit{Temporally Adaptive Interpolated Distillation (TAID)}$, a novel knowledge distillation approach that dynamically interpolates student and teacher distributions through an adaptive intermediate distribution, gradually shifting from the student's initial distribution towards the teacher's distribution. We provide a theoretical analysis demonstrating TAID's ability to prevent mode collapse and empirically show its effectiveness in addressing the capacity gap while balancing mode averaging and mode collapse. Our comprehensive experiments demonstrate TAID's superior performance across various model sizes and architectures in both instruction tuning and pre-training scenarios. Furthermore, we showcase TAID's practical impact by developing two state-of-the-art compact foundation models: $\texttt{TAID-LLM-1.5B}$ for language tasks and $\texttt{TAID-VLM-2B}$ for vision-language tasks. These results demonstrate TAID's effectiveness in creating high-performing and efficient models, advancing the development of more accessible AI technologies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。