Abstract:The projection of planar MRI data onto a spherical surface is equivalent to a nonlinear image transformation that retains global anatomical information. By incorporating this image transformation process in our proposed spherical projection-based U-Net (SPU-Net) segmentation model design, multiple independent segmentation predictions can be obtained from a single MRI. The final segmentation is the average of all available results, and the variation can be visualized as a pixel-wise uncertainty map. An uncertainty score was introduced to evaluate and compare the performance of uncertainty measurements. The proposed SPU-Net model was implemented on the basis of 369 glioma patients with MP-MRI scans (T1, T1-Ce, T2, and FLAIR). Three SPU-Net models were trained to segment enhancing tumor (ET), tumor core (TC), and whole tumor (WT), respectively. The SPU-Net model was compared with (1) the classic U-Net model with test-time augmentation (TTA) and (2) linear scaling-based U-Net (LSU-Net) segmentation models in terms of both segmentation accuracy (Dice coefficient, sensitivity, specificity, and accuracy) and segmentation uncertainty (uncertainty map and uncertainty score). The developed SPU-Net model successfully achieved low uncertainty for correct segmentation predictions (e.g., tumor interior or healthy tissue interior) and high uncertainty for incorrect results (e.g., tumor boundaries). This model could allow the identification of missed tumor targets or segmentation errors in U-Net. Quantitatively, the SPU-Net model achieved the highest uncertainty scores for three segmentation targets (ET/TC/WT): 0.826/0.848/0.936, compared to 0.784/0.643/0.872 using the U-Net with TTA and 0.743/0.702/0.876 with the LSU-Net (scaling factor = 2). The SPU-Net also achieved statistically significantly higher Dice coefficients, underscoring the improved segmentation accuracy.

Effect of latent space distribution on the segmentation of images with multiple annotations

Generalized Probabilistic U-Net for medical image segementation

Investigating and Improving Latent Density Segmentation Models for Aleatoric Uncertainty Quantification in Medical Imaging

3D U$$^$$-Net: A 3D Universal U-Net for Multi-domain Medical Image Segmentation

Probabilistic U-Net with Kendall Shape Spaces for Geometry-Aware Segmentations of Images

Quantifying U-Net Uncertainty in Multi-Parametric MRI-based Glioma Segmentation by Spherical Image Projection

Hypernet-Ensemble Learning of Segmentation Probability for Medical Image Segmentation with Ambiguous Labels

Distributional Gaussian Processes Layers for Out-of-Distribution Detection

Learning to segment subcortical structures from noisy annotations with a novel uncertainty-reliability aware learning framework

LATUP-Net: A Lightweight 3D Attention U-Net with Parallel Convolutions for Brain Tumor Segmentation

Effect of learning parameters on the performance of U-Net Model in segmentation of Brain tumor

ProPanDL: A Modular Architecture for Uncertainty-Aware Panoptic Segmentation

Probabilistic Deep Learning for Instance Segmentation

Unsupervised Model Adaptation for Source-free Segmentation of Medical Images

Effective data augmentation for brain tumor segmentation

Denoising Diffusions in Latent Space for Medical Image Segmentation

Brain tumor image segmentation algorithm based on multimodal feature fusion of Bayesian weight distribution

Modeling the Probabilistic Distribution of Unlabeled Data forOne-shot Medical Image Segmentation

Latent Noise Segmentation: How Neural Noise Leads to the Emergence of Segmentation and Grouping

Uncertainty-aware U-Net for Medical Landmark Detection

Medical image segmentation using scalable functional variational Bayesian neural networks with Gaussian processes