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.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
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
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
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
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
Chollet, F.: Xception: deep learning with depthwise separable convolutions. CoRR abs/1610.02357 (2016). arXiv: 1610.02357. http://arxiv.org/abs/1610.02357
Dwaracherla, V., et al.: Hypermodels for Exploration. 2020. eprint. arXiv:2006.07464
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
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
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
Hinton, G., Vinyals, O., Dean, J.: Distilling the knowledge in a neural network. eprint: arXiv:1503.02531 (2015)
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
Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. arXiv: 1412.6980 [cs.LG] (2017)
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
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
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
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
Tang, C., et al.: Look closer to segment better: boundary patch refinement for instance segmentation. arXiv preprint arXiv:2104.05239 (2021)
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
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
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
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
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)