Brain Tumor Segmentation from Multi-Spectral MRI Data Using Cascaded Ensemble Learning

Ensemble learning methods are frequently employed in medical decision support. In image segmentation problems the ensemble based decisions require a postprocessing, because the ensemble cannot adequately handle the strong correlation of neighbor voxels. This paper proposes a brain tumor segmentation procedure based on an ensemble cascade. The first ensemble consisting of binary decision trees is trained to separate focal lesions from normal tissues based on four observed and 100 computed features. Starting from the intermediary labels provided by the first ensemble, six local features are computed for each voxel that serve as input for the second ensemble. The second ensemble is a classical random forest that enforces the correlation between neighbor pixels, regularizes the shape of the lesions. The segmentation accuracy is characterized by 85.5% overall Dice Score, 0.5% above previous solutions.