A Two-Stage Coarse-to-Fine Brain Tumor Segmentation Framework

CHEN Hao,QIN Zhi-guang,DING Yi
DOI: https://doi.org/10.12178/1001-0548.2019285
2020-01-01
Abstract:The accurate extraction of brain tumor from magnetic resonance imaging (MRI) images is the key of clinical diagnostics and treatment planning. A two-stage coarse-to-fine brain tumor segmentation framework is proposed for brain tumor segmentation in multi-modal MRI images. There are two parts in our framework. One is coarse segmentation part, the other one is fine segmentation part. A five-classification task is performed in coarse segmentation part with a deep convolutional neural network. Four coarse probability maps are generated according the five-classification results. The fine segmentation part takes these coarse maps as the mask to guide the network to pay more attention on regions of high probability. Besides, to alleviate the data imbalance problem, there is a two-branch output structure in fine segmentation part. One branch outputs five-classification results with a mask soft-max cross entropy loss function. The other branch outputs a binary result which labels the whole tumor with a mean-square loss function. Our proposal was validated in the BRATS 2015 dataset. It can be proven that our approach achieves a competitive result.
What problem does this paper attempt to address?