Improved brain tumour segmentation using modified U-Net model with inception and attention modules on multimodal MRI images

被引:0
作者
Hechri A. [1 ,2 ]
Boudaka A. [1 ]
Hamed A. [1 ]
机构
[1] Department of Electrical and Electronics Engineering, British Applied College, Umm Al Quwain
[2] Faculty of Sciences, University of Tunis El Manar, Laboratory of Electronics and Micro-electronics, University of Monastir, Monastir
关键词
attention gate; brain tumour segmentation; fusion module; inception module; MR images; U-Net architecture;
D O I
10.1080/1448837X.2024.2309427
中图分类号
学科分类号
摘要
Brain tumours are currently recognised as one of the most dangerous diseases worldwide. Manual segmentation of brain tumours poses a challenging task heavily reliant on individual expertise, highlighting the need for the development of fully automated approaches. Deep learning techniques have gained significant popularity in recent years for medical image segmentation due to their exceptional accuracy and efficiency. In this study, we present a new brain tumour segmentation model based on the U-Net architecture. Our model incorporates an innovative inception module and a multi-level attention gate, effectively enhancing local feature expression and achieving superior segmentation accuracy. Additionally, we integrated a multiscale prediction fusion block to leverage global information across multiple scales. The performance of our proposed model was evaluated using the BRATS 2020 and BRATS 2018 datasets. On the BRATS 2018 dataset, our system achieved dice scores of 0.899, 0.851, and 0.838 for the whole tumour, tumour core, and enhancing tumour regions, respectively. Similarly, on the BRATS 2020 dataset, we obtained dice scores of 0.907, 0.842, and 0.829, respectively. Our proposed system demonstrated competitive performance across all brain tumour regions, as shown by the comparative analysis against state-of-the-art CNN models. ©, Engineers Australia.
引用
收藏
页码:48 / 58
页数:10
相关论文
共 40 条
[1]  
Aggarwal M., Tiwari A.K., Sarathi M.P., Bijalwan A., An Early Detection and Segmentation of Brain Tumor Using Deep Neural Network, BMC Medical Informaticae Decision Mak, 23, 1, (2023)
[2]  
Ahmad P., Qamar S., Shen L., Saeed A., Context Aware 3D UNet for Brain Tumor Segmentation
[3]  
Ballestar L.M., Vilaplana V., MRI Brain Tumor Segmentation and Uncertainty Estimation Using 3D-UNet Architectures, Lect Notes Comput Sci (Including Subser Lect Notes Artif Intell Lect Notes Bioinformatics), 12658, pp. 376-390, (2021)
[4]  
Ben Naceur M., Akil M., Saouli R., Kachouri R., Fully Automatic Brain Tumor Segmentation with Deep Learning-Based Selective Attention Using Overlapping Patches and Multi-Class Weighted Cross-Entropy, Medical Image Analysis, 63, (2020)
[5]  
Castillo L.S., Daza L.A., Rivera L.C., Arbelaez P., Volumetric multimodality neural network for brain tumor segmentation, 10572, pp. 97-104, (2017)
[6]  
Havaei M., Davy A., Warde-Farley D., Biard A., Courville A., Bengio Y., Pal C., Et al., Brain Tumor Segmentation with Deep Neural Networks, Medical Image Analysis, 35, pp. 18-31, (2017)
[7]  
Henry T., Carre A., Lerousseau M., Estienne T., Robert C., Paragios N., Deutsch E., Brain tumor segmentation with self-ensembled, deeply-supervised 3D U-net neural networks: a BraTS 2020 challenge solution, Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries Lecture Notes in Computer Science, 12658, pp. 227-339, (2020)
[8]  
Huang D., Wang M., Zhang L., Li H., Ye M., Li A., Learning Rich Features with Hybrid Loss for Brain Tumor Segmentation, BMC Medical Informaticae Decision Mak, 21, pp. 1-13, (2021)
[9]  
Hu H.X., Mao W.J., Lin Z.Z., Hu Q., Zhang Y., Multimodal Brain Tumor Segmentation Based on an Intelligent UNET-LSTM Algorithm in Smart Hospitals, ACM Trans Internet Technol, 21, 3, pp. 1-14, (2021)
[10]  
Hu Y., Xia Y., 3D Deep Neural Network-Based Brain Tumor Segmentation Using Multimodality Magnetic Resonance Sequences, Lect Notes Comput Sci (Including Subser Lect Notes Artif Intell Lect Notes Bioinformatics), 10670, pp. 423-434, (2017)