mlsanityData to 7 Oct 2026

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

arXiv 2603.29941 ↗First posted 31 Mar 2026cs.CV cs.LGcited by 1

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

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

50 most cited of 55 All 55 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.