Dual-force convolutional neural networks for accurate brain tumor segmentation

被引:133
作者
Chen, Shengcong [1 ]
Ding, Changxing [1 ]
Liu, Minfeng [2 ]
机构
[1] South China Univ Technol, Sch Elect & Informat Engn, Guangzhou 510000, Guangdong, Peoples R China
[2] Southern Med Univ, Nanfang Hosp, Guangzhou 510515, Guangdong, Peoples R China
基金
中国国家自然科学基金;
关键词
Brain tumor segmentation; Dual-force network; Convolutional neural network; Label distribution; Post-processing; LESION SEGMENTATION; MODEL;
D O I
10.1016/j.patcog.2018.11.009
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Brain tumor segmentation from Magnetic Resonance Imaging scans is vital for both the diagnosis and treatment of brain cancers. It is widely accepted that accurate segmentation depends on multi-level information. However, exiting deep architectures for brain tumor segmentation fail to explicitly encourage the models to learn high-quality hierarchical features. In this paper, we propose a series of approaches to enhance the quality of the learnt hierarchical features. Our contributions incorporate four aspects. First, we extend the popular DeepMedic model to Multi-Level DeepMedic to make use of multi-level information for more accurate segmentation. Second, we propose a novel dual-force training scheme to promote the quality of multi-level features learnt from deep models. It is a general training scheme and can be applied to many exiting architectures, e.g., DeepMedic and U-Net. Third, we design a label distribution based loss function as an auxiliary classifier to encourage the high-level layers of deep models to learn more abstract information. Finally, we propose a novel Multi-Layer Perceptron-based post-processing approach to refine the prediction results of deep models. Extensive experiments are conducted on two most recent brain tumor segmentation datasets, i.e., BRATS 2017 and BRATS 2015 datasets. Results on the two databases indicate that the proposed approaches consistently promote the segmentation performance of the two popular deep models. (C) 2018 Elsevier Ltd. All rights reserved.
引用
收藏
页码:90 / 100
页数:11
相关论文
共 60 条
[1]  
[Anonymous], ARXIV160806993V1
[2]  
[Anonymous], P 3 INT C LEARNING R
[3]  
[Anonymous], PROC CVPR IEEE
[4]  
[Anonymous], 2015, ARXIV PREPRINT ARXIV
[5]  
[Anonymous], P MICCAI CHALL MULT
[6]  
[Anonymous], 2015, PROC CVPR IEEE
[7]  
[Anonymous], ARXIV170205970V1
[8]  
[Anonymous], ARXIV170407754V1
[9]  
[Anonymous], 2015, PROCIEEE CONFCOMPUT, DOI DOI 10.1109/CVPR.2015.7298594
[10]  
[Anonymous], 2017, P INT MICCAI BRATS C