A Fully Automated Deep Learning Network for Brain Tumor Segmentation

Chandan Ganesh Bangalore Yogananda, Bhavya R. Shah, Maryam Vejdani-Jahromi, Sahil S. Nalawade, Gowtham K. Murugesan, Frank F. Yu, Marco C. Pinho, Benjamin C. Wagner, Kyrre E. Emblem, Atle Bjørnerud, Baowei Fei, Ananth J. Madhuranthakam, Joseph A. Maldjian

Research output: Contribution to journalArticle

Abstract

We developed a fully automated method for brain tumor segmentation using deep learning; 285 brain tumor cases with multiparametric magnetic resonance images from the BraTS2018 data set were used. We designed 3 separate 3D-Dense-UNets to simplify the complex multiclass segmentation problem into individual binary-segmentation problems for each subcomponent. We implemented a 3-fold cross-validation to generalize the network's performance. The mean cross-validation Dice-scores for whole tumor (WT), tumor core (TC), and enhancing tumor (ET) segmentations were 0.92, 0.84, and 0.80, respectively. We then retrained the individual binary-segmentation networks using 265 of the 285 cases, with 20 cases held-out for testing. We also tested the network on 46 cases from the BraTS2017 validation data set, 66 cases from the BraTS2018 validation data set, and 52 cases from an independent clinical data set. The average Dice-scores for WT, TC, and ET were 0.90, 0.84, and 0.80, respectively, on the 20 held-out testing cases. The average Dice-scores for WT, TC, and ET on the BraTS2017 validation data set, the BraTS2018 validation data set, and the clinical data set were as follows: 0.90, 0.80, and 0.78; 0.90, 0.82, and 0.80; and 0.85, 0.80, and 0.77, respectively. A fully automated deep learning method was developed to segment brain tumors into their subcomponents, which achieved high prediction accuracy on the BraTS data set and on the independent clinical data set. This method is promising for implementation into a clinical workflow.

Original languageEnglish (US)
Pages (from-to)186-193
Number of pages8
JournalTomography (Ann Arbor, Mich.)
Volume6
Issue number2
DOIs
StatePublished - Jun 1 2020

Keywords

  • BraTS
  • Brain tumor segmentation
  • CNN (convolutional neural networks)
  • Dense UNet
  • MRI
  • deep learning
  • machine learning

Fingerprint Dive into the research topics of 'A Fully Automated Deep Learning Network for Brain Tumor Segmentation'. Together they form a unique fingerprint.

  • Cite this

    Bangalore Yogananda, C. G., Shah, B. R., Vejdani-Jahromi, M., Nalawade, S. S., Murugesan, G. K., Yu, F. F., Pinho, M. C., Wagner, B. C., Emblem, K. E., Bjørnerud, A., Fei, B., Madhuranthakam, A. J., & Maldjian, J. A. (2020). A Fully Automated Deep Learning Network for Brain Tumor Segmentation. Tomography (Ann Arbor, Mich.), 6(2), 186-193. https://doi.org/10.18383/j.tom.2019.00026