Abstract:Functional connectivity (FC) obtained from resting-state functional magnetic resonance imaging has been integrated with machine learning algorithms to deliver consistent and reliable brain disease classification outcomes. However, in classical learning procedures, custom-built specialized feature selection techniques are typically used to filter out uninformative features from FC patterns to generalize efficiently on the datasets. The ability of convolutional neural networks (CNN) and other deep learning models to extract informative features from data with grid structure (such as images) has led to the surge in popularity of these techniques. However, the designs of many existing CNN models still fail to exploit the relationships between entities of graph-structure data (such as networks). Therefore, graph convolution network (GCN) has been suggested as a means for uncovering the intricate structure of brain network data, which has the potential to substantially improve classification accuracy. Furthermore, overfitting in classifiers can be largely attributed to the limited number of available training samples. Recently, the generative adversarial network (GAN) has been widely used in the medical field for its generative aspect that can generate synthesis images to cope with the problems of data scarcity and patient privacy. In our previous work, GCN and GAN have been designed to investigate FC patterns to perform diagnosis tasks, and their effectiveness has been tested on the ABIDE-I dataset. In this paper, the models will be further applied to FC data derived from more public datasets (ADHD, ABIDE-II, and ADNI) and our in-house dataset (PTSD) to justify their generalization on all types of data. The results of a number of experiments show the powerful characteristic of GAN to mimic FC data to achieve high performance in disease prediction. When employing GAN for data augmentation, the diagnostic accuracy across ADHD-200, ABIDE-II, and ADNI datasets surpasses that of other machine learning models, including results achieved with BrainNetCNN. Specifically, in ADHD, the accuracy increased from 67.74% to 73.96% with GAN, in ABIDE-II from 70.36% to 77.40%, and in ADNI, reaching 52.84% and 88.56% for multiclass and binary classification, respectively. GCN also obtains decent results, with the best accuracy in ADHD datasets at 71.38% for multinomial and 75% for binary classification, respectively, and the second-best accuracy in the ABIDE-II dataset (72.28% and 75.16%, respectively). Both GAN and GCN achieved the highest accuracy for the PTSD dataset, reaching 97.76%. However, there are still some limitations that can be improved. Both methods have many opportunities for the prediction and diagnosis of diseases.

Benchmarking Graph Neural Networks for FMRI analysis

BrainGNN: Interpretable Brain Graph Neural Network for fMRI Analysis

Discovering robust biomarkers of neurological disorders from functional MRI using graph neural networks: A Review

Graph Neural Networks in Network Neuroscience

NeuroGraph: Benchmarks for Graph Machine Learning in Brain Connectomics

Assisted diagnosis of neuropsychiatric disorders based on functional connectivity: A survey on application and performance evaluation of graph neural network

BrainGB: A Benchmark for Brain Network Analysis With Graph Neural Networks

Balanced Graph Structure Information for Brain Disease Detection

DepressionGraph: A Two-Channel Graph Neural Network for the Diagnosis of Major Depressive Disorders Using Rs-Fmri

FBNETGEN: Task-aware GNN-based fMRI Analysis via Functional Brain Network Generation

Deep reinforcement learning guided graph neural networks for brain network analysis

Joint Embedding of Structural and Functional Brain Networks with Graph Neural Networks for Mental Illness Diagnosis

A Deep Graph Neural Network Architecture for Modelling Spatio-temporal Dynamics in resting-state functional MRI Data

Learning Task-Aware Effective Brain Connectivity for fMRI Analysis with Graph Neural Networks

Interpretable Graph Neural Networks for Connectome-Based Brain Disorder Analysis

Fmri-Based Brain Disease Diagnosis: A Graph Network Approach

CC- GCN: A novel graph-based approach for identification and detection of autism spectrum disorder

The Use of Generative Adversarial Network and Graph Convolution Network for Neuroimaging-Based Diagnostic Classification

Modeling Brain Functional Networks Using Graph Neural Networks: A Review and Clinical Application

Graph Neural Networks for Brain Graph Learning: A Survey

Graph Autoencoders for Embedding Learning in Brain Networks and Major Depressive Disorder Identification