General Instance Distillation for Object Detection

Xing Dai, Zeren Jiang, Si Liu, Erjin Zhou, Zhicheng Wang, Yiping Bao, Zhao Wu
2021-06-01

SCID:  54.1/zajuaanm
In recent years, knowledge distillation has been proved to be an effective solution for model compression. This approach can make lightweight student models acquire the knowledge extracted from cumbersome teacher models. However, previous distillation methods of detection have weak generalization for different detection frameworks and rely heavily on ground truth (GT), ignoring the valuable relation information between instances. Thus, we propose a novel distillation method for detection tasks based on discriminative instances without considering the positive or negative distinguished by GT, which is called general instance distillation (GID). Our approach contains a general instance selection module (GISM) to make full use of feature-based, relation-based and response-based knowledge for distillation. Extensive results demonstrate that the student model achieves significant AP improvement and even outperforms the teacher in various detection frame-works. Specifically, RetinaNet with ResNet-50 achieves 39.1% in mAP with GID on COCO dataset, which surpasses the baseline 36.2% by 2.9%, and even better than the ResNet-101 based teacher model with 38.1% AP.
Publication Details
Publication Date
2021-06-01
Journal
Publisher
ISSN
Access Type
Author Information
Authors
Xing Dai
Zeren Jiang
Si Liu
Erjin Zhou
Zhicheng Wang
Yiping Bao
Zhao Wu
Explore More Research
Use the citation graph to discover related papers and expand your research horizons.
Click any node to explore
Download PDF
100%