CertainNet: Sampling-free Uncertainty Estimation for Object Detection

Stefano Gasperini,Jan Haug,Mohammad-Ali Nikouei Mahani,Alvaro Marcos-Ramiro,Nassir Navab,Benjamin Busam,Federico Tombari
DOI: https://doi.org/10.1109/LRA.2021.3130976
2021-12-28
Abstract:Estimating the uncertainty of a neural network plays a fundamental role in safety-critical settings. In perception for autonomous driving, measuring the uncertainty means providing additional calibrated information to downstream tasks, such as path planning, that can use it towards safe navigation. In this work, we propose a novel sampling-free uncertainty estimation method for object detection. We call it CertainNet, and it is the first to provide separate uncertainties for each output signal: objectness, class, location and size. To achieve this, we propose an uncertainty-aware heatmap, and exploit the neighboring bounding boxes provided by the detector at inference time. We evaluate the detection performance and the quality of the different uncertainty estimates separately, also with challenging out-of-domain samples: BDD100K and nuImages with models trained on KITTI. Additionally, we propose a new metric to evaluate location and size uncertainties. When transferring to unseen datasets, CertainNet generalizes substantially better than previous methods and an ensemble, while being real-time and providing high quality and comprehensive uncertainty estimates.
Computer Vision and Pattern Recognition,Machine Learning,Robotics
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to provide comprehensive and efficient uncertainty estimation in object detection. Specifically, the author proposes a new method named CertainNet, which can provide independent uncertainty estimates for each output signal (such as objectness, class, location, and size) of object detection without sampling. This method is especially suitable for scenarios requiring high security, such as perception tasks in autonomous driving. By providing these detailed uncertainty information, downstream tasks (e.g., path planning) can utilize this information to achieve safer navigation. In addition, this method also performs well in terms of real - time performance and accuracy, and can handle challenging out - of - domain samples, thereby improving the generalization ability of the model on unknown datasets.