中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
PKD: General Distillation Framework for Object Detectors via Pearson Correlation Coefficient

文献类型:会议论文

作者Weihan, Cao2,3; Yifan, Zhang3; Jianfei, Gao1; Anda, Cheng2,3; Ke, Cheng2,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.Shanghai AI Laboratory
2.School of Artificial Intelligence, University of Chinese Academy of Sciences
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.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

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