KNOWLEDGE DISTILLATION USING GANS FOR FAST OBJECT DETECTION
- 1Federal State Unitary Enterprise «State Research Institute of Aviation Systems», Russian Federation
- 2National Research Nuclear University “MEPhI”, Russian Federation
Keywords: CNN, Object detection, Knowledge distillation, Real-time, GAN, Single shot detector
Abstract. In this paper, we propose a new method for knowledge distilling based on generative adversarial networks. Discriminator CNNs is used as an adaptive knowledge distilling loss. In experiments, single shot multibox detector SSD based on MobileNet v2 and ShuffleNet v1 are used as student networks. Our tests showed AP and mAP improvement of more than 3% on PascalVOC and 1% on MS Coco datasets compared with the baseline algorithm without any architecture or dataset changes. The proposed approach is general and can be used not only with SSD but also with any type of object detection algorithms.