用自注意力机制分析脑连接图,准确识别阿尔茨海默病。
An Attention-Based Framework for Alzheimers Disease Classification Using Resting-State fMRI

- 将脑区视为令牌,用Transformer自注意力建模远距离脑网络关联
- 在ADNI数据集上达88.95%准确率,AUC为0.90,平衡了精准与召回
- 无需人工特征工程,适合神经影像与临床辅助诊断研究者
基于静息态功能磁共振成像(rs-fMRI)准确识别阿尔茨海默病(AD)仍面临挑战,因其功能脑连接具有高维、噪声大及区域间复杂依赖关系,传统手工特征或机器学习方法效果有限。本文提出一种基于注意力的深度学习框架,直接处理rs-fMRI功能连接矩阵,将脑区视为令牌,采用类Transformer自注意力机制建模跨分布式脑网络的长程与全局依赖关系。该框架无需依赖人工特征工程,可自动学习判别性功能表征,在阿尔茨海默病神经影像倡议(ADNI)纵向队列数据上进行评估,包含认知正常与阿尔茨海默病受试者多时间点数据。采用个体级评估协议防止时间点间信息泄露,并引入类别加权优化缓解轻微类别不平衡。二分类任务(AD vs 认知正常)实验结果显示,该注意力模型达到88.95%准确率和0.90的ROC-AUC,具备良好的精确率-召回率平衡,验证了自注意力驱动的功能连接建模在基于静息态fMRI的阿尔茨海默病检测中具有鲁棒性和可解释性。
原文摘要 · Abstract (English)
Accurate identification of Alzheimers disease (AD) using resting-state functional magnetic resonance imaging (rs-fMRI) remains challenging due to the high dimensionality, noise, and complex inter-regional dependencies inherent in functional brain connectivity, which limit the effectiveness of traditional approaches based on handcrafted connectivity features or conventional machine learning models. In this work, we present an attention-based deep learning framework for Alzheimers disease classification that operates directly on rs-fMRI functional connectivity matrices by treating brain regions as tokens and employing a Transformer-inspired self-attention mechanism to model long-range and global functional dependencies across distributed brain networks. The proposed framework learns discriminative functional representations without reliance on manual feature engineering and is evaluated on a longitudinal cohort from the Alzheimers Disease Neuroimaging Initiative (ADNI) comprising cognitively normal and Alzheimers disease subjects with multiple visits. A subject-wise evaluation protocol is adopted to prevent information leakage across visits, and class-weighted optimization is incorporated to address mild class imbalance. Experimental results for binary AD versus cognitively normal classification demonstrate that the proposed attention- based rs-fMRI model achieves an accuracy of 88.95% and a ROC-AUC of 0.90, along with a favorable precision-recall balance, highlighting the effectiveness of self-attention-driven functional connectivity modeling as a robust and interpretable approach for Alzheimers disease detection using resting-state fMRI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。