PKD: General Distillation Framework for Object Detectors via Pearson Correlation Coefficient
Weihan, Cao1,3; Yifan, Zhang3; Jianfei, Gao2; Anda, Cheng1,3; Ke, Cheng1,3; Jian, Cheng3
2022
会议日期Monday November 28th through Friday December 9th
会议地点New Orleans, America
关键词Knowledge Distillation Model Compression Object Detection
卷号35
页码15394-15406
英文摘要

Knowledge distillation(KD) is a widely-used technique to train compact models in object detection. However, there is still a lack of study on how to distill between heterogeneous detectors. In this paper, we empirically find that better FPN features from a heterogeneous teacher detector can help the student although their detection heads and label assignments are different. However, directly aligning the feature maps to distill detectors suffers from two problems. First, the difference in feature magnitude between the teacher and the student could enforce overly strict constraints on the student. Second, the FPN stages and channels with large feature magnitude from the teacher model could dominate the gradient of distillation loss, which will overwhelm the effects of other features in KD and introduce much noise. To address the above issues, we propose to imitate features with Pearson Correlation Coefficient to focus on the relational information from the teacher and relax constraints on the magnitude of the features. Our method consistently outperforms the existing detection KD methods and works for both homogeneous and heterogeneous student-teacher pairs. Furthermore, it converges faster. With a powerful MaskRCNN-Swin detector as the teacher, ResNet-50 based RetinaNet and FCOS achieve 41.5% and 43.9% mAP on COCO2017, which are 4.1% and 4.8% higher than the baseline, respectively.

语种英语
URL标识查看原文
内容类型会议论文
源URL[http://ir.ia.ac.cn/handle/173211/52086]  
专题类脑芯片与系统研究
通讯作者Yifan, Zhang
作者单位1.School of Artificial Intelligence, University of Chinese Academy of Sciences
2.Shanghai AI Laboratory
3.NLPR & AIRIA, Institute of Automation, Chinese Academy of Sciences
推荐引用方式
GB/T 7714
Weihan, Cao,Yifan, Zhang,Jianfei, Gao,et al. PKD: General Distillation Framework for Object Detectors via Pearson Correlation Coefficient[C]. 见:. New Orleans, America. Monday November 28th through Friday December 9th.
个性服务
查看访问统计
相关权益政策
暂无数据
收藏/分享
所有评论 (0)
暂无评论
 

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。


©版权所有 ©2017 CSpace - Powered by CSpace