Learning global and cluster-specific classifiers for robust brain extraction in MR data

Abstract

We present a learning-based framework for automatic brain extraction in MR images. It accepts single or multi-contrast brain MR data, builds global binary random forests classifiers at multiple resolution levels, hierarchically performs voxelwise classifications for a test subject, and refines the brain surface using a narrow-band level set technique on the classification map. We further develop a data-driven schema to improve the model performance, which clusters patches of co-registered training images and learns cluster-specific classifiers. We validate our framework via experiments on single and multi-contrast datasets acquired using scanners with different magnetic field strengths. Compared to the state-of-the-art methods, it yields the best performance with statistically significant improvement of the cluster-specific method (with a Dice coefficient of 97.6±0.4% and an average surface distance of 0.8±0.1mm) over the global method.