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.

An Auto-Encoding Generative Adversarial Networks for Generating Brain Network.

A Unified Framework of Graph Structure Learning, Graph Generation and Classification for Brain Network Analysis

Collaborative Learning of Graph Generation, Clustering and Classification for Brain Networks Diagnosis.

Brain Graph Synthesis for Multi-Site Autism Spectrum Disorder Identification

Deep Neural Generative Adversarial Model based on VAE + GAN for Disorder Diagnosis

Functional brain network identification and fMRI augmentation using a VAE-GAN framework

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

Advancing ASD identification with neuroimaging: a novel GARL methodology integrating Deep Q-Learning and generative adversarial networks

BG-GAN: Generative AI Enables Representing Brain Structure-Function Connections for Alzheimer's Disease

BrainNetGAN: Data augmentation of brain connectivity using generative adversarial network for dementia classification

ASD-GANNet: A Generative Adversarial Network-Inspired Deep Learning Approach for the Classification of Autism Brain Disorder

A Graph Attention Neural Network for Diagnosing ASD with fMRI Data

Data augmentation using generative adversarial neural networks on brain structural connectivity in multiple sclerosis

AIDS Brain MRIs Synthesis Via Generative Adversarial Networks Based on Attention-Encoder

SA-GCN: Scale Adaptive Graph Convolutional Network for ASD Identification

Graph-Regularized Manifold-Aware Conditional Wasserstein GAN for Brain Functional Connectivity Generation

Adversarial Learning Based Node-Edge Graph Attention Networks for Autism Spectrum Disorder Identification

Generation of 3D Brain MRI Using Auto-Encoding Generative Adversarial Networks

BrainGNN: Interpretable Brain Graph Neural Network for fMRI Analysis

Brain functional activity-based classification of autism spectrum disorder using an attention-based graph neural network combined with gene expression

Brain Functional Network Generation Using Distribution-Regularized Adversarial Graph Autoencoder with Transformer for Dementia Diagnosis