Skip to main content

ORIGINAL RESEARCH article

Front. Neurosci., 21 December 2023
Sec. Brain Imaging Methods
This article is part of the Research Topic Deep Learning Methods and Applications in Brain Imaging for the Diagnosis of Neurological and Psychiatric Disorders View all 9 articles

TSP-GNN: a novel neuropsychiatric disorder classification framework based on task-specific prior knowledge and graph neural network

  • 1Anhui Province Key Laboratory of Medical Physics and Technology, Institute of Health and Medical Technology, Hefei Institutes of Physical Science, Chinese Academy of Sciences, Hefei, China
  • 2University of Science and Technology of China, Hefei, China
  • 3Hefei Cancer Hospital, Chinese Academy of Sciences, Hefei, China

Neuropsychiatric disorder (ND) is often accompanied by abnormal functional connectivity (FC) patterns in specific task contexts. The distinctive task-specific FC patterns can provide valuable features for ND classification models using deep learning. However, most previous studies rely solely on the whole-brain FC matrix without considering the prior knowledge of task-specific FC patterns. Insight by the decoding studies on brain-behavior relationship, we develop TSP-GNN, which extracts task-specific prior (TSP) connectome patterns and employs graph neural network (GNN) for disease classification. TSP-GNN was validated using publicly available datasets. Our results demonstrate that different ND types show distinct task-specific connectivity patterns. Compared with the whole-brain node characteristics, utilizing task-specific nodes enhances the accuracy of ND classification. TSP-GNN comprises the first attempt to incorporate prior task-specific connectome patterns and the power of deep learning. This study elucidates the association between brain dysfunction and specific cognitive processes, offering valuable insights into the cognitive mechanism of neuropsychiatric disease.

1 Introduction

Neuropsychiatric disorder (ND) defines a wide range of psychiatric symptoms accompanying specific emotional, memory, social, or other cognitive impairments (Eddy, 2019; Porcelli et al., 2019; Jahn et al., 2021). Different subtypes of diseases, such as attention-deficit/hyperactivity disorder (ADHD) (Zepf et al., 2019), autism spectrum disorder (ASD) (Vaidya et al., 2020; Wadhera and Kakkar, 2020), and schizophrenia (SZ) (Ioakeimidis et al., 2022; Riedel et al., 2022) show abnormal brain activity during specific task context compared to healthy controls. Mental disorder diagnosis using neuroimaging and machine learning is thus promising (Lanillos et al., 2020; Perez et al., 2021).

Recent years have seen explosive growth in applying deep learning to facilitate ND classification (Chan et al., 2019; Canario et al., 2021; Liu et al., 2021). Previous studies often use brain functional connectivity (FC) or graph theory features (Farahani et al., 2019) and build convolutional neural networks (CNNs) for disease classification (Kim et al., 2016; Guo et al., 2017). However, brain networks are generally irregular and non-Euclidean structures, which can be better captured by graph neural networks (GNNs) than CNNs (Parisot et al., 2017; Zhang et al., 2020; Li L. et al., 2021; Li X. et al., 2021; Zhao et al., 2022). The benefit of GNN is due to the peculiarities of the message-passing mechanism on the graph (Ying et al., 2019). A pioneering study by Parisot and colleagues integrated the FC matrix and phenotype information to construct a sparse graph that captures participants’ relationships (Parisot et al., 2017). Subsequently, various graph structures (Li L. et al., 2021) and graph modules, such as graph pooling (Li X. et al., 2021) and even dynamic graph strategies (Zhao et al., 2022), have been proposed, significantly enhancing GNN models for neuropsychiatric disease classification. These models utilize node pooling or edge convolution layers to selectively aggregate important node features, thereby providing insights into relevant diseases from a regional perspective within the brain. For example, default mode network (DMN) and memory-associated brain regions have been identified as biological markers of ASD (Li X. et al., 2021), while damage to the DMN associated with occipital and frontal lobes may explain ADHD (Zhao et al., 2022).

The whole-brain resting-state FC matrix contains redundant and spurious correlations because of confounding or collider effects (Sanchez-Romero and Cole, 2021). It is thus valuable to extract and define distinct connectivity patterns specific to certain cognitive contexts. Recent studies have demonstrated that task-state FC patterns play an essential role in dynamically reshaping brain networks and modulating the flow of neural activity during task performance (Cole et al., 2021; Hearne et al., 2021). These task-related changes in brain network activity provide valuable prior knowledge for understanding the mechanisms underlying brain disorders (Briend et al., 2019; Xia et al., 2019; Kofler et al., 2020; Riedel et al., 2022). However, previous research on ND classification often overlooked this valuable prior information (Gupta et al., 2022; Jiang et al., 2022).

Decoding studies on brain-behavior relationships provide an insightful framework (Jiang et al., 2020; Finn, 2021). We hypothesize that incorporating prior knowledge of task-specific connectivity patterns can improve the performance of ND classification. Motivated by the underlying association between brain decoding and disease diagnosis, the present study seeks to integrate task-specific prior (TSP) knowledge (task-specific functional connectivity) and GNN into a ground-breaking framework for detecting neuropsychiatric disease, dubbed TSP-GNN. We use the Elastic-Net regression model to decode task-specific brain connectome patterns from task-state fMRI in healthy people. Then, task-specific connectome patterns were migrated to illness classification using resting-state fMRI. Finally, we build a population-based graph convolution network to detect brain disease in two neuropsychiatric datasets. The brain decoding approach reduces the dimension of the brain network while providing interpretive information relevant to the task context. Our results demonstrate that task-specific connectome improves disease categorization compared to whole-brain nodes and sheds light on the relationship between brain pathology and specific cognitive processes.

2 Materials and methods

2.1 Participants

2.1.1 HCP dataset

The Human Connectome Project (HCP) (Van Essen et al., 2013) is a remarkable and widely available dataset aimed at defining the anatomical and functional interconnection of the human brain. This dataset contains high-resolution structural MRI, resting-state fMRI, task fMRI scans, and detailed behavioral information for over 1,000 healthy individuals. Subjects completed seven scanner tasks: motor execution, language, emotion, social cognition, working memory (WM), relational, and gambling-related processes. The seven tasks, which lasted for about 20–30 frames under different conditions during each block, and the detailed task paradigm were described in Supplementary Table S1.

2.1.2 Neuropsychiatric dataset

The present study consisted of two datasets, ADHD1 and ABIDE,2 for the investigation of disease classification. The ADHD dataset consists of eight cohorts of structural MRI and resting-state fMRI scans (Bellec et al., 2017). Similarly, the ABIDE dataset has the same acquisition modalities from 20 data sites (Cameron et al., 2013). To address the potential impact of heterogeneity in equipment and scanning parameters across different sites, we selected five data sites for the ADHD dataset and three for the ABIDE dataset. Demographic information for the two datasets mentioned above can be found in Table 1.

TABLE 1
www.frontiersin.org

Table 1. Demographic and clinical characteristics of ADHD and ABIDE datasets.

2.2 fMRI data preprocessing

To ensure the reproducibility of our investigation, we utilized preprocessed fMRI results from ConnectomeDB as a basis for our subsequent analysis. We applied restricted data usage to exclude any influence of inter-individual synchronization among participants within the same family, and finally, 473 unrelated individuals were included. Additionally, we obtained two neuropsychiatric datasets that offered a standard preprocessing workflow. These datasets were directly accessible from their respective data buckets. The preprocessing of fMRI data involves numerous steps to clean and standardize the data prior to statistical analysis. All preprocessing is conducted using fMRIPrep (Esteban et al., 2019), a best-in-breed workflow that ensures high-quality preprocessing to address the challenges of robust and reproducible fMRI data preparation. The minimal preprocessing steps defined by fMRIPrep include motion correction, field unwarping, normalization, bias field correction, and brain extraction.

Subsequently, we conducted a fist-level analysis on each task-state fMRI within HCP using the general linear model (GLM). Our study used the ‘3dDeconvolve’ command in AFNI v20.3.02 to perform first-level GLM analysis. Specifically, the ‘-stim_times_FSL’ parameter was used to specify the timing of stimulus events, while the ‘-stim_file’ parameter was employed to include six head motion parameters. The ‘-mask’ parameter was also used to specify the brain mask generated by fMRIprep. The total number of stimuli ‘-num_stimts’ represented the sum of task conditions and head motion directions. All these parameters collectively constitute the design matrix for each task type, which consists of columns for each condition, nuisance variables, and a constant term, with rows corresponding to each time point of the fMRI data acquisition. The specifics of the design matrix vary according to the exact nature and timing of the task conditions within each of the seven tasks in the HCP dataset. After GLM analysis, we obtained the distribution of brain activation under different task conditions and the purified fMRI time series, devoid of noise signals from task events and motion parameters, which can enable us to investigate the neural correlates of the tasks accurately (Spencer et al., 2022).

