MLFINet : A multi-level feature interaction 3D medical image segmentation network

Chuanlin Liao,Xiaolin Gou,Kemal Polat,Jingchun Zhou,Yi Lin
DOI: https://doi.org/10.1016/j.neucom.2024.129077
IF: 6
2024-12-08
Neurocomputing
Abstract:Accurate identification of brain tumors using magnetic resonance imaging (MRI) is crucial for clinical decision-making and surgery planning. However, it is a challenging task due to the diverse and complex nature of tumors. Some existing U-Net variants are outperformed in the brain tumor segmentation tasks, but these models usually have large parameters and limited ability to extract image features. In addition, the semantic difference between encoder and decoder also affects the segmentation performance of U-Net based variant network models. To address the above problems, a new segmentation network model called MLFINet is proposed based on a 3D U-Net structure. In this model, a dual separable convolution squeeze-and-excitation (DSSE) module is first designed to extract image features more efficiently while reducing the number of model parameters, and then a multi-level feature interaction module (MFIM) is designed to enhance semantic interaction between different levels. In addition, structural similarity (SSIM) loss is considered to help the model better identify and segment complex boundaries and details. Extensive experiments are conducted on open-source datasets, and the results demonstrate that the proposed model outperforms selected baselines in terms of several metrics, achieving the dice score of 86.63%, 85.79%, 94.10% on the BraTS2018, BraTS2020 and LiTS17 dataset, respectively.
computer science, artificial intelligence
What problem does this paper attempt to address?