3D SE-Res-UNet: squeeze-excitation enhanced residual UNet for brain tumor segmentation in multi-modal MRI scans
摘要
Brain segmentation is crucial for analyzing and understanding complex neurological structures and functions, enabling precise diagnosis, treatment planning, and research in various medical and scientific fields. Current brain segmentation methods face challenges in accurately outlining complex structures, dealing with imaging artifacts, and adapting to individual and modality differences, which may impact accuracy and applicability.
MethodsWe propose a novel deep learning model, 3D SE-Res-UNet, for brain tumor segmentation in multi-parametric magnetic resonance imaging (mpMRI) scans. This architecture incorporates Squeeze-and-Excitation (SE) layers to enhance feature re-calibration and Residual (Res) layers for efficient information flow. The encoder utilizes SE layers to capture complex hierarchical features, while the decoder employs Res layers for precise reconstruction. Batch normalization and Leaky ReLU activation functions maintain stability and non-linearity throughout the network.
ResultsThe proposed 3D SE-Res-UNet model demonstrated high performance in medical image segmentation on the BraTS’21 dataset, achieving substantial overlap between predicted and ground truth masks as evidenced by Dice Similarity Coefficient (DSC) values of 0.8968 for Tumor Core (TC), 0.9342 for Whole Tumor (WT), and 0.8621 for Enhancing Tumor (ET).
ConclusionThe accuracy validated by favorable Hausdorff Distance (HD-95), Mean Intersection over Union (mIoU), Surface Distance metrics, and Surface Dice coefficients, underscores its effectiveness in capturing proximity and geometric agreement between predicted and true tumor surfaces. The proposed model had achieved competitive results with very low computational complexity against state-of-the-art models.