We utilized the ‘3dNetCorr’ command by AFNI v20.3.02 to calculate the FC matrixes for both HCP and neuropsychiatric datasets based on the fMRI time series residual preprocessed by GLM. The command will calculate the correlation matrix between the time series of each pair of ROIs defined by parameter ‘-in_rois.’ The average time series and the functional connections between brain regions can be found in the destination file. The atlas adopted in our research was the Brainnetome Atlas (Fan et al., 2016), which has been extensively employed in various clinical studies (Li et al., 2020; Lee et al., 2021). The atlas consists of 246 distinct brain areas that have been carefully delineated. These brain regions can be parcellated into eight functional subsystems (Jiang et al., 2020; Lee et al., 2021). For more details on the names of brain regions in the atlas and their corresponding network allocation, please refer to Supplementary Table S2.

2.3 HCP behavioral performance

Due to the HCP dataset consisting of seven task fMRI scans covering various cognitive abilities, we employed corresponding performance measures as markers of these abilities. For the social task, we used the ratio of precious divided by the median response time (median_RT) under random mode. Working memory ability was evaluated using the accuracy (Acc) divided by the Median_RT score under the 2-back conditions. Emotion reflection performance was assessed using the Acc/Median_RT ratio. In the language task, the story condition was selected to indicate language competence, as performance under both story and math conditions showed a substantial association. However, no significant performance-related markers were detected for the gambling and motor tasks. We used the delay discounting measure to approximate the gambling task performance involving impulsive decision-making. Specifically, we calculated the difference in the area under the curve (AUC) scores between DDisc_AUC_40k and DDisc_AUC_200 as the gambling task score (Cai et al., 2020). A smaller AUC value indicates a higher degree of decision impulsivity. For the motor task, which does not quantitatively reflect participants’ athletic ability, we substituted the endurance measure obtained from the NIH Toolbox 2-Minute Walk Test.

In addition to the task-based fMRI, we considered resting-state fMRI, which reflects a baseline state of cognitive ability without task requirements. We utilized general ability (intelligence) measures related to reasoning, problem-solving, abstract thinking, planning, and learning. These measures, which reflect individual cognitive skills like brain fingerprint, were combined into a general factor score using exploratory factor analysis (Dubois et al., 2018; Thiele et al., 2022). Task performance indicators and their corresponding calculations for all fMRI tasks mentioned above can be found in Supplementary Table S3.

2.4 Task-specific functional connectome decoding based on corresponding behavioral performance

Acknowledging the advantages of the task-state connectome in predicting cognitive traits, we constructed eight models to decode task-specific brain connectome patterns across various fMRI tasks. By incorporating task performance as a driving factor, we aimed to reveal the brain connectivity patterns that contribute to cognitive traits and potentially improve our understanding of the neural mechanisms underlying these traits. Considering the superior performance of classical linear regression methods in terms of computational efficiency and their ability to capture complex brain-behavior relationships (Sui et al., 2020; Kim et al., 2021), we developed a task performance-driven brain decoding model utilizing the Elastic-net algorithm:

min β i = 1 n ( f ( x i ) y i ) 2 + λ j = 1 p ( α | β j | + 1 2 ( 1 α ) β j 2 )     (1)

The Elastic-net algorithm is known for handling high-dimensional data and selecting relevant features. The above formula, λ represents the weight coefficient of the linear regression and regularization terms, while α determines the balance between the L1 (Lasso regression) and L2 (Ridge regression) norms. For α = 0 , the model is equivalent to ridge regression, and for α = 1 , it becomes equivalent to lasso regression. The weight coefficients assigned to the features in the Elastic-Net model can quantify the contribution of FC pairs between different brain regions to predicting cognitive traits. To construct our brain functional decoding models, we tailored them for each specific fMRI state (as depicted in the top half of Figure 1). Initially, we screened out edges highly correlated with connectome strength. Subsequently, we employed a 10-fold cross-validation approach to creating regression models to decipher task-specific connectivity patterns. By aggregating the non-zero coefficients obtained from each fold in the Elastic-Net model, we obtained a functional subnetwork that best reflected the specificity of the given task (Caunca et al., 2021). To assess the reliability of the prediction outputs, we combined the predictors from each fold and performed a permutation test. Specifically, we calculated the Pearson correlation coefficient between the predicted and observed (random shuffled) scores. The permutation test probability was determined by evaluating the frequency of correlation coefficients in a set of 10,000 permutations that exceeded the initial coefficient.

FIGURE 1
www.frontiersin.org

Figure 1. The architecture overview of the proposed TSP-GNN framework which combines task-specific patterns for disease diagnosis. The top half of the framework decodes task patterns based on cognitive performance, while the bottom extracts task-specific functional connectome and graph theoretical measures from various disease datasets. Subsequently, phenotypic information is integrated to construct the population-based graph neural network to achieve disease classification.

2.5 Graph theory measures the connectome

Changes in graph theory measures of brain connectome have been recognized as significant aspects of various brain diseases (Savanth et al., 2022). By quantifying the graph-theoretical properties, researchers can gain insights into the essential brain regions and unravel the underlying organizational principles of the brain network (Fallahi et al., 2021; Zhang T. et al., 2021; Zamani et al., 2022). Our investigation included several graph theory measures as supplementary features for disease classification. These measures, namely graph strength, clustering coefficient, local efficiency, page rank centrality, betweenness centrality, eigenvector, flow coefficient, and k-coreness centrality, were calculated based on binary or weighted graphs after implementing a sparsity threshold (Wang B. et al., 2022). The ideal sparse brain graphs were constructed by optimizing the global brain efficiency, and the graph theory features extracted from the corresponding task-specific brain nodes.

2.6 Task-specific prior-knowledge graph neural network model

The population and brain parcellation methods are two commonly used GNN frameworks for diagnosing brain diseases. The population graph methodology involves constructing a graph representation at the population level (Parisot et al., 2017, 2018), while the brain-level graph methodology focuses on building graphs based on individual brain connectivity patterns (Felouat and Oukid, 2020; Wang L. et al., 2021). In our study, we employed a population GNN for further computations after decoding task-specific brain regions (as shown in the bottom half of Figure 1). We chose the population GNN approach due to its superior classification performance demonstrated in previous studies (Pan J. et al., 2022). Neuropsychological scale score, gender, or age were considered as the set of non-imaging phenotypic features N = ( N h ) . The adjacency weights of the population graph were defined as follows:

