arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2110.01604cs.CVcs.LGcs.RO
收藏

CertainNet:用于目标检测的无采样不确定性估计

CertainNet: Sampling-free Uncertainty Estimation for Object Detection

Technische Universität München (TUM)(慕尼黑工业大学);BMW Group(宝马集团);Johns Hopkins University(约翰斯·霍普金斯大学)

Stefano Gasperini, Jan Haug, Mohammad-Ali Nikouei Mahani, Alvaro Marcos-Ramiro, Nassir Navab, Benjamin Busam, Federico Tombari

摘要:

估计神经网络的不确定性在安全关键设置中起着重要作用。在自动驾驶的感知中,测量不确定性意味着为下游任务提供额外的校准信息,例如路径规划,可以将其用于安全导航。在这项工作中,我们提出了一种新的无采样不确定性估计方法的目标检测。我们称之为CNET,它是第一个为每个输出信号提供单独的不确定性:对象,类别,位置和大小。为了实现这一点,我们提出了一个不确定性感知的热图,并利用检测器在推理时提供的相邻边界框。我们分别评估了不同不确定性估计的检测性能和质量,以及具有挑战性的域外样本:BDD100K和nuImages,模型在KITTI上训练。此外,我们提出了一个新的度量来评估位置和大小的不确定性。当传输到看不见的数据集时,CNONet的泛化能力比以前的方法和集合要好得多,同时是实时的,并提供高质量和全面的不确定性估计。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