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

被引:0
作者
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 条
  • [31] Multi-modality encoded fusion with 3D inception U-net and decoder model for brain tumor segmentation
    Punn, Narinder Singh
    Agarwal, Sonali
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (20) : 30305 - 30320
  • [32] Edge U-Net: Brain tumor segmentation using MRI based on deep U-Net model with boundary information
    Allah, Ahmed M. Gab
    Sarhan, Amany M.
    Elshennawy, Nada M.
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2023, 213
  • [33] NDNN based U-Net: An Innovative 3D Brain Tumor Segmentation Method
    Trivedi, Sandeep
    Patel, Nikhil
    Faruqui, Nuruzzaman
    [J]. 2022 IEEE 13TH ANNUAL UBIQUITOUS COMPUTING, ELECTRONICS & MOBILE COMMUNICATION CONFERENCE (UEMCON), 2022, : 538 - 546
  • [34] Inception-UDet: An Improved U-Net Architecture for Brain Tumor Segmentation
    Aboussaleh I.
    Riffi J.
    Mahraz A.M.
    Tairi H.
    [J]. Annals of Data Science, 2024, 11 (03) : 831 - 853
  • [35] MAU-Net: Mixed attention U-Net for MRI brain tumor segmentation
    Zhang, Yuqing
    Han, Yutong
    Zhang, Jianxin
    [J]. MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2023, 20 (12) : 20510 - 20527
  • [36] 3D U-Net Based Brain Tumor Segmentation and Survival Days Prediction
    Wang, Feifan
    Jiang, Runzhou
    Zheng, Liqin
    Meng, Chun
    Biswal, Bharat
    [J]. BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES (BRAINLES 2019), PT I, 2020, 11992 : 131 - 141
  • [37] A Multi Brain Tumor Region Segmentation Model Based on 3D U-Net
    Li, Zhenwei
    Wu, Xiaoqin
    Yang, Xiaoli
    [J]. APPLIED SCIENCES-BASEL, 2023, 13 (16):
  • [38] SCAU-net: 3D self-calibrated attention U-Net for brain tumor segmentation
    Liu, Dongwei
    Sheng, Ning
    Han, Yutong
    Hou, Yaqing
    Liu, Bin
    Zhang, Jianxin
    Zhang, Qiang
    [J]. NEURAL COMPUTING & APPLICATIONS, 2023, 35 (33) : 23973 - 23985
  • [39] SCAU-net: 3D self-calibrated attention U-Net for brain tumor segmentation
    Dongwei Liu
    Ning Sheng
    Yutong Han
    Yaqing Hou
    Bin Liu
    Jianxin Zhang
    Qiang Zhang
    [J]. Neural Computing and Applications, 2023, 35 : 23973 - 23985
  • [40] MVP U-Net: Multi-View Pointwise U-Net for Brain Tumor Segmentation
    Zhao, Changchen
    Zhao, Zhiming
    Zeng, Qingrun
    Feng, Yuanjing
    [J]. BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES (BRAINLES 2020), PT II, 2021, 12659 : 93 - 103