Semantic-spatial feature-fused cortical surface parcellation: a scale-unified spatial learning network with boundary contrastive loss.

Abstract

The cortical surface parcellation provides prior guidance for studying mental disorders and human cognition. Graph neural networks (GNNs) have gained popularity in this task to preserve its spatial structure. However, previous GNNs struggled to effectively exploit the information contained in the complex spatial structure of the cortical surface and generally encountered an uneven node distribution issue. Meanwhile, labeling boundary nodes was also identified as a widespread problem in this task. Accordingly, this paper develops a scale-unified spatial learning network with a boundary contrastive loss (SSLNet) for cortical surface parcellation. Its core is the scale-unified spatial learning module. It devises neighbor feature extraction and aggregation strategies by fully integrating spatial coordinates and semantic structure to learn effective spatial features of local neighborhoods. More importantly, spatial scale unification is incorporated into this module to mitigate the negative effect on spatial learning caused by node distribution differences among local areas. Additionally, a universal boundary contrastive loss is constructed, enhancing the feature discriminability of boundary nodes by constraining them to be close to the same class nodes and apart from different class nodes in the feature space. It considerably improves boundary performance without increasing parameters or changing the network structure. Experiments regarding public Mindboggle demonstrate that the dice score and accuracy of SSLNet achieve $89.8\%$ and $90.89\%$ , respectively, surpassing existing methods.