Cross-domain Fiber Cluster Shape Analysis for Language Performance Cognitive Score Prediction
Yui Lo, Yuqian Chen, Dongnan Liu, Wan Liu, Leo Zekelman, Fan Zhang, Yogesh Rathi, Nikos Makris, Alexandra J. Golby, Weidong Cai, Lauren J. O'Donnell
TL;DR
This work addresses whether the shape of white matter fiber clusters, derived from diffusion MRI tractography, can predict individual language performance. It introduces SFFormer, an encoder-only transformer with a multi-head cross-attention fusion module that integrates shape, microstructure, and connectivity features across 953 fiber clusters per subject. Evaluated on 1065 healthy young adults from the HCP-YA dataset, SFFormer with domain fusion outperforms a CNN and a baseline transformer in predicting TPVT and TORRT language scores, highlighting the contribution of fiber-cluster shape information. The findings imply that the geometry of white matter connections provides meaningful signals for language function and that cross-domain fusion of shape with microstructural and connectivity features enhances cognitive performance prediction.
Abstract
Shape plays an important role in computer graphics, offering informative features to convey an object's morphology and functionality. Shape analysis in brain imaging can help interpret structural and functionality correlations of the human brain. In this work, we investigate the shape of the brain's 3D white matter connections and its potential predictive relationship to human cognitive function. We reconstruct brain connections as sequences of 3D points using diffusion magnetic resonance imaging (dMRI) tractography. To describe each connection, we extract 12 shape descriptors in addition to traditional dMRI connectivity and tissue microstructure features. We introduce a novel framework, Shape--fused Fiber Cluster Transformer (SFFormer), that leverages a multi-head cross-attention feature fusion module to predict subject-specific language performance based on dMRI tractography. We assess the performance of the method on a large dataset including 1065 healthy young adults. The results demonstrate that both the transformer-based SFFormer model and its inter/intra feature fusion with shape, microstructure, and connectivity are informative, and together, they improve the prediction of subject-specific language performance scores. Overall, our results indicate that the shape of the brain's connections is predictive of human language function.
