Abstract
In recent years, the fully convolutional network represented by Unet has been widely used in the field of medical image segmentation. However, due to the diversity of the shapes of lesions and the differences in the structures of different organs, the segmentation of lesions using only Unet structure cannot meet the requirements of accuracy and speed. Therefore, an improved Unet network for brain tumor segmentation is proposed. To reduce the number of parameters while extracting richer features and improving the accuracy of segmentation, this article introduces the inverted residuals block to replace the convolution module in the encoding and decoding stages of Unet to improve the calculation speed and accuracy; to better combine high-order semantic information with low-order semantic information, improve for the quality of detailed features in the training process, an improved Residuals Convolutional Block Attention Module is added between the encoder and the decoder. Combining the above two points of improvement, this article proposes an improved model based on Unet. Based on the Brats2019 dataset, an ablation experiment was performed on the proposed improved Unet model and compared with the TrUE-Net, ConResNet and OM-Net methods, and the Dice coefficient and Hausdorff distance were used as evaluation indicators to analyze the segmentation effect of the model. The experimental results show that the Dice coefficient of the improved Unet network model proposed in this article is 0.020–0.027 higher than other comparative models on average, and the Haushofer distance is reduced by 2.67–10.06.
Similar content being viewed by others
Availability of data and materials
All data included in this study are available upon request by contact with the corresponding author.
References
Gorp, M., Erp, L.M.E., Maas, A., et al.: Increased health-related quality of life impairments of male and female survivors of childhood cancer: DCCSS LATER 2 psycho-oncology study. Cancer 128(5), 1074–1084 (2021)
Miller, K.D., Ostrom, Q.T., Kruchko, C., et al.: Brain and other central nervous system tumor statistics. CA Cancer J. Clin. 71(5), 381–406 (2021)
Jiang, H., Diao, Z., Yao, Y.D.: deep learning techniques for tumor segmentation: a review. J. Supercomput. 4, 1–7 (2021)
TANG Shi. Study on segmentation algorithm of brain tumor MR image. Guangdong University of Tech-nolog., 2018: 1–65. (In Chinese)
Li, Xu., Jinniu, B., Leimin, Li.: Diagnosis of tumor based on GMM feature decision-making classification of brain MR Images. Control. Eng. 32(1), 1–6 (2008)
Frucci, M., Di Baja, G.S.: A new algorithm for image segmentation via watershed transformation. In: International conference on image analysis and processing, pp 168–177. Springer, Berlin, Heidelberg (2011)
Zhang, Y., Wu, L.: An MR brain images classifier via principal component analysis and kernel support vector machine. Progress Electromagn. Res. 130, 369–388 (2012)
Khan, M.A., Lali, I.U., Rehman, A., et al.: Brain tumor detection and classification: a framework of marker-based watershed algorithm and multilevel priority features selection. Mi-crosc. Res. Tech. 82(6), 909–922 (2019)
Menon, R.V., Raha, P., Kothari, S., et al.: Automated detection and classification of mass from breast ultrasound images. Computer Vision. India: Pattern Recognition. Image Processing & Graphics, pp 1–4 (2016)
He, Q., Shao, J., Jian, Pu., Zhou, M., Xiang, S., Wei, Su.: FRFCM clustering segmentation method for medical MR image feature diagnosis. Intell. Fuzzy Syst. 40(2), 2871–2879 (2021)
Zhang, C., Zhu, G., Lian, B., et al.: Image segmentation based on multiscale fast spectral clustering. Multimed. Tools Appl. 80(16), 24969–24994 (2021)
Saxena, S., Kumari, N., Pattnaik, S.: Brain tumour segmentation in FLAIR MRI using sliding window texture feature extraction followed by fuzzy C-means clustering. Int. J. Healthc. Inf. Syst. Inform. (IJHISI) 16(3), 1–20 (2021)
Rana, R., Bhdauria, H.S., Singh, A.: Brain tumour extraction from MRI images using bounding-box with level set method. In: Contemporary computing (IC3), India: 2013 sixth international conference on IEEE, pp 319–324 (2013)
Matsuyama, E.: A novel method for automated lung region segmentation in chest X-ray images. J. Biomed. Sci. Eng. 14(6), 288–299 (2021)
Evan, S., Jonathan, L., Trevor, D.: Fully convolutional networks for semantic segmentation. IEEE Trans. Pattern Anal. Mach. Intell. 39(4), 640–651 (2017)
Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. Lecture Notes in Computer Science, pp 234–241 (2015)
Roy, A.G., Navab, N., Navab, N., et al. Concurrent spatial and channel squeeze and excitation in fully convolutional networks. In: Proceedings of the medical image computing and computer-assisted intervention, Singapore, MICCAI 2018, pp 421–429 (2018)
Ni, Z.L., Bian, G.B., Zhou, X.H., et al.: RAUNet: residual attention U-Net for semantic segmentation of cataract surgical instruments. Lect. Notes Comput. Sci. 11954(04), 139–149 (2019)
Trebing, K., Staǹczyk, T., Mehrkanoon, S.: SmaAt-UNet: Precipitation nowcasting using a small attention-Unet architecture. Pattern Recognit. Lett. 145, 178–186 (2021)
Wei, P., Tong, M.: Attention-DPU: dual-path UNet with an attention mechanism for ultrasound image segmentation. J. Phys. Conf. Ser. 1693(01), 1–5 (2020)
Howard, A.G., Zhu, M., Chen, B., et al.: MobileNets: efficient convolutional neural networks for mobile vision applications. 16(05): 4480–4490 (2017)
Sandler, M., Howard, A., Zhu, M., et al.: MobileNetV2: inverted residuals and linear bottlenecks. IEEE /CVF Conf. Comput. Vis. Pattern Recognit. (CVPR) 16(05), 4510–4520 (2018)
Howard, A., Sandler, M., Chu, G., et al.: Searching for MobileNetV3: end-to-end object detection with transformers. IEEE/CVF Int. Conf. Comput. Vis. (ICCV) 16(05), 4570–4580 (2019)
Han, K., Wang, Y., Tian, Q., et al.: GhostNet: more features from cheap operations. IEEE/cvf conference on computer vision and pattern recognition (CVPR). IEEE, 77(02): 1577–1586 (2020)
Yu, C., Xiao, B., Gao, C., et al.: Lite-HRNet: a lightweight high-resolution network. IEEE/CVF Conf. Comput. Vis. Pattern Recognit. (CVPR) 17(04), 10435–10445 (2021)
Bakas, S., Akbari, H., Sotiras, A., et al.: Advancing the cancer genome atlas glioma MRI collections with expert segmentation labels and radiomic features. Sci. Data 06(04), 1–13 (2017)
Bakas, S., Reyes, M., Jakab, A., et al.: Identifying the best machine learning algorithms for brain tumor segmentation, progression assessment, and overall survival prediction in the BRATS challenge. Sci. Data 44(10), 1–29 (2018)
Menze, B., Jakab, A., Bauer, S., et al.: The multimodal brain tumor image segmentation benchmark (BRATS). IEEE Trans. Med. Imaging 30(10), 1993–2024 (2015)
Tustison Nicholas, J., Avants Brian, B., Cook Philip, A., et al.: N4ITK: improved N3 bias correction. IEEE Trans. Med. Imaging 29(6), 1310–1320 (2010)
Sundaresan, V., Griffanti, L., Jenkinson, M.: Brain tumour segmentation using a triplanar ensemble of U-Nets on MR images. Lect. Notes Comput. Sci. 12658, 340–353 (2021)
Zhang, J., Xie, Y., Wang, Y., et al.: Inter-slice context residual learning for 3D medical image segmentation. IEEE Trans. Med. Imaging 2, 661–672 (2020)
Zhou, C., Ding, C., Wang, X., et al.: One-pass multi-task networks with cross-task guided attention for brain tumor segmentation. IEEE Trans. Image Process. 99, 1–1 (2020)
Funding
This research received by the natural science foundation of Heilongjiang Province (No. LH2020F033) and the national natural science youth foundation of china (No.11804068).
Author information
Authors and Affiliations
Contributions
Lei Zhang and Chaofeng Lan contributed to the conception of the study and contributed significantly to analysis and manuscript preparation; Lirong Fu and Meng Zhang made important contributions in made adjustments to the structure of the paper, revised the paper, edited the manuscript and english polished; Xiuhuan Mao performeds the experiment、the data analyses and wrote the manuscript. All authors reviewed the manuscript.
Corresponding authors
Ethics declarations
Conflict of interest
The authors declare that there is no conflict of interest regarding the publication of this article.
Ethics approval and consent to participate
Not applicable.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Zhang, L., Lan, C., Fu, L. et al. Segmentation of brain tumor MRI image based on improved attention module Unet network. SIViP 17, 2277–2285 (2023). https://doi.org/10.1007/s11760-022-02443-5
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11760-022-02443-5