arXiv:2503.14655q-bio.NCcs.AI2025-03被引 7

用脑网络核心-外围结构设计新模型,高效分类神经疾病

Core-Periphery Principle Guided State Space Model for Functional Connectome Classification

  • 基于脑网络核心-外围结构设计轻量级状态空间模型
  • 在两个数据集上准确率超Transformer,计算量更低
  • 适合脑影像分析、神经疾病诊断研究者使用

理解人脑网络的组织结构是神经科学的核心课题,尤其在功能连接研究中对神经疾病诊断至关重要。尽管功能性磁共振成像和机器学习技术进步显著提升了脑网络分析能力,但传统机器学习难以捕捉脑区间复杂关系,而基于Transformer的深度学习方法因长序列建模存在二次复杂度,计算开销大。为此,我们提出核心-外围状态空间模型(CP-SSM),利用具有线性复杂度的Mamba模型有效捕获功能脑网络中的长程依赖。同时,受脑网络核心-外围(CP)组织特性启发,设计了CP-MoE——一种引导型专家混合模型,增强脑连接模式的表征学习能力。在ABIDE与ADNI两个基准fMRI数据集上的实验表明,CP-SSM在分类性能上优于基于Transformer的模型,同时显著降低计算复杂度。结果验证了该模型在建模脑功能连接方面的有效性与高效性,为基于神经影像的神经疾病诊断提供了新方向。

原文摘要 · Abstract (English)

Understanding the organization of human brain networks has become a central focus in neuroscience, particularly in the study of functional connectivity, which plays a crucial role in diagnosing neurological disorders. Advances in functional magnetic resonance imaging and machine learning techniques have significantly improved brain network analysis. However, traditional machine learning approaches struggle to capture the complex relationships between brain regions, while deep learning methods, particularly Transformer-based models, face computational challenges due to their quadratic complexity in long-sequence modeling. To address these limitations, we propose a Core-Periphery State-Space Model (CP-SSM), an innovative framework for functional connectome classification. Specifically, we introduce Mamba, a selective state-space model with linear complexity, to effectively capture long-range dependencies in functional brain networks. Furthermore, inspired by the core-periphery (CP) organization, a fundamental characteristic of brain networks that enhances efficient information transmission, we design CP-MoE, a CP-guided Mixture-of-Experts that improves the representation learning of brain connectivity patterns. We evaluate CP-SSM on two benchmark fMRI datasets: ABIDE and ADNI. Experimental results demonstrate that CP-SSM surpasses Transformer-based models in classification performance while significantly reducing computational complexity. These findings highlight the effectiveness and efficiency of CP-SSM in modeling brain functional connectivity, offering a promising direction for neuroimaging-based neurological disease diagnosis.

脑网络状态空间模型功能连接神经疾病

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