Lesion Conditional Image Generation for Improved Segmentation of Intracranial Hemorrhage from CT Images

Abstract

Data augmentation can effectively resolve a scarcity of images when training machine-learning algorithms. It can make them more robust to unseen images. We present a lesion conditional Generative Adversarial Network LcGAN to generate synthetic Computed Tomography (CT) images for data augmentation. A lesion conditional image (segmented mask) is an input to both the generator and the discriminator of the LcGAN during training. The trained model generates contextual CT images based on input masks. We quantify the quality of the images by using a fully convolutional network (FCN) score and blurriness. We also train another classification network to select better synthetic images. These synthetic CT images are then augmented to our hemorrhagic lesion segmentation network. By applying this augmentation method on 2.5%, 10% and 25% of original data, segmentation improved by 12.8%, 6% and 1.6% respectively.

References in corpus (8)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Conditional Generative Adversarial Nets
• Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks
• Brain Tumor Segmentation with Deep Neural Networks
• The Effectiveness of Data Augmentation in Image Classification using Deep Learning
• Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks
• A note on the evaluation of generative models
• How much data is needed to train a medical image deep learning system to achieve necessary high accuracy?