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

BEA:使用萌芽集成架构重新审视基于锚点的目标检测深度神经网络

BEA: Revisiting anchor-based object detection DNN using Budding Ensemble Architecture

Intel Labs(英特尔实验室);Technical University of Munich(慕尼黑工业大学)

Syed Sha Qutub, Neslihan Kose, Rafael Rosales, Michael Paulitsch, Korbinian Hagn, Florian Geissler, Yang Peng, Gereon Hinz, Alois Knoll

AI总结:

本文提出萌芽集成架构(BEA),通过改进损失函数校准置信度并降低不确定性误差,提升基于锚点的目标检测模型精度,在KITTI上使YOLOv3的mAP和AP50分别提高6%和3.7%,并增强分布外检测能力。

摘要:

本文介绍了萌芽集成架构(BEA),一种用于基于锚点的目标检测模型的新型精简集成架构。目标检测模型在基于视觉的任务中至关重要,尤其是在自主系统中。它们应提供精确的边界框检测,同时校准其预测的置信度分数,从而获得更高质量的不确定性估计。然而,当前模型可能因假阳性获得高分或真阳性因低分而被丢弃而做出错误决策。BEA旨在解决这些问题。BEA中提出的损失函数改善了置信度分数的校准并降低了不确定性误差,从而更好地区分真阳性和假阳性,最终提高目标检测模型的准确性。使用BEA方法及其提出的损失函数增强了Base-YOLOv3和SSD模型。在KITTI数据集上训练的Base-YOLOv3上的BEA使得mAP和AP50分别提高了6%和3.7%。利用平衡良好的不确定性估计阈值实时丢弃样本,甚至比基础模型高出9.6%的AP50。这归因于基于AP50的保留曲线下面积增加了40%,该曲线用于衡量置信度分数校准的质量。此外,在KITTI上训练的BEA-YOLOv3在Citypersons、BDD100K和COCO数据集上提供了优于YOLOv3和Gaussian-YOLOv3的集成及原始模型的分布外检测性能。

英文摘要:

This paper introduces the Budding Ensemble Architecture (BEA), a novel reduced ensemble architecture for anchor-based object detection models. Object detection models are crucial in vision-based tasks, particularly in autonomous systems. They should provide precise bounding box detections while also calibrating their predicted confidence scores, leading to higher-quality uncertainty estimates. However, current models may make erroneous decisions due to false positives receiving high scores or true positives being discarded due to low scores. BEA aims to address these issues. The proposed loss functions in BEA improve the confidence score calibration and lower the uncertainty error, which results in a better distinction of true and false positives and, eventually, higher accuracy of the object detection models. Both Base-YOLOv3 and SSD models were enhanced using the BEA method and its proposed loss functions. The BEA on Base-YOLOv3 trained on the KITTI dataset results in a 6% and 3.7% increase in mAP and AP50, respectively. Utilizing a well-balanced uncertainty estimation threshold to discard samples in real-time even leads to a 9.6% higher AP50 than its base model. This is attributed to a 40% increase in the area under the AP50-based retention curve used to measure the quality of calibration of confidence scores. Furthermore, BEA-YOLOV3 trained on KITTI provides superior out-of-distribution detection on Citypersons, BDD100K, and COCO datasets compared to the ensembles and vanilla models of YOLOv3 and Gaussian-YOLOv3.

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

↑