Journal
MEDICAL IMAGE ANALYSIS
Volume 80, Issue -, Pages -Publisher
ELSEVIER
DOI: 10.1016/j.media.2022.102517
Keywords
Semi-supervised learning; Uncertainty rectifying; Pyramid consistency; Image segmentation
Categories
Funding
- National Natural Science Foundations of China [81771921, 61901084]
- key research and development project of Sichuan province, China [2020YFG0084]
Ask authors/readers for more resources
This paper introduces a simple and efficient consistency regularization approach called Uncertainty Rectified Pyramid Consistency (URPC) for semi-supervised medical image segmentation. Experimental results demonstrate that URPC achieves significant performance improvement by utilizing unlabeled data and outperforms or achieves comparable results with simpler pipeline compared to five existing semi-supervised methods.
Despite that Convolutional Neural Networks (CNNs) have achieved promising performance in many medical image segmentation tasks, they rely on a large set of labeled images for training, which is expensive and time-consuming to acquire. Semi-supervised learning has shown the potential to alleviate this challenge by learning from a large set of unlabeled images and limited labeled samples. In this work, we present a simple yet efficient consistency regularization approach for semi-supervised medical image segmentation, called Uncertainty Rectified Pyramid Consistency (URPC). Inspired by the pyramid feature network, we chose a pyramid-prediction network that obtains a set of segmentation predictions at different scales. For semi-supervised learning, URPC learns from unlabeled data by minimizing the discrepancy between each of the pyramid predictions and their average. We further present multi-scale uncertainty rectification to boost the pyramid consistency regularization, where the rectification seeks to temper the consistency loss at outlier pixels that may have substantially different predictions than the average, potentially due to upsampling errors or lack of enough labeled data. Experiments on two public datasets and an in-house clinical dataset showed that: 1) URPC can achieve large performance improvement by utilizing unlabeled data and 2) Compared with five existing semi-supervised methods, URPC achieved better or comparable results with a simpler pipeline. Furthermore, we build a semi-supervised medical image segmentation codebase to boost research on this topic: https://github.com/HiLab- git/SSL4MIS. (C) 2022 Published by Elsevier B.V.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available