Unsupervised Learning for CT Image Segmentation via Adversarial Redrawing

Abstract

We propose a novel adversarial learning framework for unsupervised training of CNNs in CT image segmentation. It is motivated by difficulties in collecting voxel-wise annotations, which is laborious, time-consuming and expensive. It is conceptually simple, allowing us to train an effective segmentation network without any human annotation. Specifically, we design the generator with a CNN producing the segmentation results and a decoder redrawing the CT volume based on the segmentation results. The CNN is then implicitly trained in the adversarial learning framework where a discriminator gradually enforcing the generator to generate CT volumes whose distribution well matches the distribution of the training data. We further propose two constrains as regularization schemes for the training procedure to drive the model towards optimal segmentation by avoiding some unreasonable results. We conducted extensive experiments to evaluate the proposed method on a famous publicly available dataset, and the experimental results demonstrate the effectiveness of the proposed method.