arXiv:2506.05695cs.CLcs.AI2025-06被引 3

通过渐进式训练提升大模型蒸馏效果,避免学生模型遗忘和崩溃。

Being Strong Progressively! Enhancing Knowledge Distillation of Large Language Models through a Curriculum Learning Framework

  • 按难易程度排序样本,从简单到复杂逐步训练。
  • 在不同阶段使用递增温度的损失函数,提升学习稳定性。
  • 适用于各类大模型蒸馏方法,提升性能且计算开销小。

知识蒸馏(KD)通过将大语言模型(LLM)的知识迁移到更小的学生模型中,降低推理成本与内存占用,同时保持性能。然而,现有针对LLM的KD方法常导致学生模型分布发生显著偏移,引发灾难性遗忘、模式坍缩及训练-推理不匹配等问题。为此,我们提出一种受‘渐进超载’原理启发的新型即插即用式课程学习框架(POCL),可无缝集成至现有白盒蒸馏方法,计算开销极低。该框架包含两个核心组件:(1) 难度度量器,用于对训练样本按难易程度排序并分组;(2) 训练调度器,按固定周期逐步引入这些子集,并采用温度逐级升高的损失函数。通过从最简单样本开始,逐步增加难度,该方法显著提升了学习的稳定性和效率。大量指令跟随场景下的实验表明,POCL在多种白盒蒸馏方法与模型家族上均能持续提升学生模型性能。研究揭示了有序样本训练在LLM蒸馏中的有效性,更广泛地展示了如何通过结构化数据组织增强蒸馏模型的稳定性和表现。

原文摘要 · Abstract (English)

Knowledge Distillation (KD) compresses large language models (LLMs) by transferring the teacher model's capabilities to a smaller student model, reducing inference cost and memory usage while maintaining performance. However, existing KD methods for LLMs often fail to prevent significant shifts in the student model's distribution during training, leading to issues such as catastrophic forgetting, mode collapse, and training-inference mismatch. To address these challenges, we propose a novel, plug-in curriculum learning framework inspired by the strength training principle of "progressive overload" (POCL), which can be seamlessly integrated into existing white-box KD approaches with minimal computational overhead. The framework comprises two core components: (1) a difficulty measurer that ranks and partitions training samples from easy to hard, and (2) a training scheduler that incrementally introduces these subsets into the distillation process at fixed intervals while applying loss functions with progressively rising temperatures. By starting with the easiest samples and progressively increasing the difficulty, the approach enhances both the stability and efficiency of learning. Extensive experiments in instruction-following settings demonstrate that POCL consistently improves the performance of distilled student models across various white-box KD methods and model families. Our findings highlight the effectiveness of sorted training samples in KD for LLMs. More generally, our work demonstrates how to structure training data within the KD process to enhance the stability and performance of distilled LLMs.

知识蒸馏大模型压缩课程学习

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