用双曲空间建模脑网络层次结构,提升疾病诊断准确率
Navigating Hierarchy: Hyperbolic Learning on Brain Graphs for Disorder Diagnosis

- 将脑区、功能社区与全脑网络映射到双曲空间,显式建模层级关系
- 在ABIDE-I和REST-MDD数据集上达到92.3%和89.7%的分类准确率
- 适合脑网络分析、精神疾病诊断及生物标志物挖掘的研究者
功能脑网络在脑区(ROI)、社区和全脑层面具有层次化组织,支持局部处理、跨社区协调与全局整合。近期研究证实,考虑脑区社区结构对脑网络的诊断与生物标志物识别有益。然而,现有脑图建模方法常难以捕捉脑区-社区间的交互,无法充分挖掘多层级结构信息。为此,受深层双曲学习启发,我们提出一种新框架——脑图双曲学习(HLBG),利用双曲空间固有的层次几何特性,建模脑区、功能社区与全脑网络间的层级关系,学习具有层次感知且高度区分性的脑网络表示。具体地,HLBG首先将脑区、社区及全脑网络的表示投影至洛伦兹双曲空间,再通过两个几何蕴含约束施加多层次结构。此外,引入新型图感知Mamba(GaMamba)模型,融合拓扑生成的结构提示,以捕捉长程依赖同时保留图拓扑信息。在ABIDE-I与REST-MDD数据集上的实验表明,HLBG优于现有先进方法,并识别出与疾病相关的功能生物标志物。
原文摘要 · Abstract (English)
Functional brain networks exhibit a hierarchical organization across ROI, community, and whole-brain levels, supporting local processing, inter-community coordination, and global integration. Recent studies have demonstrated that brain community-aware modeling is beneficial for both diagnosis and biomarker identification of brain networks. However, existing brain graph modeling methods often struggle to model ROI-community interactions, thereby failing to fully exploit the hierarchy across ROI, community, and whole-brain network levels. To address this issue, inspired by deep hyperbolic learning in modeling hierarchical structures, we propose a novel framework, termed Hyperbolic Learning on Brain Graphs (HLBG), for brain network analysis. The core idea of HLBG is to exploit the inherent hierarchical geometry of hyperbolic space to model the hierarchical relationships among ROIs, functional communities, and the whole-brain network, thereby learning hierarchy-aware and highly discriminative representations for brain network data. Specifically, HLBG first projects representations from ROIs, communities, and the whole-brain network into Lorentzian hyperbolic space. Then, the multi-level hierarchy is imposed via two geometric entailment constraints. In addition, we introduce a new Graph-aware Mamba (GaMamba) model, which incorporates topology-derived structural prompts into Mamba to capture long-range dependencies while preserving graph topological information. Experiments on ABIDE-I and REST-MDD demonstrate that HLBG outperforms state-of-the-art methods and identifies disorder-relevant functional biomarkers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。