Generalized Probabilistic U-Net for medical image segementation

07/26/2022

∙

We propose the Generalized Probabilistic U-Net, which extends the Probabilistic U-Net by allowing more general forms of the Gaussian distribution as the latent space distribution that can better approximate the uncertainty in the reference segmentations. We study the effect the choice of latent space distribution has on capturing the uncertainty in the reference segmentations using the LIDC-IDRI dataset. We show that the choice of distribution affects the sample diversity of the predictions and their overlap with respect to the reference segmentations. For the LIDC-IDRI dataset, we show that using a mixture of Gaussians results in a statistically significant improvement in the generalized energy distance (GED) metric with respect to the standard Probabilistic U-Net. We have made our implementation available at https://github.com/ishaanb92/GeneralizedProbabilisticUNet

READ FULL TEXT

Generalized Probabilistic U-Net for medical image segementation

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

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

Learning Probabilistic Ordinal Embeddings for Uncertainty-Aware Regression

CRISP - Reliable Uncertainty Estimation for Medical Image Segmentation

Deep Variational Clustering Framework for Self-labeling of Large-scale Medical Images

LOHO: Latent Optimization of Hairstyles via Orthogonalization

Multivariate Probabilistic Regression with Natural Gradient Boosting

Generalized Probabilistic U-Net for medical image segementation

Related Research

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

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

Learning Probabilistic Ordinal Embeddings for Uncertainty-Aware Regression

CRISP - Reliable Uncertainty Estimation for Medical Image Segmentation

Deep Variational Clustering Framework for Self-labeling of Large-scale Medical Images

LOHO: Latent Optimization of Hairstyles via Orthogonalization

Multivariate Probabilistic Regression with Natural Gradient Boosting