NeuroPath: A Neural Pathway Transformer for Joining the Dots of Human Connectomes

TL;DR

NeuroPath is a biologically-inspired Transformer model that captures complex structural and functional brain connectivity patterns to improve understanding and prediction of cognitive functions and diseases.

Contribution

The paper introduces NeuroPath, a novel Transformer-based model that incorporates topological detours to better represent brain connectomes by integrating structural and functional connectivity.

Findings

NeuroPath achieves state-of-the-art results on HCP and UK Biobank datasets.

The model effectively captures high-order topological features of brain networks.

NeuroPath demonstrates strong performance in task recognition and disease diagnosis.

Abstract

Although modern imaging technologies allow us to study connectivity between two distinct brain regions in-vivo, an in-depth understanding of how anatomical structure supports brain function and how spontaneous functional fluctuations emerge remarkable cognition is still elusive. Meanwhile, tremendous efforts have been made in the realm of machine learning to establish the nonlinear mapping between neuroimaging data and phenotypic traits. However, the absence of neuroscience insight in the current approaches poses significant challenges in understanding cognitive behavior from transient neural activities. To address this challenge, we put the spotlight on the coupling mechanism of structural connectivity (SC) and functional connectivity (FC) by formulating such network neuroscience question into an expressive graph representation learning problem for high-order topology. Specifically, we introduce the concept of topological detour to characterize how a ubiquitous instance of FC (direct link) is supported by neural pathways (detour) physically wired by SC, which forms a cyclic loop interacted by brain structure and function. In the clich\'e of machine learning, the multi-hop detour pathway underlying SC-FC coupling allows us to devise a novel multi-head self-attention mechanism within Transformer to capture multi-modal feature representation from paired graphs of SC and FC. Taken together, we propose a biological-inspired deep model, coined as NeuroPath, to find putative connectomic feature representations from the unprecedented amount of neuroimages, which can be plugged into various downstream applications such as task recognition and disease diagnosis. We have evaluated NeuroPath on large-scale public datasets including HCP and UK Biobank under supervised and zero-shot learning, where the state-of-the-art performance by our NeuroPath indicates great potential in network neuroscience.