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

Cao Peng,Wen Guangqi,Yang Wenju,Liu Xiaoli,Yang Jinzhu,Zaiane Osmar
DOI: https://doi.org/10.1007/s10489-022-03891-9
IF: 5.3
2022-01-01
Applied Intelligence
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.
What problem does this paper attempt to address?