arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.14922cs.LGcs.AI
收藏

面向预训练模型的知识蒸馏实用见解

Practical Insights into Knowledge Distillation for Pre-Trained Models

Queen Mary University of London(伦敦女王学院)

Norah Alballa, Ahmed M. Abdelmoniem, Marco Canini

AI总结:

本文系统比较多种知识蒸馏方法在预训练模型与联邦学习场景中的效果,揭示不同数据划分下的最优超参数设置,并提出利用KD减少通信轮次、加速训练的实用框架。

摘要:

本研究探讨了预训练模型中知识蒸馏(KD)过程的增强,这是知识迁移领域的一个新兴方向,对分布式训练和联邦学习环境具有重要意义。这些环境受益于通信需求的降低,并能够适应多种模型架构。尽管已有多种KD方法被用于在预训练模型之间迁移知识,但目前仍缺乏对KD在这些场景中应用的全面理解。我们的研究对多种KD技术进行了广泛比较,包括标准KD、调优KD(通过优化温度和权重参数)、深度互学习和数据划分KD。我们在不同的数据分布策略下评估这些方法,以确定每种方法最有效的应用场景。通过对超参数调优的详细考察,并结合广泛的网格搜索评估,我们明确了何时需要进行调整以提升模型性能。本文揭示了不同数据划分场景下的最优超参数设置,并研究了KD在通过减少通信轮次和加速训练过程来改进联邦学习方面的作用。通过填补当前研究中的一个显著空白,我们的研究结果为在协作学习和联邦学习框架中利用预训练模型的KD提供了实用框架。

英文摘要:

This research investigates the enhancement of knowledge distillation (KD) processes in pre-trained models, an emerging field in knowledge transfer with significant implications for distributed training and federated learning environments. These environments benefit from reduced communication demands and accommodate various model architectures. Despite the adoption of numerous KD approaches for transferring knowledge among pre-trained models, a comprehensive understanding of KD's application in these scenarios is lacking. Our study conducts an extensive comparison of multiple KD techniques, including standard KD, tuned KD (via optimized temperature and weight parameters), deep mutual learning, and data partitioning KD. We assess these methods across various data distribution strategies to identify the most effective contexts for each. Through detailed examination of hyperparameter tuning, informed by extensive grid search evaluations, we pinpoint when adjustments are crucial to enhance model performance. This paper sheds light on optimal hyperparameter settings for distinct data partitioning scenarios and investigates KD's role in improving federated learning by minimizing communication rounds and expediting the training process. By filling a notable void in current research, our findings serve as a practical framework for leveraging KD in pre-trained models within collaborative and federated learning frameworks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