Redundant co-training: Semi-supervised segmentation of medical images using informative redundancy

Abstract

Pseudo-labeling, consistency regularization, and co-training are common paradigms for semi-supervised learning. In this paper, we propose a novel method based on co-training and pseudo-labeling for the semi-supervised segmentation of the left ventricle. Our co-training strategy is novel and unlike most previous works does not rely on using multiple-view datasets, performing weak/strong augmentations on the input images or perturbations on the networks. We proposed creating redundant labels by utilizing the provided ground-truths and training networks segmenting different overlapping regions corresponding to the created labels. Although the new labels seem to be redundant, we demonstrated that they provide valuable information to the networks. The predictions of the redundant networks (which are trained on the redundant labels) can be used in the pixels where the primary network’s predictions are not reliable. This enables extracting a secondary source of information without requiring any additional ground-truths. The common practice in pseudo-labeling is using the reliable predictions of the unlabeled data and discarding the unreliable ones. However, we proposed utilizing predictions from the redundant networks to generate pseudo-labels for the unreliable pixels in the primary network’s predictions, rather than simply discarding them. We validated our method on two left ventricle segmentation datasets, and it surpassed the state-of-the-art semi-supervised learning approaches. Furthermore, we conducted extensive studies to analyze the proposed method from different aspects. Implementation of our work is available at https://github.com/behnam-rahmati/redundant-cotraining.