U-net combined with CRF and anatomical based spatial features to segment white matter hyperintensities.

Abstract

White matter hyperintensities (WMH) are important biomarkers for cerebral small vessel disease and closely associated with other neurodegenerative process. In this paper, we proposed a fully automatic WMH segmentation method based on U-net architecture. CRF were combined with U-net to refine segmentation results. We used a new anatomical based spatial feature produced by brain tissue segmentation based on T1 image, along with intensities of T1 and T2-FLAIR images to train our neural network. We compared 8 forms of automated WMH segmentation methods, range from traditional statistical learnng methods to deep learning based methods, with different architecture and used different features. Results showed our proposed method achieved best performance in terms of most metrics, and the inclusion of anatomical based spatial features strongly increase the segmentation performance.