Enhancing Alzheimer's Diagnosis: Leveraging Anatomical Landmarks in Graph Convolutional Neural Networks on Tetrahedral Meshes
作者: Yanxi Chen, Mohammad Farazi, Zhangsihao Yang, Yonghui Fan, Nicholas Ashton, Eric M Reiman, Yi Su, Yalin Wang
发布时间: 2025-03-10
来源: arxiv
研究方向: 阿尔茨海默病诊断与脑部淀粉样蛋白阳性分类
主要内容
该研究提出了一种基于Transformer架构的几何深度学习模型,用于阿尔茨海默病的诊断和脑部淀粉样蛋白阳性分类。该模型通过结合结构磁共振成像(sMRI)和基于血液的生物标志物(BBBMs),旨在提高诊断准确率,同时减少对昂贵的PET扫描的依赖。
主要贡献
1. 提出了首个适用于不同大小和拓扑结构的四面体网格的Transformer模型。
2. 采用了稀疏局部注意力策略,显著降低了计算成本。
3. 引入了节点级特征学习模块,提高了性能和局部上下文感知。
4. 在ADNI数据集的子集上对多个任务进行了严格评估,一致性地优于最先进的方法。
5. 将BBBMs与学习到的特征相结合,显著提高了性能,突出了sMRI分析在AD研究中的互补作用。
研究方法
1. 基于高斯过程(GP)模型的解剖标志物生成。
2. 四面体网格的新型标记化方案。
3. 节点和图级别的嵌入。
4. Transformer架构,包括自注意力机制和稀疏图注意力。
5. Ablation研究以验证模型组件的贡献。
实验结果
在阿尔茨海默病分类任务中,LETetCNN模型在所有临床诊断组对中均优于现有方法,特别是在MCI vs CN比较中表现出色。在脑部淀粉样蛋白阳性分类任务中,LETetCNN模型优于所有基线模型,尤其是当结合pTau-217信息时,表现更佳。
未来工作
探索多尺度标记化策略、集成额外的解剖先验以及在其他需要灵活表示学习的3D结构领域的应用。