arXiv:2603.01847cs.CV2026-03中稿 · IEEE IV 2026被引 1

提出GroupEnsemble,让DETR模型一次推理就估算空间不确定性。

GroupEnsemble: Efficient Uncertainty Estimation for DETR-based Object Detection

  • 用多组查询并行生成检测结果,共享解码器但互不干扰。
  • 单次前向传播完成不确定性估计,内存和延迟远低于深度集成。
  • 适合自动驾驶等需快速可靠检测的实时场景。

检测变压器(DETR)及其变体在目标检测任务中表现优异,是自动驾驶系统的关键技术。然而,这些模型的置信度分数仅反映语义不确定性,未能捕捉同样重要的空间不确定性,导致对检测可靠性评估不完整。深度集成虽能有效估计空间不确定性,但内存开销巨大,难以实际应用。蒙特卡洛丢弃(MC-Dropout)虽然更轻量,却因需多次前向传播而引入高延迟。为此,我们提出GroupEnsemble,一种高效可靠的DETR类模型不确定性估计方法。该方法在推理时向变压器解码器输入多组不同的对象查询,每组独立通过共享解码器生成完整的检测集。通过在解码器中施加注意力掩码,阻止组间查询交互,确保各组独立检测,从而实现基于集成的可靠不确定性估计。利用解码器的内在并行性,GroupEnsemble可在单次前向传播中完成不确定性估计,无需串行重复。我们在城市景观和日常场景下分别使用Cityscapes和COCO数据集验证了该方法。结果表明,结合MC-Dropout与GroupEnsemble的混合策略,在多项指标上优于深度集成,且成本仅为后者的极小部分。代码已开源:https://github.com/yutongy98/GroupEnsemble。

原文摘要 · Abstract (English)

Detection Transformer (DETR) and its variants show strong performance on object detection, a key task for autonomous systems. However, a critical limitation of these models is that their confidence scores only reflect semantic uncertainty, failing to capture the equally important spatial uncertainty. This results in an incomplete assessment of the detection reliability. On the other hand, Deep Ensembles can tackle this by providing high-quality spatial uncertainty estimates. However, their immense memory consumption makes them impractical for real-world applications. A cheaper alternative, Monte Carlo (MC) Dropout, suffers from high latency due to the need of multiple forward passes during inference to estimate uncertainty. To address these limitations, we introduce GroupEnsemble, an efficient and effective uncertainty estimation method for DETR-like models. GroupEnsemble simultaneously predicts multiple individual detection sets by feeding additional diverse groups of object queries to the transformer decoder during inference. Each query group is transformed by the shared decoder in isolation and predicts a complete detection set for the same input. An attention mask is applied to the decoder to prevent inter-group query interactions, ensuring each group detects independently to achieve reliable ensemble-based uncertainty estimation. By leveraging the decoder's inherent parallelism, GroupEnsemble efficiently estimates uncertainty in a single forward pass without sequential repetition. We validated our method under autonomous driving scenes and common daily scenes using the Cityscapes and COCO datasets, respectively. The results show that a hybrid approach combining MC-Dropout and GroupEnsemble outperforms Deep Ensembles on several metrics at a fraction of the cost. The code is available at https://github.com/yutongy98/GroupEnsemble.

目标检测不确定性估计DETR高效推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。