arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.00299cs.CVcs.AIcs.LG
收藏

学生是最好的老师:多出口出-增蒸馏

Students are the Best Teacher: Exit-Ensemble Distillation with Multi-Exits

Yonsei University(延世大学)

Hojung Lee, Jong-Seok Lee

摘要:

本文提出了一种新的基于知识蒸馏的学习方法,以提高卷积神经网络(CNN)的分类性能,而无需预先训练的教师网络,称为出口集成蒸馏。我们的方法利用多出口架构,在传统的CNN中间添加辅助分类器(称为出口),通过它可以获得早期的推理结果。我们的方法的思想是训练网络使用出口的集合作为蒸馏目标,这大大提高了整个网络的分类性能。我们的方法提出了一个新的范式知识蒸馏,不像传统的蒸馏概念,教师只教学生,我们表明,学生也可以帮助其他学生,甚至教师学习得更好。实验结果表明,我们的方法在各种流行的CNN架构(VGG,ResNet,ResNeXt,WideResNet等)上实现了分类性能的显着改善。此外,所提出的方法可以加快学习的收敛速度,提高稳定性。我们的代码将在Github上提供。

英文摘要:

This paper proposes a novel knowledge distillation-based learning method to improve the classification performance of convolutional neural networks (CNNs) without a pre-trained teacher network, called exit-ensemble distillation. Our method exploits the multi-exit architecture that adds auxiliary classifiers (called exits) in the middle of a conventional CNN, through which early inference results can be obtained. The idea of our method is to train the network using the ensemble of the exits as the distillation target, which greatly improves the classification performance of the overall network. Our method suggests a new paradigm of knowledge distillation; unlike the conventional notion of distillation where teachers only teach students, we show that students can also help other students and even the teacher to learn better. Experimental results demonstrate that our method achieves significant improvement of classification performance on various popular CNN architectures (VGG, ResNet, ResNeXt, WideResNet, etc.). Furthermore, the proposed method can expedite the convergence of learning with improved stability. Our code will be available on Github.

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

↑