arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.03037cs.CVcs.AI
收藏

EvCenterNet:使用证据学习进行目标检测的不确定性估计

EvCenterNet: Uncertainty Estimation for Object Detection using Evidential Learning

Fraunhofer IKS(弗劳恩霍夫智能认知系统研究所);University of Freiburg(弗赖堡大学);Federal University of Rio Grande(里奥格兰德联邦大学);University of Tech. Nuremberg(纽伦堡工业大学);TU München(慕尼黑工业大学)

Monish R. Nallapareddy, Kshitij Sirohi, Paulo L. J. Drews-Jr, Wolfram Burgard, Chih-Hong Cheng, Abhinav Valada

AI总结:

本文提出 EvCenterNet 证据学习二维目标检测框架,联合估计分类与回归不确定性,并通过组合损失、类别平衡加权和不确定性聚焦训练提升分布外检测精度。

摘要:

不确定性估计在自动驾驶等安全关键场景中至关重要,因为它为包括高层决策和路径规划在内的多项下游任务提供有价值的信息。在本研究中,我们提出 EvCenterNet,这是一种新颖的、具备不确定性感知能力的二维目标检测框架,使用证据学习直接估计分类和回归不确定性。为了将证据学习用于目标检测,我们针对稀疏热图输入设计了证据损失与焦点损失的组合。我们引入用于回归和热图预测的类别平衡加权,以应对证据学习遇到的类别不平衡问题。此外,我们提出一种学习方案,主动利用预测出的热图不确定性,通过聚焦最不确定的点来提升检测性能。我们在 KITTI 数据集上训练模型,并在包括 BDD100K 和 nuImages 在内的具有挑战性的分布外数据集上进行评估。实验表明,与基础模型相比,我们的方法提高了精度,并最小化了执行时间损失。

英文摘要:

Uncertainty estimation is crucial in safety-critical settings such as automated driving as it provides valuable information for several downstream tasks including high-level decision making and path planning. In this work, we propose EvCenterNet, a novel uncertainty-aware 2D object detection framework using evidential learning to directly estimate both classification and regression uncertainties. To employ evidential learning for object detection, we devise a combination of evidential and focal loss functions for the sparse heatmap inputs. We introduce class-balanced weighting for regression and heatmap prediction to tackle the class imbalance encountered by evidential learning. Moreover, we propose a learning scheme to actively utilize the predicted heatmap uncertainties to improve the detection performance by focusing on the most uncertain points. We train our model on the KITTI dataset and evaluate it on challenging out-of-distribution datasets including BDD100K and nuImages. Our experiments demonstrate that our approach improves the precision and minimizes the execution time loss in relation to the base model.

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

↑