用3D视觉变换器与图神经网络,结合脑图谱提升抑郁症影像诊断准确率。
3DViT-GAT: A Unified Atlas-Based 3D Vision Transformer and Graph Learning Framework for Major Depressive Disorder Detection Using Structural MRI Data
- 融合3D ViT与GNN,从结构MRI中提取脑区特征并建模区域间关系。
- 在REST-meta-MDD数据集上达到81.51%准确率,敏感性达85.94%。
- 基于脑图谱的区域划分优于无先验的立方体方法,适合医学影像分析研究者。
重度抑郁障碍(MDD)是一种普遍的精神健康问题,对个人福祉和全球公共卫生造成负面影响。利用结构磁共振成像(sMRI)与深度学习(DL)实现自动化检测,有望提高诊断准确性并支持早期干预。现有方法多依赖体素级特征或预定义脑图谱的手工区域表示,难以捕捉复杂脑部模式。本文提出一种统一框架:使用视觉变压器(ViTs)从sMRI中提取3D脑区嵌入,并通过图神经网络(GNN)进行分类。探索两种区域定义策略:(1) 基于预定义解剖与功能脑图谱的图谱法;(2) 通过训练ViT直接从均匀提取的3D块中识别区域的立方体法。进一步,基于余弦相似性构建脑区间关系图,引导GNN分类。在REST-meta-MDD数据集上进行大量实验,采用分层10折交叉验证,最佳模型获得81.51%准确率、85.94%敏感性、76.36%特异性、80.88%精确率和83.33%F1分数。结果表明,基于图谱的方法始终优于立方体法,凸显使用领域特定解剖先验对MDD检测的重要性。
原文摘要 · Abstract (English)
Major depressive disorder (MDD) is a prevalent mental health condition that negatively impacts both individual well-being and global public health. Automated detection of MDD using structural magnetic resonance imaging (sMRI) and deep learning (DL) methods holds increasing promise for improving diagnostic accuracy and enabling early intervention. Most existing methods employ either voxel-level features or handcrafted regional representations built from predefined brain atlases, limiting their ability to capture complex brain patterns. This paper develops a unified pipeline that utilizes Vision Transformers (ViTs) for extracting 3D region embeddings from sMRI data and Graph Neural Network (GNN) for classification. We explore two strategies for defining regions: (1) an atlas-based approach using predefined structural and functional brain atlases, and (2) an cube-based method by which ViTs are trained directly to identify regions from uniformly extracted 3D patches. Further, cosine similarity graphs are generated to model interregional relationships, and guide GNN-based classification. Extensive experiments were conducted using the REST-meta-MDD dataset to demonstrate the effectiveness of our model. With stratified 10-fold cross-validation, the best model obtained 81.51\% accuracy, 85.94\% sensitivity, 76.36\% specificity, 80.88\% precision, and 83.33\% F1-score. Further, atlas-based models consistently outperformed the cube-based approach, highlighting the importance of using domain-specific anatomical priors for MDD detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。