Brain Tumor Segmentation of HGG and LGG MRI Images Using WFL-Based 3D U-Net

Abstract

The semantic segmentation of a brain tumor is the essential stage in medical treatment planning. Due to the different characteristics of tumors, one of the main difficulties in image segmentation is the severe imbalance between classes. Also, a dataset with imbalanced classes is a common problem in multimodal 3D brain MRIs. Despite these problems, most studies in brain tumor segmentation are biased toward the overrepresented tumor class (majority class) and ignore the small size tumor class (minority class). In this paper, we propose an improved loss function Weighted Focal Loss (WFL), based on 3D U-Net to enhance the prediction of brain tumor segmentation. Using our proposed loss function (WFL) solves the imbalance between classes and the imbalance between weights by giving higher weights to the minority and lower weights to the majority. After assigning these weights to different pixel values, our work is able to resolve pixel degradation, which is one of the limitations of the loss function during model training. Based on our experiments, the proposed function (WFL) on the 3D U-Net model for high-grade glioma (HGG) and low-grade glioma (LGG) in the Brain Tumor Segmentation Challenge (BraTS) 2019 dataset has shown promising results for tumor core (TC), whole tumor (WT) and enhanced tumor (ET) with average dice scores of HGG: 0.

Brain tumor segmentation by combining MRI multimodal information and improved Unet network

“…In this study, the public Glioma data set Brats2018 provided by MICCAI was adopted. The Brats2018 data set mainly consists of two types: High Grade Glioma (HGG) image and Low Grade Glioma (LGG) image [3]. The relevant information of the Brats18 brain tumor image dataset is shown in Table 1.…”

Brain Tumor Semantic Segmentation using Residual U-Net++ Encoder-Decoder Architecture

“…BraTS stands for Brain Tumor Segmentation, collected and prepared as a challenge per year. It is the most commonly used dataset for brain tumor segmentation as it is public [7], [8], [9], [10], [11], [12], [13], [14], [15]. It consists of a collection of MRI brain images, and all brain images are stripped of the skull and oriented similarly.…”