W ( x , y ) = S i m ( A x , A y ) h = 1 H γ ( N h ( x ) , N h ( y ) )     (2)
where S i m ( A x , A y ) is a similarity measure between subjects x and y , γ is the distance between phenotypic measures. For every category in h, we adopt a threshold θ and define γ as a unit-step function:
γ ( N h ( x ) , N h ( y ) ) = { 1 , i f | N h ( x ) N h ( y ) | < θ 0 , otherwise     (3)

The similarity of graph features was defined as:

S i m ( A x , A y ) = exp ( [ ρ ( F ( x ) , F ( y ) ) ] 2 2 σ 2 )     (4)

Where ρ is the correlation distance, and σ determines the width of the kernel. Due to network connectivity and graph theory measures based on the interconnected nodes on both sides of the edges to form subnetworks, the features remain in a relatively high dimension. We adopt a ridge classifier to perform recursive feature elimination (RFE) with a fixed number of features (Ravishankar et al., 2016). In the graph convolutional component of the TSP-GNN model, the normalized graph Laplacian function of a weighted graph G = ( { V , E , W } ) is defined as L = I N D 1 / 2 W D 1 / 2 where I N and D are, respectively, the identity matrix of size N N and diagonal degree matrix. The GNN architecture is derived from (Parisot et al., 2018), consists with L fully convolutional hidden layers activated using the Rectified Linear Unit (ReLU) function.

g θ Χ = g θ ( L ) Χ = g θ ( U Λ U T ) Χ = U g θ ( Λ ) U T Χ     (5)

The input layer encompasses the entire population graph, while a SoftMax activation function follows the output layer. To evaluate the performance of our model, we employed a five-fold cross-validation approach across all databases. During training, the training fold consisted of a subset of tagged graph nodes, the loss function was assessed, and gradients were backpropagated on this subset.

2.7 Compare with other classification methods

The current research comprehensively compared the TSP-GNN method with various machine learning techniques, deep learning models, and graph neural networks. Specifically, the comparison included support vector machine (SVM), K-nearest neighbor (KNN), and several ensemble learning methods. In addition, we included two deep neural networks (DNN) methods, namely multilayer perceptron (MLP) and convolutional neural networks (CNN). The MLP method, a supervised feedforward neural network which consists of one hidden layer, was connected to the stacked autoencoder (Parisot et al., 2018). The CNN method uses the most classical design, using dropout and linear layers to achieve reduction and forecast. As for the GNN model, we employed MAGE and EV-GNN, which have demonstrated superior performance in previous studies. It is worth noting that the original MAGE utilizes a variety of brain atlas features to improve the accuracy of disease diagnosis (Wang Y. et al., 2022). We adopted this concept in our paper to effectively integrate relevant prior information from multiple task modalities. Additionally, the EV-GNN model demonstrated the ability to automatically integrate imaging data and phenotype data within a learnable adaptive population graph (Huang and Chung, 2020).

3 Results

3.1 Functional connectivity patterns of different cognitive tasks

Our study demonstrates that brain connectivity patterns exhibit both task-specific characteristics and commonalities. We observed that the decoded edges traverse multiple functional brain regions and are distributed across various intrinsic resting-state networks (RSNs), indicating shared patterns across different tasks. The assessment metrics presented in Table 2 indicate the strength of the decoding results, with all expected correlation coefficients (r values) exceeding 0.3 and the corresponding value of ps being less than 0.05. Notably, we found that the prediction models for all tasks passed the permutation test, confirming the reliability and consistency of our decoding results (Figure 2). In addition to the permutation test, we employed several evaluation measures to assess the performance of the decoding models. These measures included the mean squared error (MSE), explained variance score (EVS), and mean absolute error (MAE). By examining these metrics, we gained further insights into the accuracy and precision of our prediction models.

TABLE 2
www.frontiersin.org

Table 2. Prediction and evaluations of various cognitive abilities.

FIGURE 2
www.frontiersin.org

Figure 2. Results of permutation tests on task-state and resting-state fMRI decoding. The green histograms illustrate the correlation values’ distribution between the predicted task performances and those obtained from 10,000 permutation tests. The red line marks the correlation from the predictions of the original Elastic-Net model to the actual outcomes, clearly showing that the permutation test outcomes systematically register below the baseline correlation.

3.2 Anatomical and functional localization of task-specific network edges

Significant interconnections were identified by analyzing the non-zero coefficients in the Elastic-Net model. Our analysis results revealed the most prominent interconnections associated with each task state, with the following number of edges identified: emotion (47 edges), gambling (46 edges), language (21 edges), motor task (15 edges), relational (27 edges), social (44 edges), working memory (22 edges), and rest (99 edges). Importantly, it was observed that the seven task-specific regions were widely distributed across different anatomical locations, and the number of specific edges involved in rest-state fMRI was greater than that in task fMRI. A circular diagram has depicted the distribution of the essential connected edges of social cognition and gambling tasks (Figure 3). The specific connectivity distribution patterns of the brain networks for the other five tasks and resting-state fMRI are presented in Supplementary Figures S1, S2.

FIGURE 3
www.frontiersin.org

Figure 3. FCs with the best task performance prediction capability. The nodes and edges of the brain network are created by averaging the FC strength of a particular task across all people, and the strength determines the node size and edge thickness. Connections within a module are depicted using the same color as the module in which it is situated, whereas gray lines represent inter-module connections.

The social task-related FC patterns were distributed inter-LIM-VIS, LIM-SUB, VAN-SUB networks, and intra-DMN and SUB networks. In the gambling task, participants were asked to guess the number of a mystery card. Decoding results showed significant ROIs, such as inter-insular subsystem, angular gyrus (IPL_L_6_2), supramarginal gyrus (IPL_L_6_3), superior parietal lobule (SPL), precuneus (Pcun_L_4_1), right cuneus (Cun_R_5_3, Cun_R_5_4). From the RSN perspective, brain edges related to gambling or risk decision were mainly distributed inter- SMN-VIS, DAN-VIS, DAN-FPN, and LIM-SUB networks (Figure 4), indicating a broader cross-network interaction. In the resting state, the FC pattern has the highest number of brain edges and almost exhibits the highest proportion of connections within brain anatomical locations. Resting-state fMRI predicts general intelligence, which includes reasoning, problem-solving, abstract thinking, planning, and learning, in our model.

FIGURE 4
www.frontiersin.org

Figure 4. The distribution of functional brain networks associated with edges differs across decoding modes of task states. DAN, dorsal attention network; DMN, default mode network; FPN, frontoparietal network; LIM, limbic network; SMN, somatomotor network; SUB, subcortical network; VAN, ventral attention network; VIS, visual network.

3.3 Task-specific brain connectivity for disease classification

To evaluate the impact of task-specific prior knowledge on brain disease classification, we extracted a subnetwork comprising all the nodes involved in the task-based connectome. Additionally, we incorporated graph-theoretical properties of these task-specific nodes derived from binary and weighted brain network analyzes. These steps allowed us to amalgamate FC strength with graph metrics, culminating in a refined set of input features for the GNN model. This particular methodology facilitated a comprehensive exploration of the influence of prior knowledge on disease categorization. Notably, demographics and behavioral statistics are also incorporated into the construction of the population graph. Table 3 presents the classification performance ranking by task paradigm of each dataset. The findings suggest that the classification of different types of mental illnesses exhibited a preference for the specific task prior knowledge. As two prevalent neurodevelopmental disorders, ASD and ADHD frequently co-occur. Interestingly, they exhibited distinct task preferences in classification tasks. For ADHD, task-specific features related to social and relational processing tasks can achieve higher classification accuracy. In contrast, the ABIDE dataset has shown that gambling, motor, and relational processing are the top three task-specific patterns that yielded the best classification performance.

TABLE 3
www.frontiersin.org

Table 3. The implications of priori information decoded by different tasks on neuropsychiatric disease classification.

3.4 Investigate the categorization effect of various task combination models

We further conducted task-specific prior knowledge experiments on disease classification to evaluate previous task information’s influence on disease classification and investigate if information complementarity between tasks may enhance diagnosis performance. We selected four, five, and six tasks from seven different task categories to create diverse combinations, C 7 4 , C 7 5 and C 7 6 . We presented the top three ranking AUC results for each combination of task quantities, as shown in Table 4. Our findings reveal that C 7 4 yields the best classification performance, whereas increasing the accuracy of C 7 5 and C 7 6 .

TABLE 4
www.frontiersin.org

Table 4. The effects of task decoding information combination patterns on neuropsychiatric disease classification.

From the perspective of the classification effect of the combination mode, brain diseases exhibit differential task combination preferences. Specifically, the combination of M_R_S_W achieved the best classification results on the ADHD dataset. Not exactly consistently, the combination of E_G_S_W performed best on the ABIDE dataset. Compared with single-task experiments, the classification performance is slightly improved by selecting task-specific information for combinations. Additionally, the types of tasks frequently appearing in the 4-task combination also perform well in single-task experiments.

We displayed the task-specific brain node interactions effect for best task combinations under ADHD and ABIDE datasets (Figure 5). The best task combinations for these two diseases involve working memory and social cognition. In ADHD, social cognition and working memory tasks contribute the most nodes, whereas gambling and social cognition do in ABIDE. Table 5 shows that when all ROIs are included, i.e., FC features (246 * 245/2 = 30,135) or graph theory features (15 attributes, 246 * 15 = 3,690), the classification accuracy decreases, further highlighting the superiority of task-specific nodes.

FIGURE 5
www.frontiersin.org

Figure 5. Visualizing the distribution set of nodes involved in the optimal task combination: (A) for ADHD dataset, (B) for ABIDE dataset. The findings indicate that the decoded results (node distribution) are relatively independent, with a low proportion of nodes belonging to the intersection of multiple tasks.

TABLE 5
www.frontiersin.org

Table 5. Comparing the classification performance of task-specific features and whole brain features during two datasets.

3.5 Comparison results with other baseline models

In this present investigation, various machine learning and deep learning methods were used to illustrate the superiority of the TSP-GNN model in ND diagnosis. To ensure the uniformity of input features, we conducted experiments using the optimal task combination stated in section 3.4. In the classification experiments of ADHD and ABIDE datasets, TSP-GNN has obtained the optimal results (Figure 6), and the detailed numerical values of the classification results can be found in Supplementary Tables S2, S3. In comparison to classic machine learning approaches such as SVM (Abraham et al., 2017) and ensemble learning (Liu et al., 2020), GNN models the individual-based topologies structure (Zhou and Zhang, 2021) between subjects utilizing participant similarity, which is advantageous for enhancing classification performance. After numerous layers of graph convolution computation, highly relevant characteristics are continually aggregated (Wang L. et al., 2021). MLP and CNN apply fully-connected and convolutional layers to achieve dimensionality reduction on brain network features, which are spatial topological graphs between brain areas and cannot be equated to the image receptive field (Dvornek et al., 2017; Khosla et al., 2018). The TSP-GNN architecture blends multi-task information from FC characteristics and graph measures to collect better and characterize the most discriminative information than typical machine learning and deep neural network models.

FIGURE 6
www.frontiersin.org

Figure 6. On the ADHD (A) and ABIDE (B) datasets, the classification performance of the TSP-GNN framework was compared to that of various machine learning and deep learning methods. The task priors used were combinations of the best four, five, and six combinations described in section 3.3. M_R_S_W stands for a task combination of motor, relational processing, social cognitive, and working memory tasks, and the remaining acronyms are similar.

4 Discussions

This study represents the first investigation in brain disease classification that focuses explicitly on task-specific FC patterns. Task-based fMRI offers distinct advantages in exploring and understanding the mechanisms and brain-behavior relationships specific to cognitive impairments, which may not be evident in resting-state fMRI. Task paradigms provide structured cognitive engagement (Jiang et al., 2020; Yoo et al., 2022), allowing for a better examination of individual differences in critical neural circuits (Greene et al., 2018). Given the advantages of task fMRI, we employed the Elastic-Net regression model to explore task-specific FC patterns decoded relying on brain-behavior relationships. Additionally, we used resting-state fMRI to decode general intelligence as a baseline for comparison with task-specific FC (Dubois et al., 2018; Thiele et al., 2022; Anderson and Barbey, 2023). The decoding results for different tasks exhibited high heterogeneity, highlighting the brain regions and connectivity patterns that are more representative of the current task, in contrast to traditional supervised models based on task labels alone (Zhang et al., 2022).

In decoding brain-behavior relationships, selecting predictors and outcomes for the predictive model is a topic worthy of exploration. While predictions about various behaviors can be made based on resting-state data, our research prioritizes focus on the relationship between task-state fMRI and corresponding cognitive performance under task scenarios. The predictive modeling based on task-state fMRI is inspired by the potential of task-state FC to enhance cognitive outcome prediction (Jiang et al., 2020). Additionally, it fully explores the multiple task states within the HCP dataset. We consider utilizing the Acc/RT ratio as a behavioral index for tasks with accuracy and response speed metrics in predicting behavioral performance. Literature also conceptualizes the trade-off between speed and accuracy as ‘throughput’ (Thorne, 2006; Heitz, 2014). It reflects the accuracy of the response and its rapidity, thereby providing a composite measure of cognitive processing efficiency.

Our research corroborates the efficacy of integrating task-specific connectome priors into classification models for diagnosing a spectrum of psychiatric disorders across various datasets. Specifically, enhanced classification performance is observed in differentiating diseases when utilizing FC patterns associated with specific cognitive domains (Chauvin et al., 2021). Network patterns related to working memory tasks contribute significantly to both ADHD and ASD datasets. The previous study also reveals that impairments in working memory are prevalent across psychiatric conditions (Wang X. L. et al., 2021), and memory assessments are crucial for predicting and mitigating high-risk disorders (Seabury and Cannon, 2020). In classifying ADHD, leading tasks also encompass motor task, social cognition, and relational processing. Previous studies have demonstrated that severe declines in social cognition and motor speed (Haining et al., 2020) correlate with a high risk of clinical psychiatric conditions. ADHD is also associated with abnormalities in the large-scale cognitive control network that impact social attention (Fateh et al., 2022), with adolescents among the patient population exhibiting impairments in social cognition and communication abilities (Chen and Chen, 2020). Children with ADHD have a deficit in relational reasoning (Brunamonti et al., 2017), a skill subtending the acquisition of many cognitive abilities and social rules. In the classification of the ABIDE dataset, leading tasks also encompass emotion, gambling, and social cognition. Facial emotion recognition disorder is typical of people with autism. Facial emotion recognition disorder is a classic symptom of autism (Yeung, 2022). Cognitive inflexibility in people with autism appears characterized by the unwillingness to switch toward processing socio-emotional information (Latinus et al., 2019). Individuals with ASD frequently report difficulty making flexible decisions across various contexts to resolve social or moral conflicts(Tei et al., 2022). Concurrently, studies based on gambling paradigms also suggest they tend to exhibit a more cautious decision-making style (Hosozawa et al., 2021).

Integrating multi-task FC and graph theory has further enhanced classification accuracy, achieving optimal performance using four task combinations. However, the addition of features from more tasks did not continue to improve classification results, presenting an intriguing avenue for investigation. In constructing brain FC-based diagnostic models, selecting features is more critical than quantity (Du et al., 2018; Chen et al., 2020). An increased number of features may offer a richer representation of task-specific FC information, but it can also lead to the “curse of dimensionality”—a phenomenon where the introduction of noise, overfitting, and the increased difficulty of identifying meaningful patterns in high-dimensional spaces may decrease classification performance (Wee et al., 2014; Barbieri et al., 2022). Our research also validates that opting for a more suitable selection of features, rather than simply increasing their number, is the superior strategy.

The TSP-GNN system achieves a balanced trade-off between model interpretability and classification performance. In contrast to previous studies that incorporated whole-brain connectome features, our model utilizes a task-specific FC pattern, which enhances the interpretability of features by linking them to specific cognitive activities. Furthermore, the classification stage of the TSP-GNN framework employs a population graph model, simplifying the modeling of brain areas as nodes and improving classification performance. Regarding classification performance, our TSP-GNN outperforms various classical machine learning and deep network models, underscoring the superiority of our task-prioritized population graph model in detecting brain diseases. Although our classification accuracy may differ from recent studies (Chen et al., 2021; Pan J. et al., 2022), this may be due to trade-offs and parameter adjustments made during model construction. Our framework prioritizes the interpretation of cognitive processes and their extended values related to underlying disease, and task-specific prior information from brain areas can be easily transferred to other studies of cognitive brain disease and disorders. In summary, we consider the decoding model in our TSP-GNN framework as a pre-task, effectively reducing feature dimensionality and elucidating the role of task-specific prior information in the classification model for brain disease diagnosis. The model effectively bridges the gap between cognitive behavior decoding and brain illness research, offering valuable insights and serving as a reference for task-related investigations in brain diseases.

Several considerations need to be addressed in our research. Firstly, it should be acknowledged that the ADHD and ABIDE illness cohorts in our study were not comprehensive and may not represent all available data sources. The inherent imbalance resulting from variations in data collection parameters and equipment across different locations is a significant challenge in our investigation. Constrained by the differing intended uses of data acquisition between HCP and ND, a strict age match between groups was not feasible, thus warranting further investigation into the exclusion of age-related differences in brain network impacts (Zhang et al., 2023). Secondly, the task-specific FC derived from the regression process has enhanced the efficacy of disease diagnosis and is considered, to some extent, correlative rather than causally direct. Employing causal correlation-based FC (Sanchez-Romero et al., 2023) and evidence of neural modulation (Zhou et al., 2020) based on brain networks holds promise for overcoming this limitation. Lastly, our current classification results can be further enhanced by refining the incorporation of prior information and optimizing future models to approach state-of-the-art performance. Continual efforts to improve the quality of prior knowledge and refine model development are necessary to ensure our approach remains at the forefront of research in this field.

Future research aims to develop deep learning models integrating cognitive performance and task state labels for brain decoding. Recognizing the intricate relationship between brain decoding and classification, despite their distinct objectives, we intend to explore the application of zero-shot learning and advanced transfer learning models that can achieve mutual benefits for both brain function decoding and disease classification tasks (Zhang P. et al., 2021). An exciting prospect is the collection of psychiatric disorder data using appropriate task paradigms in clinical settings (Birba et al., 2022). By incorporating task performance in actual clinical circumstances, we can investigate and evaluate the underlying causes of illnesses, expand our prior knowledge about task-based brain activity, and further optimize our models accordingly. Our future endeavors aim to bridge the gap between brain decoding and disease classification by developing advanced deep-learning models informed by clinical data and task performance. This approach has the potential to significantly contribute to the field by providing valuable insights into the underlying mechanisms of brain disorders and facilitating more accurate diagnoses.

5 Conclusion

The present study introduces a novel TSP-GNN framework to improve brain disease classification. By leveraging functional connection-based cognitive performance prediction, this study decodes task-specific FC patterns and transfers them as prior knowledge for diagnosing ND. As far as we know, this study represents the first attempt to transfer task-specific connectivity patterns as a priori knowledge in brain disease research. Our results demonstrate that integrating task-specific priors leads to improved classification accuracy compared to traditional methods. The finding highlights the informativeness of task-specific connection patterns. Besides, the optimal task combinations for each kind of ND offer valuable insights into the underlying mechanisms of that brain disease. By incorporating task-specific connectivity patterns, our framework enhances the understanding and prediction of brain diseases, opening up new avenues for future investigations in this domain.

Data availability statement

The original contributions presented in the study are included in the article/Supplementary material, further inquiries can be directed to the corresponding authors.

Author contributions

JL: Conceptualization, Investigation, Methodology, Software, Writing – original draft, Writing – review & editing. L-ZY: Conceptualization, Methodology, Supervision, Writing – review & editing. HL: Conceptualization, Supervision, Writing – review & editing.

Funding

The author(s) declare financial support was received for the research, authorship, and/or publication of this article. We thank all participants for their time and effort. Preliminary analyzes were sponsored by the Natural Science Fund of China (82371931), the Natural Science Fund of Anhui Province (2008085MC69), the Natural Science Fund of Hefei City (2021033), HFIPS Director’s Fund (YZJJ202207-TS), the General scientific research project of Anhui Provincial Health Commission (AHWJ2021b150), Collaborative Innovation Program of Hefei Science Center, CAS (2021HSC-CIP013), Anhui Province Key Laboratory of Medical Physics and Technology (LMPT201904).

Conflict of interest

The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.

Publisher’s note

All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.

Supplementary material

The Supplementary material for this article can be found online at: https://www.frontiersin.org/articles/10.3389/fnins.2023.1288882/full#supplementary-material

Footnotes

References

Abraham, A., Milham, M. P., Di Martino, A., Craddock, R. C., Samaras, D., Thirion, B., et al. (2017). Deriving reproducible biomarkers from multi-site resting-state data: an autism-based example. Neuroimage 147, 736–745. doi: 10.1016/j.neuroimage.2016.10.045

PubMed Abstract | CrossRef Full Text | Google Scholar

Anderson, E. D., and Barbey, A. K. (2023). Investigating cognitive neuroscience theories of human intelligence: a connectome-based predictive modeling approach. Hum. Brain Mapp. 44, 1647–1665. doi: 10.1002/hbm.26164

PubMed Abstract | CrossRef Full Text | Google Scholar

Barbieri, M., Lee, P. K., Brizi, L., Giampieri, E., Solera, F., Castellani, G., et al. (2022). Circumventing the curse of dimensionality in magnetic resonance fingerprinting through a deep learning approach. NMR Biomed. 35:e4670. doi: 10.1002/nbm.4670

PubMed Abstract | CrossRef Full Text | Google Scholar

Bellec, P., Chu, C., Chouinard-Decorte, F., Benhajali, Y., Margulies, D. S., and Craddock, R. C. (2017). The neuro bureau ADHD-200 preprocessed repository. NeuroImage 144, 275–286. doi: 10.1016/j.neuroimage.2016.06.034

PubMed Abstract | CrossRef Full Text | Google Scholar

Birba, A., Fittipaldi, S., Cediel Escobar, J. C., Gonzalez Campo, C., Legaz, A., Galiani, A., et al. (2022). Multimodal neurocognitive markers of naturalistic discourse typify diverse neurodegenerative diseases. Cereb. Cortex 32, 3377–3391. doi: 10.1093/cercor/bhab421

PubMed Abstract | CrossRef Full Text | Google Scholar

Briend, F., Marzloff, V., Brazo, P., Lecardeur, L., Leroux, E., Razafimandimby, A., et al. (2019). Social cognition in schizophrenia: validation of an ecological fMRI task. Psychiatry Res. Neuroimag. 286, 60–68. doi: 10.1016/j.pscychresns.2019.03.004

PubMed Abstract | CrossRef Full Text | Google Scholar

Brunamonti, E., Costanzo, F., Mammi, A., Rufini, C., Veneziani, D., Pani, P., et al. (2017). Evaluation of relational reasoning by a transitive inference task in attention-deficit/hyperactivity disorder. Neuropsychology 31, 200–208. doi: 10.1037/neu0000332

PubMed Abstract | CrossRef Full Text | Google Scholar

Cai, H., Chen, J., Liu, S., Zhu, J., and Yu, Y. (2020). Brain functional connectome-based prediction of individual decision impulsivity. Cortex 125, 288–298. doi: 10.1016/j.cortex.2020.01.022

PubMed Abstract | CrossRef Full Text | Google Scholar

Cameron, C., Yassine, B., Carlton, C., Francois, C., Alan, E., András, J., et al. (2013). The neuro bureau preprocessing initiative: open sharing of preprocessed neuroimaging data and derivatives. Front. Neuroinform. 7:41. doi: 10.3389/conf.fninf.2013.09.00041

CrossRef Full Text | Google Scholar

Canario, E., Chen, D., and Biswal, B. (2021). A review of resting-state fMRI and its use to examine psychiatric disorders. Psychoradiology 1, 42–53. doi: 10.1093/psyrad/kkab003

CrossRef Full Text | Google Scholar

Caunca, M. R., Wang, L., Cheung, Y. K., Alperin, N., Lee, S. H., Elkind, M. S. V., et al. (2021). Machine learning-based estimation of cognitive performance using regional brain MRI markers: the northern Manhattan study. Brain Imaging Behav. 15, 1270–1278. doi: 10.1007/s11682-020-00325-3

PubMed Abstract | CrossRef Full Text | Google Scholar

Chan, N. K., Kim, J., Shah, P., Brown, E. E., Plitman, E., Carravaggio, F., et al. (2019). Resting-state functional connectivity in treatment response and resistance in schizophrenia: a systematic review. Schizophr. Res. 211, 10–20. doi: 10.1016/j.schres.2019.07.020

PubMed Abstract | CrossRef Full Text | Google Scholar

Chauvin, R. J., Buitelaar, J. K., Sprooten, E., Oldehinkel, M., Franke, B., Hartman, C., et al. (2021). Task-generic and task-specific connectivity modulations in the ADHD brain: an integrated analysis across multiple tasks. Transl. Psychiatry 11:159. doi: 10.1038/s41398-021-01284-z

PubMed Abstract | CrossRef Full Text | Google Scholar

Chen, M. H., and Chen, Y. L. (2020). Functional connectivity of specific brain networks related to social and communication dysfunction in adolescents with attention-deficit hyperactivity disorder. Psychiatry Res. 284:112785. doi: 10.1016/j.psychres.2020.112785

PubMed Abstract | CrossRef Full Text | Google Scholar

Chen, Y. L., Tu, P. C., Huang, T. H., Bai, Y. M., Su, T. P., Chen, M. H., et al. (2020). Using minimal-redundant and maximal-relevant whole-brain functional connectivity to classify bipolar disorder. Front. Neurosci. 14:563368. doi: 10.3389/fnins.2020.563368

PubMed Abstract | CrossRef Full Text | Google Scholar

Chen, H., Zhuang, F., Xiao, L., Ma, L., Liu, H., Zhang, R., et al. (2021). AMA-GCN: Adaptive multi-layer aggregation graph convolutional network for disease prediction. Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence (IJCAI-21).

Google Scholar

Cole, M. W., Ito, T., Cocuzza, C., and Sanchez-Romero, R. (2021). The functional relevance of task-state functional connectivity. J. Neurosci. 41, 2684–2702. doi: 10.1523/jneurosci.1713-20.2021

PubMed Abstract | CrossRef Full Text | Google Scholar

Du, Y., Fu, Z., and Calhoun, V. D. (2018). Classification and prediction of brain disorders using functional connectivity: promising but challenging. Front. Neurosci. 12:525. doi: 10.3389/fnins.2018.00525

PubMed Abstract | CrossRef Full Text | Google Scholar

Dubois, J., Galdi, P., Paul, L. K., and Adolphs, R. (2018). A distributed brain network predicts general intelligence from resting-state human neuroimaging data. Philos. Trans. R. Soc. Lond. Ser. B Biol. Sci. 373:20170284. doi: 10.1098/rstb.2017.0284

PubMed Abstract | CrossRef Full Text | Google Scholar

Dvornek, N. C., Ventola, P., Pelphrey, K. A., and Duncan, J. S. (2017). Identifying autism from resting-state fMRI using long short-term memory networks. Berlin: Springer International Publishing, pp. 362–370.

Google Scholar

Eddy, C. M. (2019). What do You have in mind? Measures to assess mental state reasoning in neuropsychiatric populations. Front. Psych. 10:425. doi: 10.3389/fpsyt.2019.00425

PubMed Abstract | CrossRef Full Text | Google Scholar

Esteban, O., Markiewicz, C. J., Blair, R. W., Moodie, C. A., Isik, A. I., Erramuzpe, A., et al. (2019). fMRIPrep: a robust preprocessing pipeline for functional MRI. Nat. Methods 16, 111–116. doi: 10.1038/s41592-018-0235-4

PubMed Abstract | CrossRef Full Text | Google Scholar

Fallahi, A., Pooyan, M., Lotfi, N., Baniasad, F., Tapak, L., Mohammadi-Mobarakeh, N., et al. (2021). Dynamic functional connectivity in temporal lobe epilepsy: a graph theoretical and machine learning approach. Neurol. Sci. 42, 2379–2390. doi: 10.1007/s10072-020-04759-x

CrossRef Full Text | Google Scholar

Fan, L., Li, H., Zhuo, J., Zhang, Y., Wang, J., Chen, L., et al. (2016). The human Brainnetome atlas: a new brain atlas based on connectional architecture. Cereb Cortex 26, 3508–3526. doi: 10.1093/cercor/bhw157

CrossRef Full Text | Google Scholar

Farahani, F. V., Karwowski, W., and Lighthall, N. R. (2019). Application of graph theory for identifying connectivity patterns in human brain networks: a systematic review. Front. Neurosci. 13:585. doi: 10.3389/fnins.2019.00585

CrossRef Full Text | Google Scholar

Fateh, A. A., Huang, W. X., Mo, T., Wang, X. Y., Luo, Y., Yang, B. R., et al. (2022). Abnormal insular dynamic functional connectivity and its relation to social Dysfunctioning in children with attention deficit/hyperactivity disorder. Front. Neurosci. 16:596. doi: 10.3389/fnins.2022.890596

PubMed Abstract | CrossRef Full Text | Google Scholar

Felouat, H., and Oukid, S. (2020). Graph convolutional networks and functional connectivity for identification of autism Spectrum disorder. 2020 Second International Conference on Embedded & Distributed Systems (EDiS).

Google Scholar

Finn, E. S. (2021). Is it time to put rest to rest? Trends Cogn. Sci. 25, 1021–1032. doi: 10.1016/j.tics.2021.09.005

PubMed Abstract | CrossRef Full Text | Google Scholar

Greene, A. S., Gao, S. Y., Scheinost, D., and Constable, R. T. (2018). Task-induced brain state manipulation improves prediction of individual traits. Nat. Commun. 9:920. doi: 10.1038/s41467-018-04920-3

CrossRef Full Text | Google Scholar

Guo, X., Dominick, K. C., Minai, A. A., Li, H., Erickson, C. A., and Lu, L. J. (2017). Diagnosing autism Spectrum disorder from brain resting-state functional connectivity patterns using a deep neural network with a novel feature selection method. Front. Neurosci. 11:460. doi: 10.3389/fnins.2017.00460

PubMed Abstract | CrossRef Full Text | Google Scholar

Gupta, S., Lim, M., and Rajapakse, J. C. (2022). Decoding task specific and task general functional architectures of the brain. Hum. Brain Mapp. 43, 2801–2816. doi: 10.1002/hbm.25817

PubMed Abstract | CrossRef Full Text | Google Scholar

Haining, K., Matrunola, C., Mitchell, L., Gajwani, R., Gross, J., Gumley, A. I., et al. (2020). Neuropsychological deficits in participants at clinical high risk for psychosis recruited from the community: relationships to functioning and clinical symptoms. Psychol. Med. 50, 77–85. doi: 10.1017/s0033291718003975

PubMed Abstract | CrossRef Full Text | Google Scholar

Hearne, L., Mill, R., Keane, B., Repovs, G., Anticevic, A., and Cole, M. (2021). Activity flow underlying abnormalities in brain activations and cognition in schizophrenia. Sci. Adv. 7:eabf2513. doi: 10.1126/sciadv.abf2513

PubMed Abstract | CrossRef Full Text | Google Scholar

Heitz, R. P. (2014). The speed-accuracy tradeoff: history, physiology, methodology, and behavior. Front. Neurosci. 8:150. doi: 10.3389/fnins.2014.00150

PubMed Abstract | CrossRef Full Text | Google Scholar

Hosozawa, M., Mandy, W., Cable, N., and Flouri, E. (2021). The role of decision-making in psychological wellbeing and risky Behaviours in autistic adolescents without ADHD: longitudinal evidence from the UK millennium cohort study. J. Autism Dev. Disord. 51, 3212–3223. doi: 10.1007/s10803-020-04783-y

PubMed Abstract | CrossRef Full Text | Google Scholar

Huang, Y., and Chung, A. C. S. (2020). “Edge-Variational graph convolutional networks for uncertainty-aware disease prediction” in Medical image computing and computer assisted intervention – MICCAI 2020. eds. A. L. Martel, P. Abolmaesumi, D. Stoyanov, D. Mateus, M. A. Zuluaga, and S. K. Zhou (Berlin: Springer International Publishing), 562–572.

Google Scholar

Ioakeimidis, V., Haenschel, C., Fett, A.-K., Kyriakopoulos, M., and Dima, D. (2022). Functional neurodevelopment of working memory in early-onset schizophrenia: a longitudinal FMRI study. Schizophr. Res. Cogn. 30:100268. doi: 10.1016/j.scog.2022.100268

PubMed Abstract | CrossRef Full Text | Google Scholar

Jahn, F. S., Skovbye, M., Obenhausen, K., Jespersen, A. E., and Miskowiak, K. W. (2021). Cognitive training with fully immersive virtual reality in patients with neurological and psychiatric disorders: a systematic review of randomized controlled trials. Psychiatry Res. 300:113928. doi: 10.1016/j.psychres.2021.113928

PubMed Abstract | CrossRef Full Text | Google Scholar

Jiang, Z., Wang, Y., Shi, C., Wu, Y., Hu, R., Chen, S., et al. (2022). Attention module improves both performance and interpretability of four-dimensional functional magnetic resonance imaging decoding neural network. Hum. Brain Mapp. 43, 2683–2692. doi: 10.1002/hbm.25813

PubMed Abstract | CrossRef Full Text | Google Scholar

Jiang, R., Zuo, N., Ford, J. M., Qi, S., Zhi, D., Zhuo, C., et al. (2020). Task-induced brain connectivity promotes the detection of individual differences in brain-behavior relationships. NeuroImage 207:116370. doi: 10.1016/j.neuroimage.2019.116370

CrossRef Full Text | Google Scholar

Khosla, M., Jamison, K., Kuceyeski, A., and Sabuncu, M. R. (2018). “3D convolutional neural networks for classification of functional connectomes” in Deep learning in medical image analysis and multimodal learning for clinical decision support. eds. D. Stoyanov, Z. Taylor, G. Carneiro, T. Syeda-Mahmood, A. Martel, and L. Maier-Hein (Berlin: Springer International Publishing), 137–145.

Google Scholar

Kim, M., Bao, J., Liu, K., Park, B.-Y., Park, H., Baik, J. Y., et al. (2021). A structural enriched functional network: an application to predict brain cognitive performance. Med. Image Anal. 71:102026. doi: 10.1016/j.media.2021.102026

PubMed Abstract | CrossRef Full Text | Google Scholar

Kim, J., Calhoun, V. D., Shim, E., and Lee, J. H. (2016). Deep neural network with weight sparsity control and pre-training extracts hierarchical features and enhances classification performance: evidence from whole-brain resting-state functional connectivity patterns of schizophrenia. NeuroImage 124, 127–146. doi: 10.1016/j.neuroimage.2015.05.018

PubMed Abstract | CrossRef Full Text | Google Scholar

Kofler, M. J., Soto, E. F., Fosco, W. D., Irwin, L. N., Wells, E. L., and Sarver, D. E. (2020). Working memory and information processing in ADHD: evidence for directionality of effects. Neuropsychology 34, 127–143. doi: 10.1037/neu0000598

PubMed Abstract | CrossRef Full Text | Google Scholar

Lanillos, P., Oliva, D., Philippsen, A., Yamashita, Y., Nagai, Y., and Cheng, G. (2020). A review on neural network models of schizophrenia and autism spectrum disorder. Neural Netw. 122, 338–363. doi: 10.1016/j.neunet.2019.10.014

PubMed Abstract | CrossRef Full Text | Google Scholar

Latinus, M., Cléry, H., Andersson, F., Bonnet-Brilhault, F., Fonlupt, P., and Gomot, M. (2019). Inflexibility in autism Spectrum disorder: need for certainty and atypical emotion processing share the blame. Brain Cogn. 136:103599. doi: 10.1016/j.bandc.2019.103599

PubMed Abstract | CrossRef Full Text | Google Scholar

Lee, J. J., Kim, H. J., Ceko, M., Park, B. Y., Lee, S. A., Park, H., et al. (2021). A neuroimaging biomarker for sustained experimental and clinical pain. Nat. Med. 27:174-+. doi: 10.1038/s41591-020-1142-7

PubMed Abstract | CrossRef Full Text | Google Scholar

Li, L., Jiang, H., Wen, G., Cao, P., Xu, M., Liu, X., et al. (2021). TE-HI-GCN: an Ensemble of Transfer Hierarchical Graph Convolutional Networks for disorder diagnosis. Neuroinformatics 20, 353–375. doi: 10.1007/s12021-021-09548-1

CrossRef Full Text | Google Scholar

Li, A., Zalesky, A., Yue, W. H., Howes, O., Yan, H., Liu, Y., et al. (2020). A neuroimaging biomarker for striatal dysfunction in schizophrenia. Nat. Med. 26:558. doi: 10.1038/s41591-020-0793-8

PubMed Abstract | CrossRef Full Text | Google Scholar

Li, X., Zhou, Y., Dvornek, N., Zhang, M., Gao, S., Zhuang, J., et al. (2021). BrainGNN: interpretable brain graph neural network for fMRI analysis. Med. Image Anal. 74:102233. doi: 10.1016/j.media.2021.102233

CrossRef Full Text | Google Scholar

Liu, M., Li, B., and Hu, D. (2021). Autism Spectrum disorder studies using fMRI data and machine learning: a review. Front. Neurosci. 15:7870. doi: 10.3389/fnins.2021.697870

CrossRef Full Text | Google Scholar

Liu, Y., Xu, L., Li, J., Yu, J., and Yu, X. (2020). Attentional connectivity-based prediction of autism using heterogeneous rs-fMRI data from CC200 atlas. Exp. Neurobiol. 29, 27–37. doi: 10.5607/en.2020.29.1.27

PubMed Abstract | CrossRef Full Text | Google Scholar

Pan, J., Dong, Y., and Chen, H. (2022). Review of research on auxiliary diagnosis of autism based on graph neural networks. Comput. Eng. 48, 1–11. doi: 10.19678/j.issn.1000-3428.0064352

CrossRef Full Text | Google Scholar

Pan, J., Lin, H., Dong, Y., Wang, Y., and Ji, Y. (2022). MAMF-GCN: multi-scale adaptive multi-channel fusion deep graph convolutional network for predicting mental disorder. Comput. Biol. Med. 148:105823. doi: 10.1016/j.compbiomed.2022.105823

PubMed Abstract | CrossRef Full Text | Google Scholar

Parisot, S., Ktena, S. I., Ferrante, E., Lee, M., Guerrero, R., Glocker, B., et al. (2018). Disease prediction using graph convolutional networks: application to autism Spectrum disorder and Alzheimer's disease. Med. Image Anal. 48, 117–130. doi: 10.1016/j.media.2018.06.001

PubMed Abstract | CrossRef Full Text | Google Scholar

Parisot, S., Ktena, S. I., Ferrante, E., Lee, M., Moreno, R., Glocker, B., et al. (2017). Spectral graph convolutions for population-based disease prediction. In: International conference on medical image computing and computer-assisted intervention: Springer International Publishing, pp. 177–185.

Google Scholar

Perez, D. L., Nicholson, T. R., Asadi-Pooya, A. A., Begue, I., Butler, M., Carson, A. J., et al. (2021). Neuroimaging in functional neurological disorder: state of the field and research agenda. Neuroimage Clin. 30:102623. doi: 10.1016/j.nicl.2021.102623

PubMed Abstract | CrossRef Full Text | Google Scholar

Porcelli, S., Van der Wee, N., van der Werff, S., Aghajani, M., Glennon, J. C., van Heukelum, S., et al. (2019). Social brain, social dysfunction and social withdrawal. Neurosci. Biobehav. Rev. 97, 10–33. doi: 10.1016/j.neubiorev.2018.09.012

CrossRef Full Text | Google Scholar

Ravishankar, H., Madhavan, R., Mullick, R., Shetty, T., Marinelli, L., and Joel, S. E. (2016). Recursive feature elimination for biomarker discovery in resting-state functional connectivity. In: 38th annual international conference of the IEEE-Engineering-in-Medicine-and-Biology-Society (EMBC), pp. 4071–4074.

Google Scholar

Riedel, P., Lee, J. H., Watson, C. G., Jimenez, A. M., Reavis, E. A., and Green, M. F. (2022). Reorganization of the functional connectome from rest to a visual perception task in schizophrenia and bipolar disorder. Psychiatr. Res. Neuroimag. 327:111556. doi: 10.1016/j.pscychresns.2022.111556

PubMed Abstract | CrossRef Full Text | Google Scholar

Sanchez-Romero, R., and Cole, M. W. (2021). Combining multiple functional connectivity methods to improve causal inferences. J. Cogn. Neurosci. 33, 180–194. doi: 10.1162/jocn_a_01580

CrossRef Full Text | Google Scholar

Sanchez-Romero, R., Ito, T., Mill, R. D., Hanson, S. J., and Cole, M. W. (2023). Causally informed activity flow models provide mechanistic insight into network-generated cognitive activations. NeuroImage 278:120300. doi: 10.1016/j.neuroimage.2023.120300

PubMed Abstract | CrossRef Full Text | Google Scholar

Savanth, A. S., Vijaya, P. A., Nair, A. K., and Kutty, B. M. (2022). Classification of Rajayoga meditators based on the duration of practice using graph theoretical measures of functional connectivity from task-based functional magnetic resonance imaging. Int. J. Yoga 15, 96–105. doi: 10.4103/ijoy.ijoy_17_22

PubMed Abstract | CrossRef Full Text | Google Scholar

Seabury, R. D., and Cannon, T. D. (2020). Memory impairments and psychosis prediction: a scoping review and theoretical overview. Neuropsychol. Rev. 30, 521–545. doi: 10.1007/s11065-020-09464-2

PubMed Abstract | CrossRef Full Text | Google Scholar

Spencer, D., Yue, Y. R., Bolin, D., Ryan, S., and Mejia, A. F. (2022). Spatial Bayesian GLM on the cortical surface produces reliable task activations in individuals and groups. NeuroImage 249:118908. doi: 10.1016/j.neuroimage.2022.118908

CrossRef Full Text | Google Scholar

Sui, J., Jiang, R., Bustillo, J., and Calhoun, V. (2020). Neuroimaging-based individualized prediction of cognition and behavior for mental disorders and health: methods and promises. Biol. Psychiatry 88, 818–828. doi: 10.1016/j.biopsych.2020.02.016

PubMed Abstract | CrossRef Full Text | Google Scholar

Tei, S. S., Tanicha, M., Itahashi, T., Aoki, Y. Y., Ohta, H., Qian, C. Y., et al. (2022). Decision flexibilities in autism spectrum disorder: an fMRI study of moral dilemmas. Soc. Cogn. Affect. Neurosci. 17, 904–911. doi: 10.1093/scan/nsac023

PubMed Abstract | CrossRef Full Text | Google Scholar

Thiele, J. A., Faskowitz, J., Sporns, O., and Hilger, K. (2022). Multitask brain network reconfiguration is inversely associated with human intelligence. Cereb. Cortex 32, 4172–4182. doi: 10.1093/cercor/bhab473

PubMed Abstract | CrossRef Full Text | Google Scholar

Thorne, D. R. (2006). Throughput: a simple performance index with desirable characteristics. Behav. Res. Methods 38, 569–573. doi: 10.3758/bf03193886

PubMed Abstract | CrossRef Full Text | Google Scholar

Vaidya, C. J., You, X., Mostofsky, S., Pereira, F., Berl, M. M., and Kenworthy, L. (2020). Data-driven identification of subtypes of executive function across typical development, attention deficit hyperactivity disorder, and autism spectrum disorders. J. Child Psychol. Psychiatry 61, 51–61. doi: 10.1111/jcpp.13114

PubMed Abstract | CrossRef Full Text | Google Scholar

Van Essen, D. C., Smith, S. M., Barch, D. M., Behrens, T. E. J., Yacoub, E., Ugurbil, K., et al. (2013). The WU-Minn human connectome project: an overview. NeuroImage 80, 62–79. doi: 10.1016/j.neuroimage.2013.05.041

PubMed Abstract | CrossRef Full Text | Google Scholar

Wadhera, T., and Kakkar, D. (2020). Multiplex temporal measures reflecting neural underpinnings of brain functional connectivity under cognitive load in autism Spectrum disorder. Neurol. Res. 42, 327–337. doi: 10.1080/01616412.2020.1726586

PubMed Abstract | CrossRef Full Text | Google Scholar

Wang, X. L., Cheng, B. C., Roberts, N., Wang, S., Luo, Y., Tian, F. F., et al. (2021). Shared and distinct brain fMRI response during performance of working memory tasks in adult patients with schizophrenia and major depressive disorder. Hum. Brain Mapp. 42, 5458–5476. doi: 10.1002/hbm.25618

CrossRef Full Text | Google Scholar

Wang, L., Li, K., and Hu, X. P. (2021). Graph convolutional network for fMRI analysis based on connectivity neighborhood. Netw. Neurosci. 5, 83–95. doi: 10.1162/netn_a_00171

PubMed Abstract | CrossRef Full Text | Google Scholar

Wang, B., Li, L., Peng, L., Jiang, Z., Dai, K., Xie, Q., et al. (2022). Multigroup recognition of dementia patients with dynamic brain connectivity under multimodal cortex parcellation. Biomed. Sig. Proc. Control 76:103725. doi: 10.1016/j.bspc.2022.103725

CrossRef Full Text | Google Scholar

Wang, Y., Liu, J., Xiang, Y., Wang, J., Chen, Q., and Chong, J. (2022). MAGE: automatic diagnosis of autism spectrum disorders using multi-atlas graph convolutional networks and ensemble learning. Neurocomputing 469, 346–353. doi: 10.1016/j.neucom.2020.06.152

CrossRef Full Text | Google Scholar

Wee, C. Y., Yap, P. T., Zhang, D., Wang, L., and Shen, D. (2014). Group-constrained sparse fMRI connectivity modeling for mild cognitive impairment identification. Brain Struct. Funct. 219, 641–656. doi: 10.1007/s00429-013-0524-8

PubMed Abstract | CrossRef Full Text | Google Scholar

Xia, M., Womer, F. Y., Chang, M., Zhu, Y., Zhou, Q., Edmiston, E. K., et al. (2019). Shared and distinct functional architectures of brain networks across psychiatric disorders. Schizophr. Bull. 45, 450–463. doi: 10.1093/schbul/sby046

PubMed Abstract | CrossRef Full Text | Google Scholar

Yeung, M. K. (2022). A systematic review and meta-analysis of facial emotion recognition in autism spectrum disorder: the specificity of deficits and the role of task characteristics. Neurosci. Biobehav. Rev. 133:104518. doi: 10.1016/j.neubiorev.2021.104518

PubMed Abstract | CrossRef Full Text | Google Scholar

Ying, R., Bourgeois, D., You, J., Zitnik, M., and Leskovec, J. J. A. (2019). GNNExplainer: generating explanations for graph neural networks. Available at: https://ui.adsabs.harvard.edu/abs/2019arXiv190303894Y (Accessed March 1, 2019).

Google Scholar

Yoo, K., Rosenberg, M. D., Kwon, Y. H., Scheinost, D., Constable, R. T., and Chun, M. M. (2022). A cognitive state transformation model for task-general and task-specific subsystems of the brain connectome. NeuroImage 257:119279. doi: 10.1016/j.neuroimage.2022.119279

PubMed Abstract | CrossRef Full Text | Google Scholar

Zamani, J., Sadr, A., and Javadi, A.-H. (2022). Classification of early-MCI patients from healthy controls using evolutionary optimization of graph measures of resting-state fMRI, for the Alzheimer’s disease neuroimaging initiative. PLoS One 17:608. doi: 10.1371/journal.pone.0267608

CrossRef Full Text | Google Scholar

Zepf, F. D., Bubenzer-Busch, S., Runions, K. C., Rao, P., Wong, J. W. Y., Mahfouda, S., et al. (2019). Functional connectivity of the vigilant-attention network in children and adolescents with attention-deficit/hyperactivity disorder. Brain Cogn. 131, 56–65. doi: 10.1016/j.bandc.2017.10.005

PubMed Abstract | CrossRef Full Text | Google Scholar

Zhang, Z., Cui, P., and Zhu, W. (2020). Deep learning on graphs: a survey. IEEE Trans. Knowl. Data Eng. 34, 249–270. doi: 10.1109/TKDE.2020.2981333

CrossRef Full Text | Google Scholar

Zhang, Y., Farrugia, N., and Bellec, P. (2022). Deep learning models of cognitive processes constrained by human brain connectomes. Med. Image Anal. 80:102507. doi: 10.1016/j.media.2022.102507

PubMed Abstract | CrossRef Full Text | Google Scholar

Zhang, P., Li, W., Ma, X., He, J., Huang, J., and Li, Q. (2021). Feature-selection-based transfer learning for Intracortical brain–machine Interface decoding. IEEE Trans. Neural Syst. Rehabil. Eng. 29, 60–73. doi: 10.1109/TNSRE.2020.3034234

CrossRef Full Text | Google Scholar

Zhang, T., Liao, Q., Zhang, D., Zhang, C., Yan, J., Ngetich, R., et al. (2021). Predicting MCI to AD conversation using integrated sMRI and rs-fMRI: machine learning and graph theory approach. Front. Aging Neurosci. 13:688926. doi: 10.3389/fnagi.2021.688926

PubMed Abstract | CrossRef Full Text | Google Scholar

Zhang, B., Zhang, S., Feng, J., and Zhang, S. (2023). Age-level bias correction in brain age prediction. Neuroimage Clin. 37:103319. doi: 10.1016/j.nicl.2023.103319

PubMed Abstract | CrossRef Full Text | Google Scholar

Zhao, K., Duka, B., Xie, H., Oathes, D. J., Calhoun, V., and Zhang, Y. (2022). A dynamic graph convolutional neural network framework reveals new insights into connectome dysfunctions in ADHD. NeuroImage 246:118774. doi: 10.1016/j.neuroimage.2021.118774

PubMed Abstract | CrossRef Full Text | Google Scholar

Zhou, T., Kang, J., Li, Z., Chen, H., and Li, X. (2020). Transcranial direct current stimulation modulates brain functional connectivity in autism. Neuroimage Clin. 28:102500. doi: 10.1016/j.nicl.2020.102500

PubMed Abstract | CrossRef Full Text | Google Scholar

Zhou, H., and Zhang, D. (2021). Graph-in-graph convolutional networks for brain disease diagnosis. In: 2021 IEEE International Conference on Image Processing (ICIP)), pp. 111–115.

Google Scholar

Keywords: neuropsychiatric disorders, task-specific prior knowledge, brain decoding, functional connectivity, graph neural network

Citation: Lang J, Yang L-Z and Li H (2023) TSP-GNN: a novel neuropsychiatric disorder classification framework based on task-specific prior knowledge and graph neural network. Front. Neurosci. 17:1288882. doi: 10.3389/fnins.2023.1288882

Received: 05 September 2023; Accepted: 01 December 2023;
Published: 21 December 2023.

Edited by:

Da Ma, Wake Forest University, United States

Reviewed by:

Xiaoxiao Wang, University of Science and Technology of China, China
Yuanqiang Zhu, Fourth Military Medical University, China

Copyright © 2023 Lang, Yang and Li. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

*Correspondence: Li-Zhuang Yang, lzyang@cmpt.ac.cn; Hai Li, hli@cmpt.ac.cn

Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.