
In the domain of medical imaging, many supervised learning based methods for segmentation face several challenges such as high variability in annotations from multiple experts, paucity of labelled data and class imbalanced datasets. These issues may result in segmentations that lack the requisite precision for clinical analysis and can be misleadingly overconfident without associated uncertainty quantification. This work proposes the PULASki method as a computationally efficient generative tool for biomedical image segmentation that accurately captures variability in expert annotations, even in small datasets. This approach makes use of an improved loss function based on statistical distances in a conditional variational autoencoder structure (Probabilistic UNet), which improves learning of the conditional decoder compared to the standard cross-entropy particularly in class imbalanced problems. The proposed method was analysed for two structurally different segmentation tasks (intracranial vessel and multiple sclerosis (MS) lesion) and compare our results to four well-established baselines in terms of quantitative metrics and qualitative output. These experiments involve class-imbalanced datasets characterised by challenging features, including suboptimal signal-to-noise ratios and high ambiguity. Empirical results demonstrate the PULASKi method outperforms all baselines at the 5\% significance level. Our experiments are also of the first to present a comparative study of the computationally feasible segmentation of complex geometries using 3D patches and the traditional use of 2D slices. The generated segmentations are shown to be much more anatomically plausible than in the 2D case, particularly for the vessel task.
Observer Variation, FOS: Computer and information sciences, Conditional VAE, Computer Science - Machine Learning, Computer Science - Artificial Intelligence, diagnostic imaging [Multiple Sclerosis], Computer Vision and Pattern Recognition (cs.CV), Computer Science - Computer Vision and Pattern Recognition, Computer Science - Human-Computer Interaction, methods [Image Interpretation, Computer-Assisted], Probabilistic UNet, Human-Computer Interaction (cs.HC), Machine Learning (cs.LG), methods [Magnetic Resonance Imaging], Artificial Intelligence (cs.AI), Humans, Multiple sclerosis segmentation, Supervised Machine Learning, Distribution distance, Vessel segmentation, Algorithms, ddc: ddc:610
Observer Variation, FOS: Computer and information sciences, Conditional VAE, Computer Science - Machine Learning, Computer Science - Artificial Intelligence, diagnostic imaging [Multiple Sclerosis], Computer Vision and Pattern Recognition (cs.CV), Computer Science - Computer Vision and Pattern Recognition, Computer Science - Human-Computer Interaction, methods [Image Interpretation, Computer-Assisted], Probabilistic UNet, Human-Computer Interaction (cs.HC), Machine Learning (cs.LG), methods [Magnetic Resonance Imaging], Artificial Intelligence (cs.AI), Humans, Multiple sclerosis segmentation, Supervised Machine Learning, Distribution distance, Vessel segmentation, Algorithms, ddc: ddc:610
| selected citations These citations are derived from selected sources. This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically). | 0 | |
| popularity This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network. | Average | |
| influence This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically). | Average | |
| impulse This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network. | Average |
