Skip to main content

Multi-plane UNet++ Ensemble for Glioblastoma Segmentation

  • Conference paper
  • First Online:
Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries (BrainLes 2021)

Part of the book series: Lecture Notes in Computer Science ((LNCS,volume 12962))

Included in the following conference series:

Abstract

Glioblastoma multiforme (grade four glioma, GBM) is the most aggressive malignant tumor in the brain and usually treated by combined surgery, chemo- and radiotherapy. The O-6-methylguanine-DNA methyltransferase (MGMT) promoter methylation status was shown to be predictive of GBM sensitivity to alkylating agent chemotherapy and is a promising marker for personalized treatment. In this paper we propose to use a multi-plane ensemble of UNet++ models for the segmentation of gliomas in MRI scans, using a combination of Dice loss and boundary loss for training. For the prediction of MGMT promoter methylation, we use an ensemble of 3D EfficientNet (one per MRI modality). Both, the UNet++ ensemble and EfficientNet are trained and validated on data provided in the context of the Brain Tumor Segmentation Challenge (BraTS) 2021, containing 2.000 fully annotated glioma samples with four different MRI modalities. We achieve Dice scores of 0.792, 0.835, and 0.906 as well as Hausdorff distances of 16.61, 10.11, and 4.54 for enhancing tumor, tumor core and whole tumor, respectively. For MGMT promoter methylation status prediction, an AUROC of 0.577 is obtained.

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
USD 109.00
Price excludes VAT (USA)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
USD 139.99
Price excludes VAT (USA)
  • Compact, lightweight edition
  • Dispatched in 3 to 5 business days
  • Free shipping worldwide - see info

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

References

  1. Baid, U., et al.: The RSNA-ASNR-MICCAI BraTS 2021 benchmark on brain tumor segmentation and radiogenomic classification. CoRR abs/2107.02314. arXiv: 2107.02314 (2021). http://arxiv.org/abs/2107.02314

  2. Bakas, S., et al.: Advancing the cancer genome atlas glioma MRI collections with expert segmentation labels and radiomic features. Sci. Data 41 (2017). https://doi.org/10.1038/sdata.2017.117

  3. Bakas, S., et al.: Segmentation Labels for the Pre-operative Scans of the TCGA-GBM collection (2017). https://doi.org/10.7937/K9/TCIA.2017.KLXWJJ1Q. https://wiki.cancerimagingarchive.net/x/KoZyAQ

  4. Bakas, S., et al.: Segmentation Labels for the Pre-operative Scans of the TCGA-LGG collection (2017). https://doi.org/10.7937/K9/TCIA.2017.GJQ7R0EF. https://wiki.cancerimagingarchive.net/x/LIZyAQ

  5. Chollet, F.: Xception: deep learning with depthwise separable convolutions. CoRR abs/1610.02357 (2016). arXiv: 1610.02357. http://arxiv.org/abs/1610.02357

  6. Dwaracherla, V., et al.: Hypermodels for Exploration. 2020. eprint. arXiv:2006.07464

  7. Goodenberger, M.L., Jenkins. R.B.: Genetics of adult glioma. Can. Gen. 205(12), 613–621 (2012). https://doi.org/10.1016/j.cancergen.2012.10.009

  8. Hansen, L.K., Salamon, P.: Neural network ensembles. IEEE Trans. Pattern Anal. Mach. Intell. 12(10), 993–1001 (1990). https://doi.org/10.1109/34.58871

    Article  Google Scholar 

  9. He, K., et al.: Deep Residual Learning for Image Recognition. CoRR abs/1512.03385 (2015). arXiv: 1512.03385 (2015). http://arxiv.org/abs/1512.03385

  10. Hinton, G., Vinyals, O., Dean, J.: Distilling the knowledge in a neural network. eprint: arXiv:1503.02531 (2015)

  11. Kervadec, H., et al.: Boundary loss for highly unbalanced segmentation. Med. Image Anal. 67, 101851 (2021). https://doi.org/10.1016/j.media.2020.101851. ISSN: 1361-8415

    Article  Google Scholar 

  12. Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. arXiv: 1412.6980 [cs.LG] (2017)

  13. Menze, B.H., et al.: The multimodal brain tumor image segmentation benchmark (BRATS). IEEE Trans. Med. Imaging 34(10), 1993–2024 (2015). https://doi.org/10.1109/tmi.2014.2377694

    Article  Google Scholar 

  14. Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. CoRR abs/1505.04597. arXiv: 1505.04597 (2015).http://arxiv.org/abs/1505.04597

  15. Sandler, M., et al.: Inverted residuals and linear bottlenecks: mobile networks for classification, detection and segmentation. CoRR abs/1801.04381, arXiv: 1801.04381 (2018). http://arxiv.org/abs/1801.04381

  16. Tan, M., Le, Q.V.: EfficientNet: rethinking model scaling for convolutional neural networks. CoRR abs/1905.11946. arXiv: 1905.11946 (2019). http://arxiv.org/abs/1905.11946

  17. Tang, C., et al.: Look closer to segment better: boundary patch refinement for instance segmentation. arXiv preprint arXiv:2104.05239 (2021)

  18. Weller, M., et al.: MGMT promoter methylation in malignant gliomas: ready for personalized medicine? Nat. Rev. Neurol. 6(1), 39–51 (2009). https://doi.org/10.1038/nrneurol.2009.197

    Article  Google Scholar 

  19. Zhou, Z., et al.: UNet++: a nested U-Net architecture for medical image segmentation. CoRR abs/1807.10165. arXiv: 1807.10165 (2018). http://arxiv.org/abs/1807.10165

Download references

Acknowledgements

Computations for this work were done (in part) using resources of the Leipzig University Computing Centre. This work was funded by the German Federal Ministry of Education and Research (BMBF) grant number 1IS18026B (Johannes Roth - ScaDS.AI Dresden/Leipzig) and 03Z1L512 (Johannes Keller, Stefan Franke, and Daniel Schneider - ICCAS).

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Daniel Schneider .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Roth, J., Keller, J., Franke, S., Neumuth, T., Schneider, D. (2022). Multi-plane UNet++ Ensemble for Glioblastoma Segmentation. In: Crimi, A., Bakas, S. (eds) Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries. BrainLes 2021. Lecture Notes in Computer Science, vol 12962. Springer, Cham. https://doi.org/10.1007/978-3-031-08999-2_23

Download citation

  • DOI: https://doi.org/10.1007/978-3-031-08999-2_23

  • Published:

  • Publisher Name: Springer, Cham

  • Print ISBN: 978-3-031-08998-5

  • Online ISBN: 978-3-031-08999-2

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics