Long-range Brain Graph Transformer
Journal:
arXiv
Published Date:
Jan 2, 2025
Abstract
Understanding communication and information processing among brain regions of
interest (ROIs) is highly dependent on long-range connectivity, which plays a
crucial role in facilitating diverse functional neural integration across the
entire brain. However, previous studies generally focused on the short-range
dependencies within brain networks while neglecting the long-range
dependencies, limiting an integrated understanding of brain-wide communication.
To address this limitation, we propose Adaptive Long-range aware TransformER
(ALTER), a brain graph transformer to capture long-range dependencies between
brain ROIs utilizing biased random walk. Specifically, we present a novel
long-range aware strategy to explicitly capture long-range dependencies between
brain ROIs. By guiding the walker towards the next hop with higher correlation
value, our strategy simulates the real-world brain-wide communication.
Furthermore, by employing the transformer framework, ALERT adaptively
integrates both short- and long-range dependencies between brain ROIs, enabling
an integrated understanding of multi-level communication across the entire
brain. Extensive experiments on ABIDE and ADNI datasets demonstrate that ALTER
consistently outperforms generalized state-of-the-art graph learning methods
(including SAN, Graphormer, GraphTrans, and LRGNN) and other graph learning
based brain network analysis methods (including FBNETGEN, BrainNetGNN,
BrainGNN, and BrainNETTF) in neurological disease diagnosis. Cases of
long-range dependencies are also presented to further illustrate the
effectiveness of ALTER. The implementation is available at
https://github.com/yushuowiki/ALTER.