关键词: 人工智能, 知识蒸馏, 特征图自适应, 模型迁移, 图像分类

Abstract: Aiming at the problem that computational and storage resources of embedded and mobile devices were limited, and the compact  network optimization was easy to  converge to poor local optimal solutions,  we proposed an  activation map adaptation model for knowledge distillation,  which was composed  of an activation map adapter and an activation map adaptation knowledge distillation strategy. Firstly, the activation map adapter realized activation map size matching, synchronous transformation of teacher-student network features, and adaptive semantic information matching by heterogeneous convolution and stacking of visual feature expression modules. Secondly, the activation map adaptation knowledge distillation strategy embedded the adapter into the teacher network to reconstruct it, and realized adaptively search suitable for the  supervision features of the hidden layer of the student network  during training process,  the front output  of the adapter was used to prompt the front training of the 
student network, so as to realize knowledge transfer from the teacher to the student network, and further optimize it  under the constraint of learning rate. Finally, experimental verification was carried out on the image classification task  dataset cifar-10. The results show  that the classification accuracy of the activation map adaptive knowledge distillation model is improved by 0.6%, the inference loss is reduced by 6.5%, and  the time to converge to 78.2% accuracy is reduced to 5.6% when it is not migrated.

Key words: artificial intelligence, knowledge distillation, activation map adaptation, model transfer, image classification