Efficient Uncertainty Estimation for Monocular 3D Object Detection in Autonomous Driving

Zechen Liu,Zhihua Han
DOI: https://doi.org/10.1109/itsc48978.2021.9564433
2021-09-19
Abstract:Monocular 3D object detection is an ill-posed problem due to the loss of geometry information in 2D images. To improve the reliability of monocular vision system, we propose to replace the deterministic estimation with a probabilistic prediction by computing both aleatoric and epistemic uncertainty in 3D detection task. We further design a continuous focal loss that can improve location accuracy and when combining with aleatoric uncertainty estimation, also reduce the effect of noise data during training. Our method is capable of estimating uncertainty in a single forward pass while achieving nearly real-time performance with 0.1s per image on a modern platform. Additionally, to make fair comparison between point and probabilistic prediction, we present a novel probabilistic localization evaluation metric which computes the average error over the predicted location distribution. Our experiments indicate that we achieve competitive result on the official KITTI 3D detection leaderboard, among other deterministic methods. Furthermore, our method gains more accurate localization accuracy under different localization evaluation metrics.
What problem does this paper attempt to address?