Multimodal Learning 相关度: 7/10

Better than Average: Spatially-Aware Aggregation of Segmentation Uncertainty Improves Downstream Performance

Vanessa Emanuela Guarino, Claudia Winklmayr, Jannik Franzen, Josef Lorenz Rumberger, Manuel Pfeuffer, Sonja Greven, Klaus Maier-Hein, Carsten T. Lüth, Christoph Karg, Dagmar Kainmueller
arXiv: 2603.29941v1 发布: 2026-03-31 更新: 2026-03-31

AI 摘要

该论文研究了分割不确定性聚合方法对下游任务的影响,并提出了空间感知的聚合策略。

主要贡献

  • 分析了常用聚合策略的性质、局限性和陷阱
  • 提出了新的空间不确定性结构聚合策略
  • 提出了一个跨数据集稳健的元聚合器

方法论

形式化分析现有方法,提出新方法,并在多个数据集上进行基准测试,评估OoD和失败检测性能。

原文摘要

Uncertainty Quantification (UQ) is crucial for ensuring the reliability of automated image segmentations in safety-critical domains like biomedical image analysis or autonomous driving. In segmentation, UQ generates pixel-wise uncertainty scores that must be aggregated into image-level scores for downstream tasks like Out-of-Distribution (OoD) or failure detection. Despite routine use of aggregation strategies, their properties and impact on downstream task performance have not yet been comprehensively studied. Global Average is the default choice, yet it does not account for spatial and structural features of segmentation uncertainty. Alternatives like patch-, class- and threshold-based strategies exist, but lack systematic comparison, leading to inconsistent reporting and unclear best practices. We address this gap by (1) formally analyzing properties, limitations, and pitfalls of common strategies; (2) proposing novel strategies that incorporate spatial uncertainty structure and (3) benchmarking their performance on OoD and failure detection across ten datasets that vary in image geometry and structure. We find that aggregators leveraging spatial structure yield stronger performance in both downstream tasks studied. However, the performance of individual aggregators depends heavily on dataset characteristics, so we (4) propose a meta-aggregator that integrates multiple aggregators and performs robustly across datasets.

标签

分割不确定性 不确定性量化 图像分割 Out-of-Distribution Detection Failure Detection

arXiv 分类

cs.CV cs.LG