Abstract
Brain tumour segmentation on 3D MRI imaging is one of the most critical deep learning applications. In this paper, for the segmentation of tumour sub-regions in brain MRI images, we study some popular architecture for medical imaging segmentation. We further, inspired by them, proposed an architecture that is an end-to-end trainable, fully convolutional neural network that uses attention block to learn localization of different features of the multiple sub-regions of a tumour. We also experiment with a combination of the weighted cross-entropy loss function and dice loss function on the model’s performance and the quality of the output segmented labels. The results of the evaluation of our model are received through BraTS’19 dataset challenge. The model can achieve a dice score of 0.80 for the whole tumour segmentation and dice scores of 0.639 and 0.536 for the other two sub-regions within the tumour on the validation dataset.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Casamitjana, A., Catà, M., Sánchez, I., Combalia, M., Vilaplana, V.: Cascaded V-Net using ROI masks for brain tumor segmentation. In: Crimi, A., Bakas, S., Kuijf, H., Menze, B., Reyes, M. (eds.) BrainLes 2017. LNCS, vol. 10670, pp. 381–391. Springer, Cham (2018). https://doi.org/10.1007/978-3-319-75238-9_33
Çiçek, Ö., Abdulkadir, A., Lienkamp, S.S., Brox, T., Ronneberger, O.: 3D U-Net: learning dense volumetric segmentation from sparse annotation. In: Ourselin, S., Joskowicz, L., Sabuncu, M.R., Unal, G., Wells, W. (eds.) MICCAI 2016. LNCS, vol. 9901, pp. 424–432. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46723-8_49
Litjens, G., et al.: Evaluation of prostate segmentation algorithms for MRI: the PROMISE12 challenge. Med. Image Anal. 18(2), 359–373 (2014)
Frey, M., Nau, M.: Memory efficient brain tumor segmentation using an autoencoder-regularized U-Net. In: Crimi, A., Bakas, S. (eds.) BrainLes 2019. LNCS, vol. 11992, pp. 388–396. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-46640-4_37
Giri, C., Goodwin, M., Oppedal, K.: Deep 3D convolution neural network for Alzheimer’s detection. In: Nicosia, G., et al. (eds.) LOD 2020. LNCS, vol. 12565, pp. 347–358. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-64583-0_32
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition (2015)
Kamnitsas, K., et al.: DeepMedic for brain tumor segmentation. In: Crimi, A., Menze, B., Maier, O., Reyes, M., Winzeck, S., Handels, H. (eds.) Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries. BrainLes 2016. LNCS, vol. 10154, pp. 138–149. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-55524-9_14
Malathi, M., Sinthia, P.: Brain tumour segmentation using convolutional neural network with tensor flow. Asian Pac. J. Cancer Prev. 20(7), 2095–2101 (2019)
Milletari, F., Navab, N., Ahmadi, S.-A.: V-Net: fully convolutional neural networks for volumetric medical image segmentation (2016)
Myronenko, A.: 3D MRI brain tumor segmentation using autoencoder regularization. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11384, pp. 311–320. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11726-9_28
Oktay, O., et al.: Attention U-Net: Learning where to look for the pancreas (2018)
Rohlfing, T., Zahr, N., Sullivan, E., Pfefferbaum, A.: The SRI24 multichannel atlas of normal adult human brain structure. Hum. Brain Map. 31, 798–819 (2009)
Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. In: Navab, N., Hornegger, J., Wells, W.M., Frangi, A.F. (eds.) MICCAI 2015. LNCS, vol. 9351, pp. 234–241. Springer, Cham (2015). https://doi.org/10.1007/978-3-319-24574-4_28
Thaha, M., Pradeep Kumar, K., Murugan, B., Dhanasekeran, S., Vijayakarthick, P., Selvi, A.: Brain tumor segmentation using convolutional neural networks in MRI images. J. Med. Syst. 43, 294 (2019). https://doi.org/10.1007/s10916-019-1416-0
Wang, C., MacGillivray, T., Macnaught, G., Yang, G., Newby, D.E.: A two-stage 3D Unet framework for multi-class segmentation on full resolution image. arXiv abs/1804.04341 (2018)
Wang, G., Li, W., Ourselin, S., Vercauteren, T.: Automatic brain tumor segmentation based on cascaded convolutional neural networks with uncertainty estimation. Front. Comput. Neurosci. 13, 08 (2019)
Weninger, L., Rippel, O., Koppers, S., Merhof, D.: Segmentation of brain tumors and patient survival prediction: methods for the BraTS 2018 challenge. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11384, pp. 3–12. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11726-9_1
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 Springer Nature Switzerland AG
About this paper
Cite this paper
Giri, C., Sharma, J., Goodwin, M. (2022). Brain Tumour Segmentation on 3D MRI Using Attention V-Net. In: Iliadis, L., Jayne, C., Tefas, A., Pimenidis, E. (eds) Engineering Applications of Neural Networks. EANN 2022. Communications in Computer and Information Science, vol 1600. Springer, Cham. https://doi.org/10.1007/978-3-031-08223-8_28
Download citation
DOI: https://doi.org/10.1007/978-3-031-08223-8_28
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-08222-1
Online ISBN: 978-3-031-08223-8
eBook Packages: Computer ScienceComputer Science (R0)