Spatial Warping Network for 3D Segmentation of the Hippocampus in MR Images

Abstract

Accurate segmentations of neuroanatomical structures are essential for volumetric and morphological assessment, but manual segmentation is time-consuming and error-prone. We propose a convolutional neural network for structural segmentation based on deformation of an example mask that is disease-state agnostic, which we apply to the hippocampus. The hippocampus is one of the first subcortical structures affected by Alzheimer’s disease, atrophying as the disease progresses. As the disease state may be unknown, and due to the varying degrees of atrophy, an accurate shape prior is not always available. The proposed network is based on an adapted spatial transformer network that learns a deformation field to resample an initial binary mask, to create an output segmentation. This segmentation is learnt by the network from the input T1-weighted MRI in an end-to-end manner. Experiments on the HarP dataset show that the network outperforms other segmentation methods and is consistent across disease states, independent of the degree of disease-related atrophy. We also explore the effect of the initial binary mask on the segmentation, showing that the it is insensitive to the size and initial location of the binary mask.