Hierarchical Uncertainty Estimation for Medical Image Segmentation Networks
Learning a medical image segmentation model is an inherently ambiguous task, as uncertainties exist in both images (noise) and manual annotations (human errors and bias) used for model training. To build a trustworthy image segmentation model, it is important to not just evaluate its performance but also estimate the uncertainty of the model prediction. Most state-of-the-art image segmentation networks adopt a hierarchical encoder architecture, extracting image features at multiple resolution levels from fine to coarse. In this work, we leverage this hierarchical image representation and propose a simple yet effective method for estimating uncertainties at multiple levels. The multi-level uncertainties are modelled via the skip-connection module and then sampled to generate an uncertainty map for the predicted image segmentation. We demonstrate that a deep learning segmentation network such as U-net, when implemented with such hierarchical uncertainty estimation module, can achieve a high segmentation performance, while at the same time provide meaningful uncertainty maps that can be used for out-of-distribution detection.
Code (0)
등록된 구현이 없습니다.
Tasks
Image SegmentationMedical Image SegmentationOut-of-Distribution DetectionSegmentationSemantic SegmentationSimilar Papers 제목 키워드 기반
On the Effect of Inter-observer Variability for a Reliable Estimation of Uncertainty of Medical Image Segmentation
Uncertainty estimation methods are expected to improve the understanding and quality of computer-assisted methods used in medical applications (e.g., neurosurgical interventions, radiotherapy planning), where automated m…
Image SegmentationMedical Image SegmentationSegmentationSemantic SegmentationCRISP - Reliable Uncertainty Estimation for Medical Image Segmentation
Accurate uncertainty estimation is a critical need for the medical imaging community. A variety of methods have been proposed, all direct extensions of classification uncertainty estimations techniques. The independent p…
Image SegmentationMedical Image SegmentationSegmentationSemantic Segmentation+1Efficient Bayesian Uncertainty Estimation for nnU-Net
The self-configuring nnU-Net has achieved leading performance in a large range of medical image segmentation challenges. It is widely considered as the model of choice and a strong baseline for medical image segmentation…
Image SegmentationMedical Image SegmentationSegmentationSemantic SegmentationConfidence Calibration and Predictive Uncertainty Estimation for Deep Medical Image Segmentation
Fully convolutional neural networks (FCNs), and in particular U-Nets, have achieved state-of-the-art results in semantic segmentation for numerous medical imaging applications. Moreover, batch normalization and Dice loss…
Image SegmentationMedical Image SegmentationOut-of-Distribution DetectionSegmentation+1Inconsistency-aware Uncertainty Estimation for Semi-supervised Medical Image Segmentation
In semi-supervised medical image segmentation, most previous works draw on the common assumption that higher entropy means higher uncertainty. In this paper, we investigate a novel method of estimating uncertainty. We ob…
Image SegmentationMedical Image SegmentationSegmentationSemantic Segmentation+1