STARFormer通过重构脑区连接与时间序列,提升自闭症和多动症诊断准确率
STARFormer: A Novel Spatio-Temporal Aggregation Reorganization Transformer of FMRI for Brain Disorder Diagnosis
- 基于特征中心性重排脑区,捕捉关键空间关系
- 分段时间窗口+跨窗口注意力,提取多尺度时间特征
- 双分支并行架构融合时空特征,适合神经疾病诊断研究
现有基于功能磁共振成像(fMRI)的脑病分类方法常忽略血氧水平依赖(BOLD)信号的空间与时间关联性,导致分类不准确。为此,本文提出星型时空聚合重组变换器(STARFormer),通过三个核心模块有效捕捉BOLD信号的时空特征:区域兴趣(ROI)空间结构分析模块利用特征中心性(EC)根据有效连接重排脑区,突出与脑病相关的关键空间关系;时间特征重组模块将时间序列划分为等维窗口片段,通过可变窗口与跨窗口注意力机制捕获多尺度特征;时空特征融合模块采用并行变压器架构,分别处理时空分支以提取整合特征。在两个公开数据集上对自闭症(ASD)和多动症(ADHD)分类任务的严格评估表明,STARFormer在多个指标上达到当前最优性能,为脑病诊断与生物医学研究提供了更精准可靠的工具。代码已开源:https://github.com/NZWANG/STARFormer。
原文摘要 · Abstract (English)
Many existing methods that use functional magnetic resonance imaging (fMRI) classify brain disorders, such as autism spectrum disorder (ASD) and attention deficit hyperactivity disorder (ADHD), often overlook the integration of spatial and temporal dependencies of the blood oxygen level-dependent (BOLD) signals, which may lead to inaccurate or imprecise classification results. To solve this problem, we propose a Spatio-Temporal Aggregation eorganization ransformer (STARFormer) that effectively captures both spatial and temporal features of BOLD signals by incorporating three key modules. The region of interest (ROI) spatial structure analysis module uses eigenvector centrality (EC) to reorganize brain regions based on effective connectivity, highlighting critical spatial relationships relevant to the brain disorder. The temporal feature reorganization module systematically segments the time series into equal-dimensional window tokens and captures multiscale features through variable window and cross-window attention. The spatio-temporal feature fusion module employs a parallel transformer architecture with dedicated temporal and spatial branches to extract integrated features. The proposed STARFormer has been rigorously evaluated on two publicly available datasets for the classification of ASD and ADHD. The experimental results confirm that the STARFormer achieves state-of-the-art performance across multiple evaluation metrics, providing a more accurate and reliable tool for the diagnosis of brain disorders and biomedical research. The codes are available at: https://github.com/NZWANG/STARFormer.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。