Brain Tumor Classification Using a Combination of Variational Autoencoders and Generative Adversarial Networks

Bilal Ahmad, Jun Sun, Qi You, Vasile Palade, Zhongjie Mao

Research output: Contribution to journalArticlepeer-review

52 Downloads (Pure)

Abstract

Brain tumors are a pernicious cancer with one of the lowest five-year survival rates. Neurologists often use magnetic resonance imaging (MRI) to diagnose the type of brain tumor. Automated computer-assisted tools can help them speed up the diagnosis process and reduce the burden on the health care systems. Recent advances in deep learning for medical imaging have shown remarkable results, especially in the automatic and instant diagnosis of various cancers. However, we need a large amount of data (images) to train the deep learning models in order to obtain good results. Large public datasets are rare in medicine. This paper proposes a framework based on unsupervised deep generative neural networks to solve this limitation. We combine two generative models in the proposed framework: Variational autoencoders (VAEs) and generative adversarial networks (GANs). We swap the encoder–decoder network after initially training it on the training set of available MR images. The output of this swapped network is a noise vector that has information of the image manifold, and the cascaded generative adversarial network samples the input from this informative noise vector instead of random Gaussian noise. The proposed method helps the GAN to avoid mode collapse and generate realistic-looking brain tumor magnetic resonance images. These artificially generated images could solve the limitation of small medical datasets up to a reasonable extent and help the deep learning models perform acceptably. We used the ResNet50 as a classifier, and the artificially generated brain tumor images are used to augment the real and available images during the classifier training. We compared the classification results with several existing studies and state-of-the-art machine learning models. Our proposed methodology noticeably achieved better results. By using brain tumor images generated artificially by our proposed method, the classification average accuracy improved from 72.63% to 96.25%. For the most severe class of brain tumor, glioma, we achieved 0.769, 0.837, 0.833, and 0.80 values for recall, specificity, precision, and F1-score, respectively. The proposed generative model framework could be used to generate medical images in any domain, including PET (positron emission tomography) and MRI scans of various parts of the body, and the results show that it could be a useful clinical tool for medical experts.

Original languageEnglish
Article number223
Number of pages19
JournalBiomedicines
Volume10
Issue number2
DOIs
Publication statusPublished - 21 Jan 2022

Bibliographical note

Copyright: © 2022 by the authors. Licensee MDPI, Basel, Switzerland.
This article is an open access article distributed under the terms and
conditions of the Creative Commons Attribution (CC BY) license (https://
creativecommons.org/licenses/by/ 4.0/)

Funder

This research work was supported in part by the National Natural Science Foundation of China (Projects Numbers: 61673194, 61672263, 61672265), and in part by the national first-class discipline program of Light Industry Technology and Engineering (Project Number: LITE2018-25).

Funding Information:
Funding: This research work was supported in part by the National Natural Science Foundation of China (Projects Numbers: 61673194, 61672263, 61672265), and in part by the national first‑class discipline program of Light Industry Technology and Engineering (Project Number: LITE2018‑25).

Publisher Copyright:
© 2022 by the authors.

Keywords

  • Brain tumor classification
  • Cancer classification
  • Convolutional neural networks
  • Deep learning
  • Generative adversarial networks
  • Glioma
  • Meningioma
  • MRI
  • PET
  • Pituitary
  • Radiolabeled PET
  • Variational autoencoder

ASJC Scopus subject areas

  • Medicine (miscellaneous)
  • Biochemistry, Genetics and Molecular Biology(all)

Fingerprint

Dive into the research topics of 'Brain Tumor Classification Using a Combination of Variational Autoencoders and Generative Adversarial Networks'. Together they form a unique fingerprint.

Cite this