Radioengineering (Jun 2024)
Full-automatic Segmentation Algorithm of Brain Tumor Based on RFE-UNet and Hybrid Focal Loss Function
Abstract
Semantic segmentation of glioma and its subregions plays a critical role in the entirely clinical workflow of brain cancer diagnosis, monitoring, and treatment planning. Recently, automatic tumor segmentation has attracted a lot of attention, especially supervised learning methods based on neural networks, and the popular “U-shaped” network architecture has achieved state-of-the-art performance in many fields of medical image segmentation. Despite the success of these models, the commonly used small convolution kernel can only extract local features, and more global contextual features cannot be learned, resulting in the disappointed performance of modeling long-range information. At the same time, due to the difficulty of obtaining medical image data, and the imbalance of tumor data in which tumor usually occupies a relatively small volume compared with the background, the adverse influence on the training of the model occurs. In this paper, a novel segmentation framework including TensorMixup data augmentation, improved Receptive Field Expansion UNet (RFE-UNet) and hybrid loss function is designed. Specifically, the TensorMixup algorithm in the data preprocessing phase is used to provide more high-quality training data. In the training phase, both a RFE-UNet network and a hybrid loss function are proposed respectively. RFE-UNet network adds Receptive field expansion module based on Dilated convolution in the first three stages of skip connection, which is used to learn more local and global features. In addition, hybrid loss function is mainly composed of focal loss and focal Tversky loss，focal loss increasing the weight of fewer samples and focal Tversky loss focusing on learning the characteristics of samples with incorrect predictions，which is adopted to alleviate data imbalance. The experimental results on the BraTs2019 dataset show that the average Dice value of the proposed algorithm in the intact tumor, tumor core, and enhanced tumor region can reach 91.55%, 89.23%, and 84.16% respectively, which proves the feasibility and effectiveness of using the proposed architecture.
