Effect of latent space distribution on the segmentation of images with multiple annotations
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 variation in the reference segmentations for lung tumors and white matter hyperintensities in the brain. We show that the choice of distribution affects the sample diversity of the predictions and their overlap with respect to the reference segmentations. We have made our implementation available at https://github.com/ishaanb92/GeneralizedProbabilisticUNet
Code (1)
Tasks
DiversityMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Optimal Latent Vector Alignment for Unsupervised Domain Adaptation in Medical Image Segmentation
This paper addresses the domain shift problem for segmentation. As a solution, we propose OLVA, a novel and lightweight unsupervised domain adaptation method based on a Variational Auto-Encoder (VAE) and Optimal Transpor…
Domain AdaptationHeart SegmentationImage SegmentationMedical Image Segmentation+4DGNet: Distribution Guided Efficient Learning for Oil Spill Image Segmentation
Successful implementation of oil spill segmentation in Synthetic Aperture Radar (SAR) images is vital for marine environmental protection. In this paper, we develop an effective segmentation framework named DGNet, which …
Image SegmentationSegmentationSemantic SegmentationRethinking Polyp Segmentation from an Out-of-Distribution Perspective
Unlike existing fully-supervised approaches, we rethink colorectal polyp segmentation from an out-of-distribution perspective with a simple but effective self-supervised learning approach. We leverage the ability of mask…
SegmentationSelf-Supervised LearningLatentFM: A Latent Flow Matching Approach for Generative Medical Image Segmentation
Generative models have achieved remarkable progress with the emergence of flow matching (FM). It has demonstrated strong generative capabilities and attracted significant attention as a simulation-free flow-based framewo…
Medical Image SegmentationProbabilistic U-Net with Kendall Shape Spaces for Geometry-Aware Segmentations of Images
One of the fundamental problems in computer vision is image segmentation, the task of detecting distinct regions or objects in given images. Deep Neural Networks (DNN) have been shown to be very effective in segmenting c…
Image SegmentationSegmentationSemantic Segmentation