Automatic segmentation of brain glioma based on XY-Net.

Abstract

Glioma is a malignant primary brain tumor, which can easily lead to death if it is not detected in time. Magnetic resonance imaging is the most commonly used technique to diagnose gliomas, and precise outlining of tumor areas from magnetic resonance images (MRIs) is an important aid to physicians in understanding the patient's condition and formulating treatment plans. However, relying on radiologists to manually depict tumors is a tedious and laborious task, so it is clinically important to investigate an automated method for outlining glioma regions in MRIs. To liberate radiologists from the heavy task of outlining tumors, we propose a fully convolutional network, XY-Net, based on the most popular U-Net symmetric encoder-decoder structure to perform automatic segmentation of gliomas. We construct two symmetric sub-encoders for XY-Net and build interconnected X-shaped feature map transmission paths between the sub-encoders, while maintaining the feature map concatenation between each sub-encoder and the decoder. Moreover, a loss function composed of the balanced cross-entropy loss function and the dice loss function is used in the training task of XY-Net to solve the class unevenness problem of the medical image segmentation task. The experimental results show that the proposed XY-Net has a 2.16% improvement in dice coefficient (DC) compared to the network model with a single encoder structure, and compare with some state-of-the-art image segmentation methods, XY-Net achieves the best performance. The DC, HD, recall, and precision of our method on the test set are 74.49%, 10.89mm, 78.06%, and 76.30%, respectively. The combination of sub-encoders and cross-transmission paths enables the model to perform better; based on this combination, the XY-Net achieves an end-to-end automatic segmentation of gliomas on 2D slices of MRIs, which can play a certain auxiliary role for doctors in grasping the state of illness.