Abstract
Recently, functional brain networks have been employed for classifying neurological disorders, such as autism spectrum disorders (ASDs). Graph convolutional networks (GCNs) have been shown to be successful in modeling applications with graph structures. However, brain network data is in general of complex structure with small sample size, and the use of GCNs on available datasets remains a big challenge. Driven by this important issue, three questions arise: 1) how to capture the critical structures of brain networks by removing noisy connections, to facilitate the following GAN and GCN; 2) how to generate graphs by generative adversarial networks (GANs) to preserve the local graph topology as well as the global data distribution; 3) how to sufficiently leverage the real and generated graphs with domain gaps for improved classification. In this paper, we proposed a three-stage framework, named BrainGC-Net, which coherently joins the power of graph pooling, GAN and GCN for brain network generation and classification. Given the original brain network with a large number of noisy connections, we propose graph pooling to enhance the important connections with a supervision scheme. Then, based on the coarsened brain networks, we propose a graph GAN model, named EG-GAN, to focus on the global data distribution in the embedding space and local graph topology in the graph space simultaneously. Finally, a domain consistent GCN model is proposed to take sufficient advantage of the two domains rather than simply merging by incorporating multiple consistent regularizations from view correlation, class correlation and sample correlation. With extensive experiments on the ASD classification problem, we validate the effectiveness of our method and it achieves consistent improvements over state-of-the-art methods on the public ABIDE dataset.
Similar content being viewed by others
Data Availability
The datasets generated during and/or analysed during the current study are available in the ABIDE repository, https://fcon_1000.projects.nitrc.org/indi/abide/.
References
Kumar V, Garg R (2021) Resting state functional connectivity alterations in individuals with autism spectrum disorders: a systematic review. Front Psychiatry, pp 1–55
Wang M, Huang J, Liu M, Zhang D (2019) Functional connectivity network analysis with discriminative hub detection for brain disease identification. In: Proceedings of the AAAI conference on artificial intelligence, vol 33. pp 1198–1205
Kipf TN, Welling M (2017) Semi-supervised classification with graph convolutional networks. ICLR, pp 1–11
Ktena SI, Parisot S, Ferrante E, Rajchl M, Lee M, Glocker B, Rueckert D (2018) Metric learning with spectral graph convolutions on brain connectivity networks. NeuroImage 169:431–442
Li X, Dvornek NC, Zhuang J, Ventola P, Duncan JS (2018) Brain biomarker interpretation in asd using deep learning and fmri. In: International conference on medical image computing and computer-assisted intervention. Springer, pp 206–214
Parisot S, Ktena SI, Ferrante E, Lee M, Moreno RG, Glocker B, Rueckert D (2017) Spectral graph convolutions for population-based disease prediction. In: International conference on medical image computing and computer-assisted intervention. Springer, pp 177–185
Li X, Zhou Y, Dvornek N, Zhang M, Gao S, Zhuang J, Scheinost D, Staib LH, Ventola P, Duncan JS (2021) Braingnn: interpretable brain graph neural network for fmri analysis. Med Image Anal 74:102233
Jiang H, Cao P, Xu M, Yang J, Zaiane O (2020) Hi-gcn: a hierarchical graph convolution network for graph embedding learning of brain network and brain disorders prediction. Comput Biol Med 127:104096
Wen G, Cao P, Bao H, Yang W, Zheng T, Zaiane O (2022) Mvs-gcn: a prior brain structure learning-guided multi-view graph convolution network for autism spectrum disorder diagnosis. Comput Biol Med, p 105239
Wang N, Yao D, Ma L, Liu M (2022) Multi-site clustering and nested feature extraction for identifying autism spectrum disorder with resting-state fmri. Med Image Anal 75:102279
Bajestani GS, Behrooz M, Khani AG, Nouri-Baygi M, Mollaei A (2019) Diagnosis of autism spectrum disorder based on complex network features. Comput Methods Prog Biomed 177:277–283
Heinsfeld AS, Franco AR, Craddock RC, Buchweitz A, Meneguzzi F (2018) Identification of autism spectrum disorder using deep learning and the abide dataset. NeuroImage: Clinical 17:16–23
Goodfellow I, Pouget-Abadie J, Mirza M, Xu B, Warde-Farley D, Ozair S, Courville A, Bengio Y (2014) Generative adversarial nets. Adv Neural Inf Process Syst 27:1–9
Gulrajani I, Ahmed F, Arjovsky M, Dumoulin V, Courville AC (2017) Improved training of wasserstein gans. Adv Neural Inf Process Syst 30:1–11
Arjovsky M, Chintala S, Bottou L (2017) Wasserstein generative adversarial networks. In: International conference on machine learning. PMLR, pp 214–223
Dauphin YN, Fan A, Auli M, Grangier D (2017) Language modeling with gated convolutional networks. In: International conference on machine learning. PMLR, pp 933–941
Di Martino A, Yan C-G, Li Q, Denio E, Castellanos FX, Alaerts K, Anderson JS, Assaf M, Bookheimer SY, Dapretto M et al (2014) The autism brain imaging data exchange: towards a large-scale evaluation of the intrinsic brain architecture in autism. Mol Psychiatry 19:659–667
Ma Y, Wang S, Aggarwal CC, Tang J (2019) Graph convolutional networks with eigenpooling. In: Proceedings of the 25th ACM SIGKDD International conference on knowledge discovery & data mining, pp 723–731
Eslami T, Mirjalili V, Fong A, Laird AR, Saeed F (2019) Asd-diagnet: a hybrid learning approach for detection of autism spectrum disorder using fmri data. Front Neuroinform 13:70
Cao B, He L, Wei X, Xing M, Yu PS, Klumpp H, Leow AD (2017) T-bne: tensor-based brain network embedding. In: Proceedings of the 2017 SIAM International conference on data mining. SIAM, pp 189–197
Mhiri I, Rekik I (2020) Joint functional brain network atlas estimation and feature selection for neurological disorder diagnosis with application to autism. Med Image Anal 60:101596
Zhang D, Huang J, Jie B, Du J, Tu L, Liu M (2018) Ordinal pattern: A new descriptor for brain connectivity networks. IEEE Trans Med Imaging 37(7):1711–1722
Kawahara J, Brown CJ, Miller SP, Booth BG, Chau V, Grunau RE, Zwicker JG, Hamarneh G (2017) Brainnetcnn: convolutional neural networks for brain networks; towards predicting neurodevelopment. NeuroImage 146:1038–1049
Rosca M, Lakshminarayanan B, Warde-Farley D, Mohamed S (2017) Variational approaches for auto-encoding generative adversarial networks, pp 1–21
Makhzani A, Shlens J, Jaitly N, Goodfellow I, Frey B (2015) Adversarial autoencoders. Int Conf Learn Representations. pp 1–6
Pan S, Hu R, Long G, Jiang J, Yao L, Zhang C (2018) Adversarially regularized graph autoencoder for graph embedding. Int Jt Conf Artif Intell, pp 2609–2615
Acknowledgements
This research was supported by the National Natural Science Foundation of China (No.62076059) and the Science Project of Liaoning province (2021-MS-105).
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of Interests
The authors declare that they have no conflicts of interest.
Additional information
Publisher’s note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Cao, P., Wen, G., Yang, W. et al. A unified framework of graph structure learning, graph generation and classification for brain network analysis. Appl Intell 53, 6978–6991 (2023). https://doi.org/10.1007/s10489-022-03891-9
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10489-022-03891-9