用训练好的大模型提取渐进式教学路径,让小模型更快更准地学。
Efficient Knowledge Distillation via Curriculum Extraction
- 从训练好的教师模型中提取隐藏层投影作为教学路线。
- 在双层网络上学习稀疏奇偶性任务时,性能接近渐进式蒸馏。
- 适用于Transformer架构,显著优于传统单次蒸馏方法。
知识蒸馏通过使用大模型输出来训练小模型,具有诸多实证优势。传统一次性蒸馏仅使用最终大模型的输出,而近期研究发现,利用大模型训练过程中的中间检查点作为隐式‘教学进度’进行渐进式蒸馏可显著加速训练。然而,该方法需存储大量检查点,且中间节点选择复杂,难以在大规模训练中应用。本文提出,仅需完全训练好的教师模型即可提取出等效教学课程:通过随机投影教师模型的隐藏表示,逐步指导学生模型训练,再结合最终输出。实验表明,该方法在双层网络学习稀疏奇偶性任务时,性能接近渐进式蒸馏,显著优于一次性蒸馏,并在Transformer架构下于稀疏奇偶性与语言建模任务中均表现更优,同时提供理论保障。
原文摘要 · Abstract (English)
Knowledge distillation is a technique used to train a small student network using the output generated by a large teacher network, and has many empirical advantages~\citep{Hinton2015DistillingTK}. While the standard one-shot approach to distillation only uses the output of the final teacher network, recent work~\citep{panigrahi2024progressive} has shown that using intermediate checkpoints from the teacher's training process as an implicit ``curriculum'' for progressive distillation can significantly speed up training. However, such schemes require storing these checkpoints, and often require careful selection of the intermediate checkpoints to train on, which can be impractical for large-scale training. In this paper, we show that a curriculum can be \emph{extracted} from just the fully trained teacher network, and that this extracted curriculum can give similar efficiency benefits to those of progressive distillation. Our extraction scheme is natural; we use a random projection of the hidden representations of the teacher network to progressively train the student network, before training using the output of the full network. We show that our scheme significantly outperforms one-shot distillation and achieves a performance similar to that of progressive distillation for learning sparse parities with two-layer networks, and provide theoretical guarantees for this setting. Additionally, we show that our method outperforms one-shot distillation even when using transformer-based architectures, both for sparse-parity learning, and language modeling tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。