arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.02430cs.LGcs.CV

基于分类的知识蒸馏中温度的统一回顾

A Unified Revisit of Temperature in Classification-Based Knowledge Distillation

Ohio State University(俄亥俄州立大学)

Logan Frank, Jim Davis

AI总结:

本文提出了一种统一方法,系统研究了知识蒸馏中温度选择与训练组件之间的关系,为实践者提供了温度选择的指导。

摘要:

知识蒸馏的核心思想是将教师模型权重中嵌入的关联结构暴露给学生模型学习,这通常通过温度参数来实现。尽管温度参数被广泛使用,但对其如何选择合适值,或其如何依赖于其他训练元素如优化器、教师预训练/微调等,仍存在有限的理解。在实践中,温度常通过网格搜索或采用先前工作的值来选择,这可能耗时或在训练设置不同时导致学生性能不佳。在本文中,我们提出温度与这些训练组件密切相关,并提出一项统一研究,系统地考察这些交互。通过分析这些交叉连接,我们识别并呈现对温度选择有显著影响的常见情况,为从事知识蒸馏工作的实践者提供有价值的指导。

英文摘要:

A central idea of knowledge distillation is to expose relational structure embedded in the teacher's weights for the student to learn, which is often facilitated using a temperature parameter. Despite its widespread use, there remains limited understanding on how to select an appropriate temperature value, or how this value depends on other training elements such as optimizer, teacher pretraining/finetuning, etc. In practice, temperature is commonly chosen via grid search or by adopting values from prior work, which can be time-consuming or may lead to suboptimal student performance when training setups differ. In this work, we posit that temperature is closely linked to these training components and present a unified study that systematically examines such interactions. From analyzing these cross-connections, we identify and present common situations that have a pronounced impact on temperature selection, providing valuable guidance for practitioners employing knowledge distillation in their work.

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

↑