Brain Tumor Segmentation Based on U-Net With Image Driven Level Set Loss

Abstract

This paper presents an approach for brain tumor segmentation based on deep neural networks. The paper proposes to utilize U-Net as an architecture of the approach to capture the fine and soars information from input images. Especially, to train the network, instead of using commonly used cross-entropy loss, dice loss or both, in this study, we propose to employ a new loss function including Level set loss and Dice loss function. The level set loss is inspired from Mumford-Shah functional for unsupervised task. Meanwhile, the Dice loss function measures the similarity between the predicted mask and desired mask. The proposed approach is then applied to segment brain tumor from MRI images as well as evaluated and compared with other approaches on a dataset of nearly 4000 brain MRI scans. Experiment results show that the proposed approach achieves high performance in terms of Dice coefficient and Intersection over Union (IoU) scores.