Currently, surgery remains the primary treatment for craniocerebral tumors. Before doctors perform surgeries, they need to determine the surgical plan according to the shape, location, and size of the tumor; however, various conditions of different patients make the tumor segmentation task challenging. To improve the accuracy of determining tumor shape and realizing edge segmentation, a U-shaped network combining a residual pyramid module and a dual feature attention module is proposed. The residual pyramid module can enlarge the receptive field, extract multiscale features, and fuse original information, which solves the problem caused by the feature pyramid pooling where the local information is not related to the remote information. In addition, the dual feature attention module is proposed to replace the skip connection in the original U-Net network, enrich the features, and improve the attention of the model to space and channel features with large amounts of information to be used for more accurate brain tumor segmentation. To evaluate the performance of the proposed model, experiments were conducted on the public datasets Kaggle_3M and BraTS2021. Because the model proposed in this study is applicable to two-dimensional image segmentation, it is necessary to obtain the crosscutting images of fair class in the BraTS2021 dataset in advance. Results show that the model accuracy, Jaccard similarity coefficient, Dice similarity coefficient, and false negative rate (FNR) on the Kaggle_3M dataset are 0.9395, 0.8812, 0.8958, and 0.007, respectively. The model accuracy, Jaccard similarity coefficient, Dice similarity coefficient, and FNR on the BraTS2021 dataset were 0.9375, 0.9072, 0.8981, and 0.0087, respectively. Compared with existing algorithms, all the indicators of the proposed algorithm have been improved, but the proposed model still has certain limitations and has not been applied to actual clinical trials. For specific datasets, the generalization ability of the model needs to be further improved. In the future work, the model will be further improved to address the aforementioned limitations.
Read full abstract