arXiv:2501.01100cs.LG2025-01NeurIPS被引 28

用随机游走捕捉脑区长程连接,提升神经疾病诊断准确率

Long-range Brain Graph Transformer

  • 设计新型长程感知策略,通过相关性引导游走路径模拟真实脑通信
  • 在ABIDE和ADNI数据集上,诊断准确率超越现有最先进方法
  • 适合脑网络分析、神经疾病建模的研究者使用

理解感兴趣脑区(ROIs)之间的通信与信息处理依赖于长程连接,其在全脑功能整合中起关键作用。然而,以往研究多关注脑网络内的短程依赖,忽视长程依赖,限制了对全脑通信的整合理解。为此,我们提出自适应长程感知变换器(ALTER),利用偏置随机游走捕捉脑区间的长程依赖。具体而言,通过引导游走者向相关性更高的下一跳前进,该策略模拟真实脑区间通信。结合Transformer框架,ALTER可自适应融合短程与长程依赖,实现对全脑多层次通信的整合理解。在ABIDE和ADNI数据集上的大量实验表明,ALTER在神经疾病诊断任务中持续优于现有最先进图学习方法(包括SAN、Graphormer、GraphTrans、LRGNN)及其他基于图学习的脑网络分析方法(如FBNETGEN、BrainNetGNN、BrainGNN、BrainNETTF)。案例分析进一步验证了长程依赖的有效性。代码已开源:https://github.com/yushuowiki/ALTER。

原文摘要 · Abstract (English)

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.

脑网络图神经网络长程连接Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。