Towards Improving Calibration in Object Detection Under Domain Shift
About
With deep neural network based solution more readily being incorporated in real-world applications, it has been pressing requirement that predictions by such models, especially in safety-critical environments, be highly accurate and well-calibrated. Although some techniques addressing DNN calibration have been proposed, they are only limited to visual classification applications and in-domain predictions. Unfortunately, very little to no attention is paid towards addressing calibration of DNN-based visual object detectors, that occupy similar space and importance in many decision making systems as their visual classification counterparts. In this work, we study the calibration of DNN-based object detection models, particularly under domain shift. To this end, we first propose a new, plug-and-play, train-time calibration loss for object detection (coined as TCD). It can be used with various application-specific loss functions as an auxiliary loss function to improve detection calibration. Second, we devise a new implicit technique for improving calibration in self-training based domain adaptive detectors, featuring a new uncertainty quantification mechanism for object detection. We demonstrate TCD is capable of enhancing calibration with notable margins (1) across different DNN-based object detection paradigms both in in-domain and out-of-domain predictions, and (2) in different domain-adaptive detectors across challenging adaptation scenarios. Finally, we empirically show that our implicit calibration technique can be used in tandem with TCD during adaptation to further boost calibration in diverse domain shift scenarios.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 3D Object Detection Classification Calibration | nuScenes v1.0-trainval (val) | NDS63.81 | 68 | |
| Object Detection | Foggy Cityscapes (val) | mAP17.6 | 67 | |
| Object Detection | Cityscapes (val) | mAP5050.8 | 31 | |
| Object Detection | CorCOCO (val) | D-ECE10.4 | 12 | |
| Object Detection | MS-COCO In-Domain (val) | D-ECE25.2 | 6 | |
| Object Detection | CorCOCO Out-Domain (val) | D-ECE26.8 | 6 |