RevPHiSeg: A Memory-Efficient Neural Network for Uncertainty Quantification in Medical Image Segmentation
Quantifying segmentation uncertainty has become an important issue in medical image analysis due to the inherent ambiguity of anatomical structures and its pathologies. Recently, neural network-based uncertainty quantification methods have been successfully applied to various problems. One of the main limitations of the existing techniques is the high memory requirement during training; which limits their application to processing smaller field-of-views (FOVs) and/or using shallower architectures. In this paper, we investigate the effect of using reversible blocks for building memory-efficient neural network architectures for quantification of segmentation uncertainty. The reversible architecture achieves memory saving by exactly computing the activations from the outputs of the subsequent layers during backpropagation instead of storing the activations for each layer. We incorporate the reversible blocks into a recently proposed architecture called PHiSeg that is developed for uncertainty quantification in medical image segmentation. The reversible architecture, RevPHiSeg, allows training neural networks for quantifying segmentation uncertainty on GPUs with limited memory and processing larger FOVs. We perform experiments on the LIDC-IDRI dataset and an in-house prostate dataset, and present comparisons with PHiSeg. The results demonstrate that RevPHiSeg consumes ~30% less memory compared to PHiSeg while achieving very similar segmentation accuracy.
Code (1)
Tasks
Efficient Neural NetworkImage SegmentationMedical Image AnalysisMedical Image SegmentationSegmentationSemantic SegmentationUncertainty QuantificationSimilar Papers 제목 키워드 기반
A review of uncertainty quantification in medical image analysis: probabilistic and non-probabilistic methods
The comprehensive integration of machine learning healthcare models within clinical practice remains suboptimal, notwithstanding the proliferation of high-performing solutions reported in the literature. A predominant fa…
Medical Image AnalysisUncertainty QuantificationQUBIQ: Uncertainty Quantification for Biomedical Image Segmentation Challenge
Uncertainty in medical image segmentation tasks, especially inter-rater variability, arising from differences in interpretations and annotations by various experts, presents a significant challenge in achieving consisten…
Image SegmentationMedical Image SegmentationSegmentationSemantic Segmentation+1Disentangled Uncertainty and Out of Distribution Detection in Medical Generative Models
Trusting the predictions of deep learning models in safety critical settings such as the medical domain is still not a viable option. Distentangled uncertainty quantification in the field of medical imaging has received …
Image-to-Image TranslationOut-of-Distribution DetectionTranslationUncertainty QuantificationTrustworthy clinical AI solutions: a unified review of uncertainty quantification in deep learning models for medical image analysis
The full acceptance of Deep Learning (DL) models in the clinical field is rather low with respect to the quantity of high-performing solutions reported in the literature. Particularly, end users are reluctant to rely on …
Medical Image AnalysisUncertainty QuantificationEfficient Prototype Consistency Learning in Medical Image Segmentation via Joint Uncertainty and Data Augmentation
Recently, prototype learning has emerged in semi-supervised medical image segmentation and achieved remarkable performance. However, the scarcity of labeled data limits the expressiveness of prototypes in previous method…
Data AugmentationImage SegmentationMedical Image SegmentationSemantic Segmentation+2