Densely Guided Knowledge Distillation using Multiple Teacher Assistants
This work addresses a specific bottleneck in model compression and transfer learning for deep neural networks, offering an incremental improvement over existing knowledge distillation techniques.
The paper tackles the problem of poor learning in knowledge distillation when there is a large size gap between teacher and student networks by proposing a densely guided method using multiple teacher assistants and stochastic teaching, achieving significant performance improvements on CIFAR-10, CIFAR-100, and ImageNet datasets with various architectures like ResNet and VGG.
With the success of deep neural networks, knowledge distillation which guides the learning of a small student network from a large teacher network is being actively studied for model compression and transfer learning. However, few studies have been performed to resolve the poor learning issue of the student network when the student and teacher model sizes significantly differ. In this paper, we propose a densely guided knowledge distillation using multiple teacher assistants that gradually decreases the model size to efficiently bridge the large gap between the teacher and student networks. To stimulate more efficient learning of the student network, we guide each teacher assistant to every other smaller teacher assistants iteratively. Specifically, when teaching a smaller teacher assistant at the next step, the existing larger teacher assistants from the previous step are used as well as the teacher network. Moreover, we design stochastic teaching where, for each mini-batch, a teacher or teacher assistants are randomly dropped. This acts as a regularizer to improve the efficiency of teaching of the student network. Thus, the student can always learn salient distilled knowledge from the multiple sources. We verified the effectiveness of the proposed method for a classification task using CIFAR-10, CIFAR-100, and ImageNet. We also achieved significant performance improvements with various backbone architectures such as ResNet, WideResNet, and VGG.