Abstract
BraTS2021 Task1 is research on segmentation of intrinsically heterogeneous brain glioblastoma sub-regions in mpMRI scans. Base on BraTS 2020 top ten team’s solution (open brats2020, ranked among the top ten teams work), we proposed a similar as 3D U-Net neural network, called as TE U-Net, to differentiate glioma sub-regions class. According that automatically learns to focus on sub-regions class structures of varying shapes and sizes, we proposed TE U-Net which is similar with U-Net++ network architecture. Firstly, we reserved encoder second and third stage’s skip connect design, then also cut off first stage skip connect design. Secondly, multiple stage features through attention gate block before features skip connect, so as to ensemble channels and space region information to suppress irrelevant regions. Finally, in order to improve model performance, on network post-processing stage, we ensemble multiple similar 3D U-Net with attention module. On the online validation database, the TE U-Net architecture get best result is that the GD-enhancing tumor (ET) dice is 83.79%, the peritumoral edematous/invaded tissue (TC) dice is 86.47%, and the necrotic tumor core (WT) dice is 91.98%, Hausdorff(95%) values is 6.39,7.81,3.86and Sensitivity values is 82.20%, 83.99%, 91.92% respectively. And our solution achieved a dice of 85.62%,86.70%,90.64% for ET,TC and WT, as well as Hausdorff(95%) is 18.70,21.06,10.88 on final private test dataset.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Neugut, A.I., et al.: Magnetic resonance imaging-based screening for asymptomatic brain tumors: a review. Oncologist 24(3), 375 (2019)
Henry, T., et al.: Brain tumor segmentation with self-ensembled, deeply-supervised 3D U-net neural networks: a BraTS 2020 challenge solution. arXiv preprint arXiv:2011.01045 (2020)
Baid, U., et al., The RSNA-ASNR-MICCAI BraTS 2021 benchmark on brain tumor segmentation and radiogenomic classification. arXiv:2107.02314 (2021)
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
Bakas, S., et al.: Advancing the cancer genome atlas glioma MRI collections with expert segmentation labels and radiomic features. Nat. Sci. Data 4, 170117 (2017). https://doi.org/10.1038/sdata.2017.117
Bakas, S., et al.: segmentation labels and radiomic features for the pre-operative scans of the TCGA-GBM collection. Cancer Imaging Arch. (2017). https://doi.org/10.7937/K9/TCIA.2017.KLXWJJ1Q
Bakas, S., et al.: Segmentation labels and radiomic features for the pre-operative scans of the TCGA-LGG collection. Cancer Imaging Arch. (2017). https://doi.org/10.7937/K9/TCIA.2017.GJQ7R0EF
Taha, A.A., Hanbury, A.: Metrics for evaluating 3D medical image segmentation: analysis, selection, and tool. BMC Med. Imaging 15(1), 1–28 (2015)
Isensee, F., et al.: nnU-Net for brain tumor segmentation. International MICCAI Brainlesion Workshop, pp. 118–132. Springer, Cham (2020)
Dice, L.R.: Measures of the amount of ecologic association between species. Ecology 26(3), 297–302 (1945)
Ronneberger, O., Fischer, P., Brox, T.: U-net: convolutional networks for biomedical image segmentation. In: International Conference on Medical image computing and computer-assisted intervention, pp. 234–241. Springer, Cham (2015)
Zhou, Z., et al.: UNet++: a nested U-Net architecture for medical image segmentation. Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, pp. 3–11. Springer, Cham (2018)
Çiçek, Ö., et al.: 3D U-Net: learning dense volumetric segmentation from sparse annotation. In: International conference on medical image computing and computer-assisted intervention, pp. 424–432. Springer, Cham (2016)
Oktay, O., et al.: Attention U-Net: Learning where to look for the pancreas. arXiv preprint arXiv:1804.03999 (2018)
Liu, Z., et al.: Deep learning based brain tumor segmentation: a survey. arXiv preprint arXiv:2007.09479 (2020)
Lei, T., et al.: Medical image segmentation using deep learning: a survey. arXiv preprint arXiv:2009.13120 (2020)
Nair, V., Hinton, G.E.: Rectified Linear Units Improve Restricted Boltzmann Machines. ICML (2010)
Wu, Y., He, K.: Group normalization. In: Ferrari, V., Hebert, M., Sminchisescu, C., Weiss, Y. (eds.) ECCV 2018. LNCS, vol. 11217, pp. 3–19. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-01261-8_1
Liu, L., et al.: On the variance of the adaptive learning rate and beyond. arXiv preprint arXiv:1908.03265 (2019)
Zhang, M.R., et al.: Lookahead optimizer: k steps forward, 1 step back. arXiv preprint arXiv:1907.08610 (2019)
Yong, H., et al.: Gradient centralization: A new optimization technique for deep neural networks. In: European Conference on Computer Vision, pp 635–652. Springer, Cham (2020)
Ruder, S.: An overview of gradient descent optimization algorithms. arXiv preprint arXiv:1609.04747 (2016)
Tan, M., Le, Q.: Efficientnet: Rethinking model scaling for convolutional neural networks. In: International Conference on Machine Learning, pp. 6105–6114. PMLR (2019)
Sagi, O., Rokach, L.: Ensemble learning: a survey. Wiley Interdiscip. Rev.: Data Min. Knowl. Discov. 8(4), e1249 (2018)
https://www.kaggle.com/c/rsna-miccai-brain-tumor-radiogenomic-classification/overview
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
Cai, X., Lou, S., Shuai, M., An, Z. (2022). Feature Learning by Attention and Ensemble with 3D U-Net to Glioma Tumor Segmentation. In: Crimi, A., Bakas, S. (eds) Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries. BrainLes 2021. Lecture Notes in Computer Science, vol 12963. Springer, Cham. https://doi.org/10.1007/978-3-031-09002-8_6
Download citation
DOI: https://doi.org/10.1007/978-3-031-09002-8_6
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-09001-1
Online ISBN: 978-3-031-09002-8
eBook Packages: Computer ScienceComputer Science (R0)