Deep neural network with generative adversarial networks pre-training for brain tumor classification based on MR images

By Navid Ghassemi, A. Shoeibi, M. Rouhani. Biomedical Signal Processing and Control, 2020

374 citations recorded by the source corpus.

Abstract In this paper, a new deep learning method for tumor classification in MR images is presented. A deep neural network is first pre-trained as a discriminator in a generative adversarial network (GAN) on different datasets of MR images to extract robust features and to learn the structure of MR images in its convolutional layers. Then the fully connected layers are replaced and the whole deep network is trained as a classifier to distinguish three tumor classes. The deep neural network classifier has six layers and about 1.7 million weight parameters. Pre-training as a discriminator of a GAN together with other techniques such as data augmentations (image rotation and mirroring) and dropout prevent the network from overtraining on a relatively small dataset. This method is applied to an MRI data set consists of 3064 T1-CE MR images from 233 patients, 13 images from each patient on average, with three different brain tumor types: meningioma (708 images), glioma (1426 images), and pituitary tumor (930 images). 5-Fold cross-validation is used to evaluate the performance of overall design, achieving the highest accuracy as compared to state-of-art methods.

Subfields: Artificial Intelligence, Biophysics, Neurology

View Paper: https://www.semanticscholar.org/paper/5d0abcbe11275a93c2fa1e896d06effa8d01c786

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Abbas Khosravi
  2. Amir F. Atiya
  3. Saeid Nahavandi
  4. Sadiq Hussain
  5. Mohamed Esmail Karar
  6. Ezz El-Din Hemdan
  7. Marwa A. Shouman
  8. Vadlamani Ravi

Citations in the corpus