Conditional Adversarial Network for Semantic Segmentation of Brain Tumor

Abstract

Automated medical image analysis has a significant value in diagnosis and treatment of lesions. Brain tumors segmentation has a special importance and difficulty due to the difference in appearances and shapes of the different tumor regions in magnetic resonance images. Additionally, the data sets are heterogeneous and usually limited in size in comparison with the computer vision problems. The recently proposed adversarial training has shown promising results in generative image modeling. In this paper, we propose a novel end-to-end trainable architecture for brain tumor semantic segmentation through conditional adversarial training. We exploit conditional Generative Adversarial Network (cGAN) and train a semantic segmentation Convolution Neural Network (CNN) along with an adversarial network that discriminates segmentation maps coming from the ground truth or from the segmentation network for BraTS 2017 segmentation task[15, 4, 2, 3]. We also propose an end-to-end trainable CNN for survival day prediction based on deep learning techniques for BraTS 2017 prediction task [15, 4, 2, 3]. The experimental results demonstrate the superior ability of the proposed approach for both tasks. The proposed model achieves on validation data a DICE score, Sensitivity and Specificity respectively 0.68, 0.99 and 0.98 for the whole tumor, regarding online judgment system.

arXiv commentSubmitted to BraTS challenges which is part of MICCAI-2017

References in corpus (8)

• Conditional Generative Adversarial Nets
• Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks
• Generating Videos with Scene Dynamics
• Deep multi-scale video prediction beyond mean square error
• Unsupervised and Semi-supervised Learning with Categorical Generative Adversarial Networks
• Unsupervised Learning for Physical Interaction through Video Prediction
• Adversarial Deep Structural Networks for Mammographic Mass Segmentation
• A-Fast-RCNN: Hard Positive Generation via Adversary for Object Detection

Cited by in corpus (8)

• Generative Adversarial Network in Medical Imaging: A Review
• Cross-Modality Deep Feature Learning for Brain Tumor Segmentation
• Multi-Task Generative Adversarial Network for Handling Imbalanced Clinical Data
• Deep Learning in Multi-organ Segmentation
• Red-GAN: Attacking class imbalance via conditioned generation. Yet another perspective on medical image synthesis for skin lesion dermoscopy and brain tumor MRI
• Mask Embedding in conditional GAN for Guided Synthesis of High Resolution Images
• 3D RegNet: Deep Learning Model for COVID-19 Diagnosis on Chest CT Image
• Automatic Liver Segmentation with Adversarial Loss and Convolutional Neural Network