PKD: General Distillation Framework for Object Detectors via Pearson Correlation Coefficient
文献类型:会议论文
作者 | Weihan, Cao2,3![]() ![]() ![]() ![]() |
出版日期 | 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收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。