CONSTRUCTION OF 4D NEONATAL CORTICAL SURFACE ATLASES USING WASSERSTEIN DISTANCE.

Abstract

Spatiotemporal (4D) neonatal cortical surface atlases with densely sampled ages are important tools for understanding the dynamic early brain development. Conventionally, after non-linear co-registration, surface atlases were constructed by simple Euclidean average of cortical attributes across different subjects, which leads to blurred folding patterns and therefore hampers the reliability and accuracy when registering new subjects onto the atlases. To better preserve the sharpness and clarity of cortical folding patterns on surface atlases, we propose to compute the Wasserstein barycenter, which represents a geometrically faithful population mean under the Wasserstein distance metric, for the construction of 4D neonatal surface atlases. The Wasserstein distance considers two distributions as heaps of sand, and quantifies their distance as the least cost to move all sand particles from one distribution to reshape it into the other. In our case, comparing to the direct vertex-wise Euclidean average, the Wasserstein distance takes into account the alignment of spatial distribution of cortical attributes, thus is robust to potential registration errors during atlas building. Using this method, we constructed 4D neonatal cortical surface atlases at each week, from 39 to 44 postmenstrual weeks, based on a large-scale dataset with 764 subjects. Our 4D atlases show sharper and more geometrically faithful cortical folding patterns than the atlases built by the state-of-the-art method, thus leading to boosted accuracy for spatial normalization and facilitating early brain development studies.