Recurrent Brain Graph Mapper for Predicting Time-Dependent Brain Graph Evaluation Trajectory

Alpay Tekin,Ahmed Nebli,Islem Rekik

DOI: https://doi.org/10.48550/arXiv.2110.11237

2021-10-06

Abstract:Several brain disorders can be detected by observing alterations in the brain's structural and functional connectivities. Neurological findings suggest that early diagnosis of brain disorders, such as mild cognitive impairment (MCI), can prevent and even reverse its development into Alzheimer's disease (AD). In this context, recent studies aimed to predict the evolution of brain connectivities over time by proposing machine learning models that work on brain images. However, such an approach is costly and time-consuming. Here, we propose to use brain connectivities as a more efficient alternative for time-dependent brain disorder diagnosis by regarding the brain as instead a large interconnected graph characterizing the interconnectivity scheme between several brain regions. We term our proposed method Recurrent Brain Graph Mapper (RBGM), a novel efficient edge-based recurrent graph neural network that predicts the time-dependent evaluation trajectory of a brain graph from a single baseline. Our RBGM contains a set of recurrent neural network-inspired mappers for each time point, where each mapper aims to project the ground-truth brain graph onto its next time point. We leverage the teacher forcing method to boost training and improve the evolved brain graph quality. To maintain the topological consistency between the predicted brain graphs and their corresponding ground-truth brain graphs at each time point, we further integrate a topological loss. We also use l1 loss to capture time-dependency and minimize the distance between the brain graph at consecutive time points for regularization. Benchmarks against several variants of RBGM and state-of-the-art methods prove that we can achieve the same accuracy in predicting brain graph evolution more efficiently, paving the way for novel graph neural network architecture and a highly efficient training scheme.

Neurons and Cognition,Computer Vision and Pattern Recognition,Machine Learning

What problem does this paper attempt to address?

The problem that this paper attempts to solve is to predict the trajectory of the brain connectivity graph over time, especially during the development of brain diseases. Specifically, the authors propose a new method - the Recurrent Brain Graph Mapper (RBGM), which aims to efficiently predict the time - dependent evolution trajectory of the brain connectivity graph from a single baseline observation point. This method focuses particularly on reducing computational complexity and training time while maintaining prediction accuracy comparable to existing state - of - the - art methods. RBGM achieves this by regarding the brain as a large interconnected graph, where each region is a node and the connections between regions are edges, so as to characterize the interconnection patterns between different regions of the brain. Through this method, researchers hope to diagnose time - dependent brain diseases more effectively, such as the evolution process from mild cognitive impairment (MCI) to Alzheimer's disease (AD), thereby providing support for early diagnosis and treatment.