Mapping dynamic spatial patterns of brain function with spatial-wise attention.

Objective: Using functional magnetic resonance imaging (fMRI) and deep learning to discover the spatial pattern of brain function, or functional brain networks (FBNs) has been attracted many reseachers. Most existing works focus on static FBNs or dynamic functional connectivity among fixed spatial network nodes, but ignore the potential dynamic/time-varying characteristics of the spatial networks themselves. And most of works based on the assumption of linearity and independence, that oversimplify the relationship between blood-oxygen level dependence signal changes and the heterogeneity of neuronal activity within voxels.Approach: To overcome these problems, we proposed a novel spatial-wise attention (SA) based method called Spatial and Channel-wise Attention Autoencoder (SCAAE) to discover the dynamic FBNs without the assumptions of linearity or independence. The core idea of SCAAE is to apply the SA to generate FBNs directly, relying solely on the spatial information present in fMRI volumes. Specifically, we trained the SCAAE in a self-supervised manner, using the autoencoder to guide the SA to focus on the activation regions. Experimental results show that the SA can generate multiple meaningful FBNs at each fMRI time point, which spatial similarity are close to the FBNs derived by known classical methods, such as independent component analysis.Main results: To validate the generalization of the method, we evaluate the approach on HCP-rest, HCP-task and ADHD-200 dataset. The results demonstrate that SA mechanism can be used to discover time-varying FBNs, and the identified dynamic FBNs over time clearly show the process of time-varying spatial patterns fading in and out.Significance: Thus we provide a novel method to understand human brain better. Code is available athttps://github.com/WhatAboutMyStar/SCAAE.

Spatial-temporal convolutional attention for discovering and characterizing functional brain networks in task fMRI.

Functional brain networks (FBNs) are spatial patterns of brain function that play a critical role in understanding human brain function. There are many proposed methods for mapping the spatial patterns of brain function, however they oversimplify the underlying assumptions of brain function and have various limitations such as linearity and independence. Additionally, current methods fail to account for the dynamic nature of FBNs, which limits their effectiveness in accurately characterizing these networks. To address these limitations, we present a novel deep learning and spatial-wise attention based model called Spatial-Temporal Convolutional Attention (STCA) to accurately model dynamic FBNs. Specifically, we train STCA in a self-supervised manner by utilizing a Convolutional Autoencoder to guide the STCA module in assigning higher attention weights to regions of functional activity. To validate the reliability of the results, we evaluate our approach on the HCP-task motor behavior dataset, the experimental results demonstrate that the STCA derived FBNs have higher spatial similarity with the templates and that the spatial similarity between the templates and the FBNs derived by STCA fluctuates with the task design over time, suggesting that STCA can reflect the dynamic changes of brain function, providing a powerful tool to better understand human brain function. Code is available at https://github.com/SNNUBIAI/STCAE.

Multi-head attention-based masked sequence model for mapping functional brain networks.

The investigation of functional brain networks (FBNs) using task-based functional magnetic resonance imaging (tfMRI) has gained significant attention in the field of neuroimaging. Despite the availability of several methods for constructing FBNs, including traditional methods like GLM and deep learning methods such as spatiotemporal self-attention mechanism (STAAE), these methods have design and training limitations. Specifically, they do not consider the intrinsic characteristics of fMRI data, such as the possibility that the same signal value at different time points could represent different brain states and meanings. Furthermore, they overlook prior knowledge, such as task designs, during training. This study aims to overcome these limitations and develop a more efficient model by drawing inspiration from techniques in the field of natural language processing (NLP). The proposed model, called the Multi-head Attention-based Masked Sequence Model (MAMSM), uses a multi-headed attention mechanism and mask training approach to learn different states corresponding to the same voxel values. Additionally, it combines cosine similarity and task design curves to construct a novel loss function. The MAMSM was applied to seven task state datasets from the Human Connectome Project (HCP) tfMRI dataset. Experimental results showed that the features acquired by the MAMSM model exhibit a Pearson correlation coefficient with the task design curves above 0.95 on average. Moreover, the model can extract more meaningful networks beyond the known task-related brain networks. The experimental results demonstrated that MAMSM has great potential in advancing the understanding of functional brain networks.

Accurate corresponding fiber tract segmentation via FiberGeoMap learner with application to autism.

Fiber tract segmentation is a prerequisite for tract-based statistical analysis. Brain fiber streamlines obtained by diffusion magnetic resonance imaging and tractography technology are usually difficult to be leveraged directly, thus need to be segmented into fiber tracts. Previous research mainly consists of two steps: defining and computing the similarity features of fiber streamlines, then adopting machine learning algorithms for fiber clustering or classification. Defining the similarity feature is the basic premise and determines its potential reliability and application. In this study, we adopt geometric features for fiber tract segmentation and develop a novel descriptor (FiberGeoMap) for the corresponding representation, which can effectively depict fiber streamlines' shapes and positions. FiberGeoMap can differentiate fiber tracts within the same subject, meanwhile preserving the shape and position consistency across subjects, thus can identify common fiber tracts across brains. We also proposed a Transformer-based encoder network called FiberGeoMap Learner, to perform segmentation based on the geometric features. Experimental results showed that the proposed method can differentiate the 103 various fiber tracts, which outperformed the existing methods in both the number of categories and segmentation accuracy. Furthermore, the proposed method identified some fiber tracts that were statistically different on fractional anisotropy (FA), mean diffusion (MD), and fiber number ration in autism.