Dec, 2018

利用特征映射进行知识蒸馏的图像分类

TL;DR本文提出了一种名为 KDFM(含有特征映射的知识蒸馏)的方法,通过学习教师网络的功能映射来提高知识蒸馏的有效性,并在实验中得出使用 4 层 CNN 模仿 DenseNet-40 和使用 MobileNet 模仿 DenseNet-100,对于 CIFAR-100 数据集,学生网络与教师模型相比准确率损失小于 1%,并且学生网络的推理速度是教师模型的 2-6 倍,而 MobileNet 的模型大小小于 DenseNet-100 的一半。