arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1907.09643cs.CVcs.LG

凸显每一步:基于协同教学的知识蒸馏

Highlight Every Step: Knowledge Distillation via Collaborative Teaching

Ocean University of China(中国海洋大学)

Haoran Zhao, Xin Sun, Junyu Dong, Changrui Chen, Zihe Dong

AI总结:

针对现有知识蒸馏忽略训练过程信息的问题,提出采用从零训练教师与专家教师协同教学的CTKD策略,结合过程引导与关键区域指导,在多个数据集上取得SOTA性能。

摘要:

高存储与计算成本阻碍了深度神经网络在资源受限设备上的部署。知识蒸馏旨在通过从更大的预训练教师模型中迁移知识,来训练一个紧凑的学生网络。然而,现有大多数知识蒸馏方法都忽略了训练过程中与训练结果相关的宝贵信息。本文提出了一种全新的协同教学知识蒸馏(Collaborative Teaching Knowledge Distillation,CTKD)策略,该策略采用两位特殊的教师。具体而言,一位从零开始训练的教师(即从零训练教师)利用其临时输出逐步辅助学生,促使学生以高精度朝着最终logits的最优路径逼近。另一位预训练教师(即专家教师)则引导学生聚焦于对任务更有用的关键区域。来自这两位特殊教师的知识相结合,能够显著提升知识蒸馏中学生网络的性能。在CIFAR-10、CIFAR-100、SVHN和Tiny ImageNet数据集上的实验结果验证了所提知识蒸馏方法的有效性,且达到了当前最优性能。

英文摘要:

High storage and computational costs obstruct deep neural networks to be deployed on resource-constrained devices. Knowledge distillation aims to train a compact student network by transferring knowledge from a larger pre-trained teacher model. However, most existing methods on knowledge distillation ignore the valuable information among training process associated with training results. In this paper, we provide a new Collaborative Teaching Knowledge Distillation (CTKD) strategy which employs two special teachers. Specifically, one teacher trained from scratch (i.e., scratch teacher) assists the student step by step using its temporary outputs. It forces the student to approach the optimal path towards the final logits with high accuracy. The other pre-trained teacher (i.e., expert teacher) guides the student to focus on a critical region which is more useful for the task. The combination of the knowledge from two special teachers can significantly improve the performance of the student network in knowledge distillation. The results of experiments on CIFAR-10, CIFAR-100, SVHN and Tiny ImageNet datasets verify that the proposed knowledge distillation method is efficient and achieves state-of-the-art performance.

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

↑