Consistency and adversarial semi-supervised learning for medical image segmentation

Abstract

Medical image segmentation based on deep learning has made enormous progress in recent years. However, the performance of existing methods generally heavily relies on a large amount of labeled data, which are commonly expensive and time-consuming to obtain. To settle above issue, in this paper, a novel semi-supervised medical image segmentation method is proposed, in which the adversarial training mechanism and the collaborative consistency learning strategy are introduced into the mean teacher model. With the adversarial training mechanism, the discriminator can generate confidence maps for unlabeled data, such that more reliable supervised information for the student network is exploited. In the process of adversarial training, we further propose a collaborative consistency learning strategy by which the auxiliary discriminator can assist the primary discriminator in achieving supervised information with higher quality. We extensively evaluate our method on three representative yet challenging medical image segmentation tasks: (1) skin lesion segmentation from dermoscopy images in the International Skin Imaging Collaboration (ISIC) 2017 dataset; (2) optic cup and optic disk (OC/OD) segmentation from fundus images in the Retinal Fundus Glaucoma Challenge (REFUGE) dataset; and (3) tumor segmentation from lower-grade glioma (LGG) tumors images. The experimental results validate the superiority and effectiveness of our proposal when compared with the state-of-the-art semi-supervised medical image segmentation methods.

Keywords: Adversarial learning; Deep neural network; Mean teacher; Medical image segmentation; Semi-supervised learning.