LIU-NET: lightweight Inception U-Net for efficient brain tumor segmentation from multimodal 3D MRI images

被引:2
作者
Shahid, Gul e Sehar [1 ]
Ahmad, Jameel [2 ]
Warraich, Chaudary Atif Raza [3 ]
Ksibi, Amel [4 ]
Alsenan, Shrooq [4 ]
Arshad, Arfan [2 ]
Raza, Rehan [5 ]
Shaikh, Zaffar Ahmed [6 ,7 ]
机构
[1] Univ Management & Technol, Dept Artificial Intelligence, Lahore, Pakistan
[2] Univ Management & Technol, Sch Syst & Technol, Dept Comp Sci, Lahore, Pakistan
[3] COMSATS Inst Informat Technol, Dept Comp Sci, Lahore, Pakistan
[4] Princess Nourah Bint Abdulrahman Univ, Coll Comp & Informat Sci, Dept Informat Syst, Riyadh, Saudi Arabia
[5] Murdoch Univ, Sch Informat Technol, Perth, Australia
[6] Benazir Bhutto Shaheed Univ Lyari, Dept Comp Sci & Informat Technol, Karachi, Pakistan
[7] Ecole Polytech Fed Lausanne, Sch Engn, Lausanne, Switzerland
关键词
3D MRI; U-NET architecture; Deep learning; BraTS; 2021; 2020; Medical image analysis; Brain tumor segmentation; Inception-style blocks; CONVOLUTION;
D O I
10.7717/peerj-cs.2787
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Segmenting brain tumors is a critical task in medical imaging that relies on advanced deep-learning methods. However, effectively handling complex tumor regions requires more comprehensive and advanced strategies to overcome challenges such as computational complexity, the gradient vanishing problem, and variations in size and visual impact. To overcome these challenges, this research presents a novel and computationally efficient method termed lightweight Inception U-Net (LIU-Net) for the accurate brain tumor segmentation task. LIU-Net balances model complexity and computational load to provide consistent performance and uses Inception blocks to capture features at different scales, which makes it relatively lightweight. Its capability to efficiently and precisely segment brain tumors, especially in challenging-to-detect regions, distinguishes it from existing models. This Inception-style convolutional block assists the model in capturing multiscale features while preserving spatial information. Moreover, the proposed model utilizes a combination of Dice loss and Focal loss to handle the class imbalance issue. The proposed LIU-Net model was evaluated on the benchmark BraTS 2021 dataset, where it generates remarkable outcomes with a Dice score of 0.8121 for the enhancing tumor (ET) region, 0.8856 for the whole tumor (WT) region, and 0.8444 for the tumor core (TC) region on the test set. To evaluate the robustness of the proposed architecture, LIU-Net was cross-validated on an external cohort BraTS 2020 dataset. The proposed method obtained a Dice score of 0.8646 for the ET region, 0.9027 for the WT region, and 0.9092 for the TC region on the external cohort BraTS 2020 dataset. These results highlight the effectiveness of integrating the Inception blocks into the U-Net architecture, making it a promising candidate for medical image segmentation.
引用
收藏
页数:34
相关论文
共 50 条
[41]   A Multi Brain Tumor Region Segmentation Model Based on 3D U-Net [J].
Li, Zhenwei ;
Wu, Xiaoqin ;
Yang, Xiaoli .
APPLIED SCIENCES-BASEL, 2023, 13 (16)
[42]   SCAU-net: 3D self-calibrated attention U-Net for brain tumor segmentation [J].
Liu, Dongwei ;
Sheng, Ning ;
Han, Yutong ;
Hou, Yaqing ;
Liu, Bin ;
Zhang, Jianxin ;
Zhang, Qiang .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (33) :23973-23985
[43]   SCAU-net: 3D self-calibrated attention U-Net for brain tumor segmentation [J].
Dongwei Liu ;
Ning Sheng ;
Yutong Han ;
Yaqing Hou ;
Bin Liu ;
Jianxin Zhang ;
Qiang Zhang .
Neural Computing and Applications, 2023, 35 :23973-23985
[44]   MVP U-Net: Multi-View Pointwise U-Net for Brain Tumor Segmentation [J].
Zhao, Changchen ;
Zhao, Zhiming ;
Zeng, Qingrun ;
Feng, Yuanjing .
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES (BRAINLES 2020), PT II, 2021, 12659 :93-103
[45]   S3D-UNet: Separable 3D U-Net for Brain Tumor Segmentation [J].
Chen, Wei ;
Liu, Boqiang ;
Peng, Suting ;
Sun, Jiawei ;
Qiao, Xu .
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2018, PT II, 2019, 11384 :358-368
[46]   Segmentation of the Multimodal Brain Tumor Images Used Res-U-Net [J].
Sun, Jindong ;
Peng, Yanjun ;
Li, Dapeng ;
Guo, Yanfei .
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES (BRAINLES 2020), PT I, 2021, 12658 :263-273
[47]   Brain Tumor Segmentation and Survival Prediction Using Patch Based Modified 3D U-Net [J].
Parmar, Bhavesh ;
Parikh, Mehul .
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES (BRAINLES 2020), PT II, 2021, 12659 :398-409
[48]   Aggregating Multi-scale Prediction Based on 3D U-Net in Brain Tumor Segmentation [J].
Chen, Minglin ;
Wu, Yaozu ;
Wu, Jianhuang .
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES (BRAINLES 2019), PT I, 2020, 11992 :142-152
[49]   RMU-Net: A Novel Residual Mobile U-Net Model for Brain Tumor Segmentation from MR Images [J].
Saeed, Muhammad Usman ;
Ali, Ghulam ;
Bin, Wang ;
Almotiri, Sultan H. ;
AlGhamdi, Mohammed A. ;
Nagra, Arfan Ali ;
Masood, Khalid ;
ul Amin, Riaz .
ELECTRONICS, 2021, 10 (16)
[50]   An MRI brain tumor segmentation method based on improved U-Net [J].
Zhu, Jiajun ;
Zhang, Rui ;
Zhang, Haifei .
MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2024, 21 (01) :778-791