arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.08271cs.CV

面向目标检测的多类别置信度与定位校准

Multiclass Confidence and Localization Calibration for Object Detection

Mohamed bin Zayed University of Artificial Intelligence(穆罕默德·本·扎耶德人工智能大学)

Bimsara Pathiraja, Malitha Gunawardhana, Muhammad Haris Khan

AI总结:

针对深度神经网络目标检测校准研究不足的问题,提出一种利用预测不确定性联合校准多类别置信度与边界框定位的训练时技术,在域内和域外基准上均优于基线方法。

摘要:

尽管深度神经网络(DNN)在诸多具有挑战性的计算机视觉问题上取得了很高的预测精度,但近期研究表明,DNN往往会产生过于自信的预测,导致其校准效果不佳。现有大多数改善DNN校准的研究仅局限于分类任务,且仅针对域内预测进行校准。令人意外的是,针对目标检测方法的校准研究极少甚至空白,而目标检测在基于视觉的安全敏感和安全关键型应用中占据核心地位。本文提出了一种用于校准现代目标检测方法的新型训练时技术,该技术能够利用预测不确定性,对多类别置信度和边界框定位进行联合校准。我们在多个域内和域外检测基准上开展了大量实验,结果表明,本文提出的训练时校准方法在降低域内和域外预测的校准误差方面,始终优于多个基线方法。我们的代码和模型可在https://github.com/bimsarapathiraja/MCCL获取。

英文摘要:

Albeit achieving high predictive accuracy across many challenging computer vision problems, recent studies suggest that deep neural networks (DNNs) tend to make overconfident predictions, rendering them poorly calibrated. Most of the existing attempts for improving DNN calibration are limited to classification tasks and restricted to calibrating in-domain predictions. Surprisingly, very little to no attempts have been made in studying the calibration of object detection methods, which occupy a pivotal space in vision-based security-sensitive, and safety-critical applications. In this paper, we propose a new train-time technique for calibrating modern object detection methods. It is capable of jointly calibrating multiclass confidence and box localization by leveraging their predictive uncertainties. We perform extensive experiments on several in-domain and out-of-domain detection benchmarks. Results demonstrate that our proposed train-time calibration method consistently outperforms several baselines in reducing calibration error for both in-domain and out-of-domain predictions. Our code and models are available at https://github.com/bimsarapathiraja/MCCL.

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

↑