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 条
  • [21] Fully Automated Segmentation of Brain Tumor from Multiparametric MRI Using 3D Context U-Net with Deep Supervision
    Lin, Mingquan
    Momin, Shadab
    Zhou, Boran
    Tang, Katherine
    Lei, Yang
    Curran, Walter J.
    Liu, Tian
    Yang, Xiaofeng
    MEDICAL IMAGING 2021: COMPUTER-AIDED DIAGNOSIS, 2021, 11597
  • [22] Fully automated segmentation of brain tumor from multiparametric MRI using 3D context deep supervised U-Net
    Lin, Mingquan
    Momin, Shadab
    Lei, Yang
    Wang, Hesheng
    Curran, Walter J.
    Liu, Tian
    Yang, Xiaofeng
    MEDICAL PHYSICS, 2021, 48 (08) : 4365 - 4374
  • [23] Optimized U-Net for Brain Tumor Segmentation
    Futrega, Michal
    Milesi, Alexandre
    Marcinkiewicz, Michal
    Ribalta, Pablo
    BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2021, PT II, 2022, 12963 : 15 - 29
  • [24] Tuning U-Net for Brain Tumor Segmentation
    Futrega, Michal
    Marcinkiewicz, Michal
    Ribalta, Pablo
    BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2022, 2023, 13769 : 162 - 173
  • [25] Deep Convolutional Neural Networks Using U-Net for Automatic Brain Tumor Segmentation in Multimodal MRI Volumes
    Kermi, Adel
    Mahmoudi, Issam
    Khadir, Mohamed Tarek
    BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2018, PT II, 2019, 11384 : 37 - 48
  • [26] LKAU-Net: 3D Large-Kernel Attention-Based U-Net for Automatic MRI Brain Tumor Segmentation
    Li, Hao
    Nan, Yang
    Yang, Guang
    MEDICAL IMAGE UNDERSTANDING AND ANALYSIS, MIUA 2022, 2022, 13413 : 313 - 327
  • [27] Multi-modality encoded fusion with 3D inception U-net and decoder model for brain tumor segmentation
    Narinder Singh Punn
    Sonali Agarwal
    Multimedia Tools and Applications, 2021, 80 : 30305 - 30320
  • [28] Breast tumor segmentation in ultrasound images: comparing U-net and U-net + +
    de Oliveira, Carlos Eduardo Gonçalves
    Vieira, Sílvio Leão
    Paranaiba, Caio Felipe Brito
    Itikawa, Emerson Nobuyuki
    Research on Biomedical Engineering, 2025, 41 (01)
  • [29] 3D Automatic Brain Tumor Segmentation Using a Multiscale Input U-Net Network
    Gonzalez, S. Rosas
    Sekou, T. Birgui
    Hidane, M.
    Tauber, C.
    BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES (BRAINLES 2019), PT II, 2020, 11993 : 113 - 123
  • [30] 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.
    EXPERT SYSTEMS WITH APPLICATIONS, 2023, 213