Brain Tumor Segmentation Based on Refined Fully Convolutional Neural Networks with A Hierarchical Dice Loss

Abstract

As a basic task in computer vision, semantic segmentation can provide fundamental information for object detection and instance segmentation to help the artificial intelligence better understand real world. Since the proposal of fully convolutional neural network (FCNN), it has been widely used in semantic segmentation because of its high accuracy of pixel-wise classification as well as high precision of localization. In this paper, we apply several famous FCNN to brain tumor segmentation, making comparisons and adjusting network architectures to achieve better performance measured by metrics such as precision, recall, mean of intersection of union (mIoU) and dice score coefficient (DSC). The adjustments to the classic FCNN include adding more connections between convolutional layers, enlarging decoders after up sample layers and changing the way shallower layers' information is reused. Besides the structure modification, we also propose a new classifier with a hierarchical dice loss. Inspired by the containing relationship between classes, the loss function converts multiple classification to multiple binary classification in order to counteract the negative effect caused by imbalance data set. Massive experiments have been done on the training set and testing set in order to assess our refined fully convolutional neural networks and new types of loss function. Competitive figures prove they are more effective than their predecessors.

14 pages, 7 figures, 6 tables

References in corpus (1)

• Brain Tumor Segmentation with Deep Neural Networks

Cited by in corpus (6)

• Association of genomic subtypes of lower-grade gliomas with shape features automatically extracted by a deep learning algorithm
• Towards Interpretable and Robust Hand Detection via Pixel-wise Prediction
• Generative Model-Based Ischemic Stroke Lesion Segmentation
• Guided Attention Network for Object Detection and Counting on Drones
• Multi-scale Attention U-Net (MsAUNet): A Modified U-Net Architecture for Scene Segmentation
• Adaptive Class Weight based Dual Focal Loss for Improved Semantic Segmentation