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.

GAN-ACNN: a design decision-making algorithm based on EEG signals from different brain regions

Neurocognition-inspired Design with Machine Learning

Space-CNN: a decision classification method based on EEG signals from different brain regions

Human-in-the-loop design with machine learning

EEG recognition model based on wavelet packet decomposition and ConvNet for art design decisions

Cross-Subject Decision Confidence Estimation from EEG Signals Using Spectral-Spatial-Temporal Adaptive GCN with Domain Adaptation.

A Visual Characteristic Land-Scape Design for EEG Signal Based on LSTM-GAN

EEGGAN-Net: enhancing EEG signal classification through data augmentation

Verifying Design through Generative Visualization of Neural Activities

EEG-Eye Movements Cross-Modal Decision Confidence Measurement with Generative Adversarial Networks

EEG Emotion Recognition Model Based on Attention and GAN

Electroencephalographic Signal Data Augmentation Based on Improved Generative Adversarial Network

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

Common Spatial Generative Adversarial Networks based EEG Data Augmentation for Cross-Subject Brain-Computer Interface

Predicting rock-paper-scissors choices based on single-trial EEG signals

Conditional generative adversarial network for EEG-based emotion fine-grained estimation and visualization.

NeuroGAN: image reconstruction from EEG signals via an attention-based GAN

EEG-Based Human Decision Confidence Measurement Using Graph Neural Networks

Discriminability of Single-Trial EEG During Decision-Making of Cooperation or Aggression: a Study Based on Machine Learning.

AssemblyNet: A Novel Deep Decision-Making Process for Whole Brain MRI Segmentation

Enhancing EEG-Based Decision-Making Performance Prediction by Maximizing Mutual Information Between Emotion and Decision-Relevant Features