Improved MRI Brain Tumor Segmentation Method Based on UNet++
DOI:
https://doi.org/10.54097/pm55y427Keywords:
Brain Tumor Segmentation, Unet++, Residual Module, CBAM, ASPP.Abstract
Brain tumors are one of the most common tumors with high mortality. Accurate detection and quantitative evaluation of brain tumors are necessary for patient treatment. A reliable and effective method for automatic segmentation of brain tumors by magnetic resonance imaging (MRI) plays a crucial role in patient treatment. In recent years, the Unet++ network structure has shown better performance in image segmentation. However, it is deficient in small-scale feature extraction and segmentation accuracy. In this study, the AC-ResUnet++ network was proposed to address the problems of Unet++ network mentioned above, which combined residual module, convolutional block attention module (CBAM) and Atrous spatial pyramidal pooling (ASPP). The residual module can solve the problem of gradient explosion in the deep network, CBAM can focus on local feature information and enhance the feature learning ability of the network for the target range, and ASPP module can generate different receptive fields and obtain multi-scale contextual feature information. The BraTS 2018 and BraTS 2019 datasets were used to test the proposed network. The Dice similarity coefficient of AC-ResUnet++ in whole tumor, tumor core and enhanced tumor are 0.8521, 0.8696 and 0.7868, respectively. Compared with Unet++ network, the Dice similarity coefficient increased by 1.22%, 5.08% and 2.03% on the baselines. The comprehensive results demonstrate that AC-ResUnet++ performs more effectively in improving the segmentation accuracy of brain tumor.
Downloads
References
MAGADZA T, VIRIRI S. Deep Learning for Brain Tumor Segmentation: A Survey of State-of-the-Art [J]. Journal of imaging, 2021, 7(2).W.-K. Chen, Linear Networks and Systems (Book style). Belmont, CA: Wadsworth, 1993, pp. 123–135.
MOHAN G, SUBASHINI M M. MRI based medical image analysis: Survey on brain tumor grade classification [J]. Biomedical Signal Processing and Control, 2018, 39: 139-61.
TANG J, LI T, SHU H, et al. Variational-Autoencoder Regularized 3D MultiResUNet for the BraTS 2020 Brain Tumor Segmentation; proceedings of the Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries, Cham, F 2021//, 2021 [C]. Springer International Publishing.
COLMAN J, ZHANG L, DUAN W, et al. DR-Unet104 for Multimodal MRI Brain Tumor Segmentation; proceedings of the Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries, Cham, F 2021//, 2021 [C]. Springer International Publishing.
RIBALTA LORENZO P, NALEPA J, BOBEK-BILLEWICZ B, et al. Segmenting brain tumors from FLAIR MRI using fully convolutional neural networks [J]. Computer methods and programs in biomedicine, 2019, 176: 135-48.
GAN X, WANG L, CHEN Q, et al. GAU-Net: U-Net Based on Global Attention Mechanism for brain tumor segmentation [J]. Journal of Physics: Conference Series, 2021, 1861(1): 012041.
ZHOU Z, RAHMAN SIDDIQUEE M M, TAJBAKHSH N, et al. UNet++: A Nested U-Net Architecture for Medical Image Segmentation; proceedings of the Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, Cham, F 2018//, 2018 [C]. Springer International Publishing.
CHEN L-C, ZHU Y, PAPANDREOU G, et al. Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation; proceedings of the Computer Vision – ECCV 2018, Cham, F 2018//, 2018 [C]. Springer International Publishing.
HOORALI F, KHOSRAVI H, MORADI B. Automatic microscopic diagnosis of diseases using an improved UNet++ architecture [J]. Tissue & cell, 2022, 76: 101816.
ZHAO Z, CHEN K, YAMANE S. CBAM-Unet++:easier to find the target with the attention module "CBAM"; proceedings of the 2021 IEEE 10th Global Conference on Consumer Electronics (GCCE), F 12-15 Oct. 2021, 2021 [C].
HE K, ZHANG X, REN S, et al. Deep Residual Learning for Image Recognition [J]. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2016: 770-8.
WANG J, GAO J, REN J, et al. DFP-ResUNet:Convolutional Neural Network with a Dilated Convolutional Feature Pyramid for Multimodal Brain Tumor Segmentation [J]. Computer methods and programs in biomedicine, 2021, 208: 106208.
WOO S, PARK J, LEE J-Y, et al. CBAM: Convolutional Block Attention Module; proceedings of the 2018 European Conference on Computer Vision (ECCV) Cham, F 2018//, 2018 [C]. Springer International Publishing.
GUAN X, YANG G, YE J, et al. 3D AGSE-VNet: an automatic brain tumor MRI data segmentation framework [J]. BMC medical imaging, 2022, 22(1): 6.
CHEN L C, PAPANDREOU G, KOKKINOS I, et al. DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2018, 40(4): 834-48.
JHA D, SMEDSRUD P H, RIEGLER M A, et al. ResUNet++: An Advanced Architecture for Medical Image Segmentation [J]. IEEE International Symposium on Multimedia (ISM), 2019: 225-2255.
ZHAO Y, LIU Y, HUANG Z, et al. GCAUNet: A group cross-channel attention residual UNet for slice based brain tumor segmentation [J]. Biomedical Signal Processing and Control, 2021, 70: 102958.
MENZE B H, JAKAB A, BAUER S, et al. The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS) [J]. IEEE transactions on medical imaging, 2015, 34(10): 1993-2024.
BAKAS S, AKBARI H, SOTIRAS A, et al. Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features [J]. Scientific Data, 2017, 4(1): 170117.
KINGMA D, BA J J C S. Adam: A Method for Stochastic Optimization [J]. International Conference on Learning Representations, 2014.
SUDRE C H, LI W, VERCAUTEREN T, et al. Generalised Dice Overlap as a Deep Learning Loss Function for Highly Unbalanced Segmentations; proceedings of the Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, Cham, F 2017//, 2017 [C]. Springer International Publishing.
Downloads
Published
Issue
Section
License
Copyright (c) 2024 Academic Journal of Science and Technology
This work is licensed under a Creative Commons Attribution 4.0 International License.
