An MRI brain tumor segmentation method based on improved U-Net.

Abstract

In order to improve the segmentation effect of brain tumor images and address the issue of feature information loss during convolutional neural network (CNN) training, we present an MRI brain tumor segmentation method that leverages an enhanced U-Net architecture. First, the ResNet50 network was used as the backbone network of the improved U-Net, the deeper CNN can improve the feature extraction effect. Next, the Residual Module was enhanced by incorporating the Convolutional Block Attention Module (CBAM). To increase characterization capabilities, focus on important features and suppress unnecessary features. Finally, the cross-entropy loss function and the Dice similarity coefficient are mixed to compose the loss function of the network. To solve the class unbalance problem of the data and enhance the tumor area segmentation outcome. The method's segmentation performance was evaluated using the test set. In this test set, the enhanced U-Net achieved an average Intersection over Union (IoU) of 86.64% and a Dice evaluation score of 87.47%. These values were 3.13% and 2.06% higher, respectively, compared to the original U-Net and R-Unet models. Consequently, the proposed enhanced U-Net in this study significantly improves the brain tumor segmentation efficacy, offering valuable technical support for MRI diagnosis and treatment.