Knowledge distillation for object detection based on Inconsistency-based Feature Imitation and Global Relation Imitation

Peng Ju,Yi Zhang

NEUROCOMPUTING(2024)

引用 0|浏览14
暂无评分
摘要
Knowledge distillation (KD) is a method that transfers information from a larger network (teacher) to a smaller network (student) to obtain stronger performance without extra computational load. It has made great success in image classification, but only attains trivial improvement in object detection. In this paper, we propose a knowledge distillation scheme for object detection with Inconsistency-Based Feature Imitation (IBFI) and Global Relation Imitation (GRI) schemes. IBFI calculates the difference between the outputs of classification head and regression head to balance the classification and localization abilities of the detector. GRI enables the student to mimic the teacher's relational information. Extensive experiments have been conducted on popular datasets including MS COCO and PASCAL VOC to validate the effectiveness of our scheme. We achieve 39.9% mAP on MS COCO by using RetinaNet + ResNet-50, which surpasses the baseline by 2.5%.
更多
查看译文
关键词
Object detection,Knowledge distillation,Feature imitation,Global Relation Imitation
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要