Better than Average: Spatially-Aware Aggregation of Segmentation Uncertainty Improves Downstream Performance
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 55 of its 81 references that are in the map. First posted on arXiv 31 Mar 2026.
Builds on
55 of 81 references.
References · 55
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,427
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,210
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
- 2018A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attackscited by 1,472
- 2019Deep High-Resolution Representation Learning for Visual Recognitioncited by 1,417
- 2020Energy-based Out-of-distribution Detectioncited by 1,296
38 more
- 2016Playing for Data: Ground Truth from Computer Gamescited by 816
- 2017Selective Classification for Deep Neural Networkscited by 799
- 2023ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoderscited by 703
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2022Out-of-Distribution Detection with Deep Nearest Neighborscited by 596
- 2020Measuring Robustness to Natural Distribution Shifts in Image Classificationcited by 452
- 2018A Probabilistic U-Net for Segmentation of Ambiguous Imagescited by 281
- 2015Bayesian SegNet: Model Uncertainty in Deep Convolutional Encoder-Decoder Architectures for Scene Understandingcited by 218
- 2021Deep Deterministic Uncertainty: A Simple Baselinecited by 200
- 2019Practical Deep Learning with Bayesian Principlescited by 158
- 2021SegmentMeIfYouCan: A Benchmark for Anomaly Segmentationcited by 150
- 2018Exploring Uncertainty Measures in Deep Networks for Multiple Sclerosis Lesion Detection and Segmentationcited by 146
- 2017Suggestive Annotation: A Deep Active Learning Framework for Biomedical Image Segmentationcited by 145
- 2018Evaluating Bayesian Deep Learning Methods for Semantic Segmentationcited by 143
- 2020Self-Adaptive Training: beyond Empirical Risk Minimizationcited by 126
- 2021Pixel-wise Anomaly Detection in Complex Driving Scenescited by 109
- 2018Bias-Reduced Uncertainty Estimation for Deep Neural Classifierscited by 106
- 2020Stochastic Segmentation Networks: Modelling Spatially Correlated Aleatoric Uncertaintycited by 100
- 2020Synthesize then Compare: Detecting Failures and Anomalies for Semantic Segmentationcited by 96
- 2021Lizard: A Large-Scale Dataset for Colonic Nuclear Instance Segmentation and Classificationcited by 89
- 2018Improving Reconstruction Autoencoder Out-of-distribution Detection with Mahalanobis Distancecited by 74
- 2021On the Practicality of Deterministic Epistemic Uncertaintycited by 62
- 2022Trustworthy clinical AI solutions: a unified review of uncertainty quantification in deep learning models for medical image analysiscited by 62
- 2022A Call to Reflect on Evaluation Practices for Failure Detection in Image Classificationcited by 51
- 2019Learning an Uncertainty-Aware Object Detector for Autonomous Drivingcited by 34
- 2024Overcoming Common Flaws in the Evaluation of Selective Classification Systemscited by 31
- 2024ValUES: A Framework for Systematic Validation of Uncertainty Estimation in Semantic Segmentationcited by 25
- 2017Checkpoint Ensembles: Ensemble Methods from a Single Training Processcited by 24
- 2020Deep Generative Model-based Quality Control for Cardiac MRI Segmentationcited by 21
- 2023Active learning for medical image segmentation with stochastic batchescited by 20
- 2020Uncertainty-Guided Efficient Interactive Refinement of Fetal Brain Segmentation from Stacks of MRI Slicescited by 16
- 2019Active Learning for Segmentation Based on Bayesian Sample Queriescited by 12
- 2024Comparative Benchmarking of Failure Detection Methods in Medical Image Segmentation: Unveiling the Role of Confidence Aggregationcited by 12
- 2021A Quantitative Comparison of Epistemic Uncertainty Maps Applied to Multi-Class Segmentationcited by 9
- 2022Novel structural-scale uncertainty measures and error retention curves: application to multiple sclerosiscited by 8
- 2024Perception Matters: Enhancing Embodied AI with Uncertainty-Aware Semantic Segmentationcited by 6
- 2024A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentationcited by 6
- 2022Panoptic segmentation with highly imbalanced semantic labelscited by 5
50 most cited of 55 All 55 (JSON)
Built on by
Citing papers per year: 2026: 1.
1 paper.