The main task of magnetic resonance imaging (MRI) automatic brain tumor segmentation is to automatically segment the brain tumor edema, peritumoral edema, endoscopic core, enhancing tumor core and nonenhancing tumor core from 3D MR images. Because the location, size, shape and intensity of brain tumors vary greatly, it is very difficult to segment these brain tumor regions automatically. In this paper, by combining the advantages of DenseNet and ResNet, we proposed a new 3D U-Net with dense encoder blocks and residual decoder blocks. We used dense blocks in the encoder part and residual blocks in the decoder part. The number of output feature maps increases with the network layers in contracting path of encoder, which is consistent with the characteristics of dense blocks. Using dense blocks can decrease the number of network parameters, deepen network layers, strengthen feature propagation, alleviate vanishing-gradient and enlarge receptive fields. The residual blocks were used in the decoder to replace the convolution neural block of original U-Net, which made the network performance better. Our proposed approach was trained and validated on the BraTS2019 training and validation data set. We obtained dice scores of 0.901, 0.815 and 0.766 for whole tumor, tumor core and enhancing tumor core respectively on the BraTS2019 validation data set. Our method has the better performance than the original 3D U-Net. The results of our experiment demonstrate that compared with some state-of-the-art methods, our approach is a competitive automatic brain tumor segmentation method.
Read full abstract