Few-Shot Object Detection with Model Calibration

被引:12
作者
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 条
[91]   Mining Latent Classes for Few-shot Segmentation [J].
Yang, Lihe ;
Zhuo, Wei ;
Qi, Lei ;
Shi, Yinghuan ;
Gao, Yang .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :8701-8710
[92]  
Yang S., 2021, INT C LEARN REPR
[93]  
Yang YY, 2020, ADV NEUR IN, V33
[94]   RepPoints: Point Set Representation for Object Detection [J].
Yang, Ze ;
Liu, Shaohui ;
Hu, Han ;
Wang, Liwei ;
Lin, Stephen .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :9656-9665
[95]  
Yousef M., 2020, CVPR
[96]  
Zareian A., 2021, CVPR
[97]   Accurate Few-shot Object Detection with Support-Query Mutual Guidance and Hybrid Loss [J].
Zhang, Lu ;
Zhou, Shuigeng ;
Guan, Jihong ;
Zhang, Ji .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :14419-14427
[98]   Bridging the Gap Between Anchor-based and Anchor-free Detection via Adaptive Training Sample Selection [J].
Zhang, Shifeng ;
Chi, Cheng ;
Yao, Yongqiang ;
Lei, Zhen ;
Li, Stan Z. .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020), 2020, :9756-9765
[99]   Single-Shot Refinement Neural Network for Object Detection [J].
Zhang, Shifeng ;
Wen, Longyin ;
Bian, Xiao ;
Lei, Zhen ;
Li, Stan Z. .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :4203-4212
[100]   Hallucination Improves Few-Shot Object Detection [J].
Zhang, Weilin ;
Wang, Yu-Xiong .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13003-13012