Uncertainty-aware temporal self-learning (UATS): Semi-supervised learning for segmentation of prostate zones and beyond. 2021

Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
Faculty of Computer Science and Research Campus STIMULATE, University of Magdeburg, Germany. Electronic address: anneke.meyer@ovgu.de.

Various convolutional neural network (CNN) based concepts have been introduced for the prostate's automatic segmentation and its coarse subdivision into transition zone (TZ) and peripheral zone (PZ). However, when targeting a fine-grained segmentation of TZ, PZ, distal prostatic urethra (DPU) and the anterior fibromuscular stroma (AFS), the task becomes more challenging and has not yet been solved at the level of human performance. One reason might be the insufficient amount of labeled data for supervised training. Therefore, we propose to apply a semi-supervised learning (SSL) technique named uncertainty-aware temporal self-learning (UATS) to overcome the expensive and time-consuming manual ground truth labeling. We combine the SSL techniques temporal ensembling and uncertainty-guided self-learning to benefit from unlabeled images, which are often readily available. Our method significantly outperforms the supervised baseline and obtained a Dice coefficient (DC) of up to 78.9%, 87.3%, 75.3%, 50.6% for TZ, PZ, DPU and AFS, respectively. The obtained results are in the range of human inter-rater performance for all structures. Moreover, we investigate the method's robustness against noise and demonstrate the generalization capability for varying ratios of labeled data and on other challenging tasks, namely the hippocampus and skin lesion segmentation. UATS achieved superiority segmentation quality compared to the supervised baseline, particularly for minimal amounts of labeled data.

UI MeSH Term Description Entries
D008297 Male Males
D011467 Prostate A gland in males that surrounds the neck of the URINARY BLADDER and the URETHRA. It secretes a substance that liquefies coagulated semen. It is situated in the pelvic cavity behind the lower part of the PUBIC SYMPHYSIS, above the deep layer of the triangular ligament, and rests upon the RECTUM. Prostates
D006624 Hippocampus A curved elevation of GRAY MATTER extending the entire length of the floor of the TEMPORAL HORN of the LATERAL VENTRICLE (see also TEMPORAL LOBE). The hippocampus proper, subiculum, and DENTATE GYRUS constitute the hippocampal formation. Sometimes authors include the ENTORHINAL CORTEX in the hippocampal formation. Ammon Horn,Cornu Ammonis,Hippocampal Formation,Subiculum,Ammon's Horn,Hippocampus Proper,Ammons Horn,Formation, Hippocampal,Formations, Hippocampal,Hippocampal Formations,Hippocampus Propers,Horn, Ammon,Horn, Ammon's,Proper, Hippocampus,Propers, Hippocampus,Subiculums
D006801 Humans Members of the species Homo sapiens. Homo sapiens,Man (Taxonomy),Human,Man, Modern,Modern Man
D000069553 Supervised Machine Learning A MACHINE LEARNING paradigm used to make predictions about future instances based on a given set of labeled paired input-output training (sample) data. Active Machine Learning,Inductive Machine Learning,Learning from Labeled Data,Machine Learning with a Teacher,Semi-supervised Learning,Learning, Active Machine,Learning, Inductive Machine,Learning, Semi-supervised,Learning, Supervised Machine,Machine Learning, Active,Machine Learning, Inductive,Machine Learning, Supervised,Semi supervised Learning
D016571 Neural Networks, Computer A computer architecture, implementable in either hardware or software, modeled after biological neural networks. Like the biological system in which the processing capability is a result of the interconnection strengths between arrays of nonlinear processing nodes, computerized neural networks, often called perceptrons or multilayer connectionist models, consist of neuron-like units. A homogeneous group of units makes up a layer. These networks are good at pattern recognition. They are adaptive, performing tasks by example, and thus are better for decision-making than are linear learning machines or cluster analysis. They do not require explicit programming. Computational Neural Networks,Connectionist Models,Models, Neural Network,Neural Network Models,Neural Networks (Computer),Perceptrons,Computational Neural Network,Computer Neural Network,Computer Neural Networks,Connectionist Model,Model, Connectionist,Model, Neural Network,Models, Connectionist,Network Model, Neural,Network Models, Neural,Network, Computational Neural,Network, Computer Neural,Network, Neural (Computer),Networks, Computational Neural,Networks, Computer Neural,Networks, Neural (Computer),Neural Network (Computer),Neural Network Model,Neural Network, Computational,Neural Network, Computer,Neural Networks, Computational,Perceptron
D035501 Uncertainty The condition in which reasonable knowledge regarding risks, benefits, or the future is not available.

Related Publications

Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
January 2021, IEEE transactions on medical imaging,
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
January 2024, Medical image analysis,
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
January 2020, Statistical atlases and computational models of the heart. STACOM (Workshop),
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
March 2022, IEEE transactions on medical imaging,
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
March 2024, Computers in biology and medicine,
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
October 2022, Computers in biology and medicine,
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
May 2023, IEEE transactions on neural networks and learning systems,
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
April 2022, International journal of neural systems,
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
June 2022, IEEE transactions on medical imaging,
Anneke Meyer, and Suhita Ghosh, and Daniel Schindele, and Martin Schostak, and Sebastian Stober, and Christian Hansen, and Marko Rak
December 2023, IEEE transactions on pattern analysis and machine intelligence,
Copied contents to your clipboard!