Estimating Uncertainty in Neural Networks for Cardiac MRI Segmentation:
A Benchmark Study
- URL: http://arxiv.org/abs/2012.15772v1
- Date: Thu, 31 Dec 2020 17:46:52 GMT
- Title: Estimating Uncertainty in Neural Networks for Cardiac MRI Segmentation:
A Benchmark Study
- Authors: Matthew Ng, Fumin Guo, Labonny Biswas, Steffen E. Petersen, Stefan K.
Piechnik, Stefan Neubauer, Graham Wright
- Abstract summary: Convolutional neural networks (CNNs) have demonstrated promise in automated cardiac magnetic resonance imaging segmentation.
It is important to quantify segmentation uncertainty in order to know which segmentations could be problematic.
- Score: 1.6222504666823843
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: Convolutional neural networks (CNNs) have demonstrated promise in automated
cardiac magnetic resonance imaging segmentation. However, when using CNNs in a
large real world dataset, it is important to quantify segmentation uncertainty
in order to know which segmentations could be problematic. In this work, we
performed a systematic study of Bayesian and non-Bayesian methods for
estimating uncertainty in segmentation neural networks. We evaluated Bayes by
Backprop (BBB), Monte Carlo (MC) Dropout, and Deep Ensembles in terms of
segmentation accuracy, probability calibration, uncertainty on
out-of-distribution images, and segmentation quality control. We tested these
algorithms on datasets with various distortions and observed that Deep
Ensembles outperformed the other methods except for images with heavy noise
distortions. For segmentation quality control, we showed that segmentation
uncertainty is correlated with segmentation accuracy. With the incorporation of
uncertainty estimates, we were able to reduce the percentage of poor
segmentation to 5% by flagging 31% to 48% of the most uncertain images for
manual review, substantially lower than random review of the results without
using neural network uncertainty.
Related papers
- Towards Reliable Medical Image Segmentation by utilizing Evidential Calibrated Uncertainty [52.03490691733464]
We introduce DEviS, an easily implementable foundational model that seamlessly integrates into various medical image segmentation networks.
By leveraging subjective logic theory, we explicitly model probability and uncertainty for the problem of medical image segmentation.
DeviS incorporates an uncertainty-aware filtering module, which utilizes the metric of uncertainty-calibrated error to filter reliable data.
arXiv Detail & Related papers (2023-01-01T05:02:46Z) - Reliable Joint Segmentation of Retinal Edema Lesions in OCT Images [55.83984261827332]
In this paper, we propose a novel reliable multi-scale wavelet-enhanced transformer network.
We develop a novel segmentation backbone that integrates a wavelet-enhanced feature extractor network and a multi-scale transformer module.
Our proposed method achieves better segmentation accuracy with a high degree of reliability as compared to other state-of-the-art segmentation approaches.
arXiv Detail & Related papers (2022-12-01T07:32:56Z) - Maximum Entropy on Erroneous Predictions (MEEP): Improving model
calibration for medical image segmentation [10.159176702917788]
We introduce MEEP, a training strategy for segmentation networks which selectively penalizes overconfident predictions, focusing only on misclassified pixels.
We benchmark the proposed strategy in two challenging segmentation tasks: white matter hyperintensity lesions in magnetic resonance images (MRI) of the brain, and atrial segmentation in cardiac MRI.
arXiv Detail & Related papers (2021-12-22T20:34:20Z) - Trustworthy Medical Segmentation with Uncertainty Estimation [0.7829352305480285]
This paper introduces a new Bayesian deep learning framework for uncertainty quantification in segmentation neural networks.
We evaluate the proposed framework on medical image segmentation data from Magnetic Resonances Imaging and Computed Tomography scans.
Our experiments on multiple benchmark datasets demonstrate that the proposed framework is more robust to noise and adversarial attacks as compared to state-of-the-art segmentation models.
arXiv Detail & Related papers (2021-11-10T22:46:05Z) - Inconsistency-aware Uncertainty Estimation for Semi-supervised Medical
Image Segmentation [92.9634065964963]
We present a new semi-supervised segmentation model, namely, conservative-radical network (CoraNet) based on our uncertainty estimation and separate self-training strategy.
Compared with the current state of the art, our CoraNet has demonstrated superior performance.
arXiv Detail & Related papers (2021-10-17T08:49:33Z) - Analyzing Epistemic and Aleatoric Uncertainty for Drusen Segmentation in
Optical Coherence Tomography Images [4.125187280299246]
Age-related macular degeneration (AMD) is one of the leading causes of permanent vision loss in people aged over 60 years.
We develop a U-Net based drusen segmentation model and quantify the segmentation uncertainty.
arXiv Detail & Related papers (2021-01-21T23:34:29Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - Automatic segmentation with detection of local segmentation failures in
cardiac MRI [1.281734910003263]
Three state-of-the-art convolutional neural networks (CNN) were trained to automatically segment cardiac anatomical structures.
Using publicly available CMR scans from the MICCAI 2017 ACDC challenge, the impact of CNN architecture and loss function for segmentation was investigated.
Experiments reveal that combining automatic segmentation with simulated manual correction of detected segmentation failures leads to statistically significant performance increase.
arXiv Detail & Related papers (2020-11-13T17:19:05Z) - Collaborative Boundary-aware Context Encoding Networks for Error Map
Prediction [65.44752447868626]
We propose collaborative boundaryaware context encoding networks called AEP-Net for error prediction task.
Specifically, we propose a collaborative feature transformation branch for better feature fusion between images and masks, and precise localization of error regions.
The AEP-Net achieves an average DSC of 0.8358, 0.8164 for error prediction task, and shows a high Pearson correlation coefficient of 0.9873.
arXiv Detail & Related papers (2020-06-25T12:42:01Z) - DenoiSeg: Joint Denoising and Segmentation [75.91760529986958]
We propose DenoiSeg, a new method that can be trained end-to-end on only a few annotated ground truth segmentations.
We achieve this by extending Noise2Void, a self-supervised denoising scheme that can be trained on noisy images alone, to also predict dense 3-class segmentations.
arXiv Detail & Related papers (2020-05-06T17:42:54Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.