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

Copyright © 2023 Elsevier Ltd. All rights reserved..

Recently, deep learning models have achieved superior performance for mapping functional brain networks from functional magnetic resonance imaging (fMRI) data compared with traditional methods. However, due to the lack of sufficient data and the high dimensionality of brain volume, deep learning models of fMRI tend to suffer from overfitting. In addition, existing methods rarely studied fMRI data augmentation and its application. To address these issues, we developed a VAE-GAN framework that combined a VAE (variational auto-encoder) with a GAN (generative adversarial net) for functional brain network identification and fMRI augmentation. As a generative model, the VAE-GAN models the distribution of fMRI so that it enables the extraction of more generalized features, and thus relieve the overfitting issue. The VAE-GAN is easier to train on fMRI than a standard GAN since it uses latent variables from VAE to generate fake data rather than relying on random noise that is used in a GAN, and it can generate higher quality of fake data than VAE since the discriminator can promote the training of the generator. In other words, the VAE-GAN inherits the advantages of VAE and GAN and avoids their limitations in modeling of fMRI data. Extensive experiments on task fMRI datasets from HCP have proved the effectiveness and superiority of the proposed VAE-GAN framework for identifying both temporal features and functional brain networks compared with existing models, and the quality of fake data is higher than those from VAE and GAN. The results on resting state fMRI of Attention Deficit Hyperactivity Disorder (ADHD)-200 dataset further demonstrated that the fake data generated by the VAE-GAN can help improve the performance of brain network modeling and ADHD classification.

Medienart:

E-Artikel

Erscheinungsjahr:

2023

Erschienen:

2023

Enthalten in:

Zur Gesamtaufnahme - volume:165

Enthalten in:

Computers in biology and medicine - 165(2023) vom: 30. Okt., Seite 107395

Sprache:

Englisch

Beteiligte Personen:

Qiang, Ning [VerfasserIn]
Gao, Jie [VerfasserIn]
Dong, Qinglin [VerfasserIn]
Yue, Huiji [VerfasserIn]
Liang, Hongtao [VerfasserIn]
Liu, Lili [VerfasserIn]
Yu, Jingjing [VerfasserIn]
Hu, Jing [VerfasserIn]
Zhang, Shu [VerfasserIn]
Ge, Bao [VerfasserIn]
Sun, Yifei [VerfasserIn]
Liu, Zhengliang [VerfasserIn]
Liu, Tianming [VerfasserIn]
Li, Jin [VerfasserIn]
Song, Hujie [VerfasserIn]
Zhao, Shijie [VerfasserIn]

Links:

Volltext

Themen:

Brain disorders
Data augmentation
FMRI
Functional brain network
Generative adversarial net
Journal Article
Research Support, Non-U.S. Gov't
Variational auto-encoder

Anmerkungen:

Date Completed 27.09.2023

Date Revised 27.09.2023

published: Print-Electronic

Citation Status MEDLINE

doi:

10.1016/j.compbiomed.2023.107395

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM36167256X