Brain Tumor Segmentation and Survival Prediction Using a Cascade of Random Forests

Brain tumor segmentation is a difficult task due to the strongly varying intensity and shape of gliomas. In this paper we propose a multi-stage discriminative framework for brain tumor segmentation based on BraTS 2018 dataset. The framework presented in this paper is a more complex segmentation system than our previous work presented at BraTS 2016. Here we propose a multi-stage discriminative segmentation model, where every stage is a binary classifier based on the random forest algorithm. Our multi-stage system attempts to follow the layered structure of tumor tissues provided in the annotation protocol. In each segmentation stage we dealt with four major difficulties: feature selection, determination of training database used, optimization of classifier performances and image post-processing. The framework was tested on the evaluation images from BraTS 2018. One of the most important results is the determination of the tumor ROI with a sensitivity of approximately 0.99 in stage I by considering only 16% of the brain in the subsequent stages. Based on the segmentation obtained we solved the survival prediction task using a random forest regressor. The results obtained are comparable to the best ones presented in previous BraTS Challenges.

[1]  Guang Yang,et al.  MRI Brain Tumor Segmentation and Patient Survival Prediction Using Random Forests and Fully Convolutional Networks , 2017, BrainLes@MICCAI.

[2]  Christos Davatzikos,et al.  Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features , 2017, Scientific Data.

[3]  et al.,et al.  Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation, Progression Assessment, and Overall Survival Prediction in the BRATS Challenge , 2018, ArXiv.

[4]  Sabine Van Huffel,et al.  Tumor Segmentation from Multimodal MRI Using Random Forest with Superpixel and Tensor Based Feature Extraction , 2017, BrainLes@MICCAI.

[5]  E. Polley,et al.  Statistical Applications in Genetics and Molecular Biology Random Forests for Genetic Association Studies , 2011 .

[6]  Lei Zhang,et al.  MRI Brain Tumor Segmentation using Random Forests and Fully Convolutional Networks , 2019, ArXiv.

[7]  Pradipta Maji,et al.  Multimodal Brain Tumor Segmentation Using Ensemble of Forest Method , 2017, BrainLes@MICCAI.

[8]  László Szilágyi,et al.  Brain Tumor Segmentation with Optimized Random Forest , 2016, BrainLes@MICCAI.

[9]  Leo Breiman,et al.  Random Forests , 2001, Machine Learning.

[10]  Mircea-Florin Vaida,et al.  Random forest feature selection approach for image segmentation , 2017, International Conference on Machine Vision.

[11]  Brian B. Avants,et al.  The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS) , 2015, IEEE Transactions on Medical Imaging.