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.

A Fine-Tuned EfficientNet B1 Based Deep Transfer Learning Framework for Multiple Types of Brain Disorder Classification

Multi-Class Brain Disease Classification Using Modified Pre-Trained Convolutional Neural Networks Model with Substantial Data Augmentation

Automated Brain Image Classification Based on VGG-16 and Transfer Learning

A Novel Transfer Learning Approach to Enhance Deep Neural Network Classification of Brain Functional Connectomes

A Unified Framework for Static and Dynamic Functional Connectivity Augmentation for Multi-Domain Brain Disorder Classification

Detection and classification of brain tumor using hybrid deep learning models

Brain Tumor Classification Using Deep Neural Network and Transfer Learning

An enhanced deep learning method for multi-class brain tumor classification using deep transfer learning

A comparative study of early stage Alzheimer's disease classification using various transfer learning CNN frameworks

Efficient framework for brain tumor detection using different deep learning techniques

BrainNet: a fusion assisted novel optimal framework of residual blocks and stacked autoencoders for multimodal brain tumor classification

Classification Framework for Medical Diagnosis of Brain Tumor with an Effective Hybrid Transfer Learning Model

Comparative Analysis of Different Deep Convolutional Neural Network Architectures for Classification of Brain Tumor on Magnetic Resonance Images

Improving brain tumor classification with combined convolutional neural networks and transfer learning

Detection and Classification of Brain tumors Using Deep Convolutional Neural Networks

Multi-Classification of Brain Tumor Images Using Transfer Learning Based Deep Neural Network

Improving Effectiveness of Different Deep Transfer Learning-Based Models for Detecting Brain Tumors From MR Images

A Novel Approach to Classify Brain Tumor with an Effective Transfer Learning based Deep Learning Model

A novel deep transfer learning framework integrating general and domain-specific features for EEG-based brain–computer interface

Classification of Brain Image Tumor using EfficientNet B1-B2 Deep Learning

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