Few-Shot Object Detection with Model Calibration

被引:11
作者
Fan, Qi [1 ]
Tang, Chi-Keung [1 ]
Tai, Yu-Wing [1 ,2 ]
机构
[1] Hong Kong Univ Sci & Technol, Clear Water Bay, Hong Kong, Peoples R China
[2] Kuaishou Technol, Beijing, Peoples R China
来源
COMPUTER VISION, ECCV 2022, PT XIX | 2022年 / 13679卷
关键词
Few-shot object detection; Model bias; Model calibration; Uncertainty-aware RPN; Detector calibration;
D O I
10.1007/978-3-031-19800-7_42
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Few-shot object detection (FSOD) targets at transferring knowledge from known to unknown classes to detect objects of novel classes. However, previous works ignore the model bias problem inherent in the transfer learning paradigm. Such model bias causes overfitting toward the training classes and destructs the well-learned transferable knowledge. In this paper, we pinpoint and comprehensively investigate the model bias problem in FSOD models and propose a simple yet effective method to address the model bias problem with the facilitation of model calibrations in three levels: 1) Backbone calibration to preserve the well-learned prior knowledge and relieve the model bias toward base classes, 2) RPN calibration to rescue unlabeled objects of novel classes and, 3) Detector calibration to prevent the model bias toward a few training samples for novel classes. Specifically, we leverage the overlooked classification dataset to facilitate our model calibration procedure, which has only been used for pre-training in other related works. We validate the effectiveness of our model calibration method on the popular Pascal VOC and MS COCO datasets, where our method achieves very promising performance. Codes are released at https://github.com/fanq15/FewX.
引用
收藏
页码:720 / 739
页数:20
相关论文
共 106 条
[1]  
Allen KR, 2019, PR MACH LEARN RES, V97
[2]  
[Anonymous], 2010, International journal of computer vision, DOI DOI 10.1007/s11263-009-0275-4
[3]  
Antoniou A., 2019, P 9 BALK C INF SOF B, DOI DOI 10.1145/3351556.3351574
[4]  
Bertinetto L., 2019, INT C LEARN REPR
[5]   Cascade R-CNN: Delving into High Quality Object Detection [J].
Cai, Zhaowei ;
Vasconcelos, Nuno .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6154-6162
[6]   A Unified Multi-scale Deep Convolutional Neural Network for Fast Object Detection [J].
Cai, Zhaowei ;
Fan, Quanfu ;
Feris, Rogerio S. ;
Vasconcelos, Nuno .
COMPUTER VISION - ECCV 2016, PT IV, 2016, 9908 :354-370
[7]  
Cao Y., 2021, NeurIPS
[8]  
Chen HC, 2018, AAAI CONF ARTIF INTE, P2127
[9]  
Chen W.Y., 2019, INT C LEARNING REPRE
[10]   Class-Balanced Loss Based on Effective Number of Samples [J].
Cui, Yin ;
Jia, Menglin ;
Lin, Tsung-Yi ;
Song, Yang ;
Belongie, Serge .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :9260-9269