Knowledge Distillation Based on Narrow-Deep Networks

Yan Zhou,Zhiqiang Wang,Jianxun Li
DOI: https://doi.org/10.1007/s11063-024-11646-5
IF: 2.565
2024-06-08
Neural Processing Letters
Abstract:Deep neural networks perform better than shallow neural networks, but the former tends to be deeper or wider, introducing large numbers of parameters and computations. We know that networks that are too wide have a high risk of overfitting and networks that are too deep require a large amount of computation. This paper proposed a narrow-deep ResNet, increasing the depth of the network while avoiding other issues caused by making the network too wide, and used the strategy of knowledge distillation, where we set up a trained teacher model to train an unmodified, wide, and narrow-deep ResNet that allows students to learn the teacher's output. To validate the effectiveness of this method, it is tested on Cifar-100 and Pascal VOC datasets. The method proposed in this paper allows a small model to have about the same accuracy rate as a large model, while dramatically shrinking the response time and computational effort.
computer science, artificial intelligence
What problem does this paper attempt to address?