Semi-supervised Segmentation with Self-training Based on Quality Estimation and Refinement.

Zhou Zheng,Xiaoxia Wang,Xiaoyun Zhang,Yu-Min Zhong,Xiaofen Yao,Ya Zhang,Yanfeng Wang
DOI: https://doi.org/10.1007/978-3-030-59861-7_4
2020-01-01
Abstract:segmentation is time-consuming and highly depends on expert knowledge. Therefore semi-supervised segmentation has been investigated by utilizing a small set of labeled data and a large set of unlabeled data with generated pseudo labels, but the quality of pseudo labels is crucial since bad labels may lead to even worse segmentation. In this paper, we propose a novel semi-supervised segmentation framework which can automatically estimate and refine the quality of pseudo labels, and select only those good samples to expand the training set for selftraining. Specifically the quality is automatically estimated in the view of shape and semantic confidence using variational auto-encoder (VAE) and CNN based network. And, the selected labels are refined in an adversarial way by distinguishing whether a label is the ground truth mask or not at pixel level. Our method is evaluated on the established neuroblastoma(NB) and BraTS18 dataset and outperforms other state-of-the-art semi-supervised medical image segmentation methods. We can achieve a fully supervised performance while requiring similar to 4x less annotation effort.
What problem does this paper attempt to address?