Template-based graph registration network for boosting the diagnosis of brain connectivity disorders.

Abstract

Brain graphs are powerful representations to explore the biological roadmaps of the human brain in its healthy and disordered states. Recently, a few graph neural networks (GNNs) have been designed for brain connectivity synthesis and diagnosis. However, such non-Euclidean deep learning architectures might fail to capture the neural interactions between different brain regions as they are trained without guidance from any prior biological template-i.e., template-free learning. Here we assume that using a population-driven brain connectional template (CBT) that captures well the connectivity patterns fingerprinting a given brain state (e.g., healthy) can better guide the GNN training in its downstream learning task such as classification or regression. To this aim we design a plug-in graph registration network (GRN) that can be coupled with any conventional graph neural network (GNN) so as to boost its learning accuracy and generalizability to unseen samples. Our GRN is a graph generative adversarial network (gGAN), which registers brain graphs to a prior CBT. Next, the registered brain graphs are used to train typical GNN models. Our GRN can be integrated into any GNN working in an end-to-end fashion to boost its prediction accuracy. Our experiments showed that GRN remarkably boosted the prediction accuracy of four conventional GNN models across four neurological datasets.