Radioengineering (Sep 2024)
EFU Net: Edge Information Fused 3D Unet for Brain Tumor Segmentation
Abstract
Brain tumors refer to abnormal cell proliferation formed in brain tissue, which can cause neurological dysfunction and cognitive impairment, posing a serious threat to human health. Therefore, it becomes a very challenging work to full-automaticly segment brain tumors using computers because of the mutual infiltration and fuzzy boundary between the focus areas and the normal brain tissue. To address the above issues, a segmentation method which integrates edge features is proposed in this paper. The overall segmentation architecture follows the encoder decoder structure, extracting rich features from the encoder. The first two layers of features are input to the edge attention module, and to extract tumor edge features which are fully fused with the features of the decoder segment. At the same time, an adaptive weighted mixed loss function is introduced to train the network by adaptively adjusting the weights of different loss parts in the training process. Relevant experiments were carried out using the public brain tumor data set. The Dice mean values of the proposed segmentation model in the whole tumor area (WT), the core tumor area (TC), and the enhancing tumor area (ET) reach 91.10%, 87.16%, and 88.86%, respectively, and the mean values of Hausdorff distance are 3.92, 5.12, and 1.92 mm, respectively. The experimental results showed that the proposed method can significantly improve segmentation accuracy, especially the segmentation effect of the edge part.
