LLM2D
基于ROI-rank基于掩码的局部时空特征增强变换器用于ADHD诊断
Local Temporal Feature Enhanced Transformer with ROI-rank Based Masking for Diagnosis of ADHD
作者: Byunggun Kim, Younghun Kwon
发布日期: 4/17/2025
arXiv ID: oai:arXiv.org:2504.11474v1

摘要

arXiv:2504.11474v1 类型: cross 摘要:在现代社会中,注意缺陷多动障碍(ADHD)是不仅在儿童中,而在成人中也发现的一种常见的精神疾病。在此背景下,我们提出了一种ADHD诊断变换器模型,该模型可以从静息态功能性磁共振成像(rs-fMRI)中有效地同时找到重要的脑空间-时间生物标志物。该模型不仅学习空间-时间个体特征,还学习与全注意力机制相关的ADHD诊断结构之间的关联。特别是,它专注于学习局部血氧水平依赖(BOLD)信号,并区分脑中的重要感兴趣区域(ROI)。具体来说,ADHD诊断变换器提出的方法如下。首先,我们设计了一个基于CNN的嵌入块,以在脑区注意力中获得更具表现力的嵌入特征。该块是基于先前的基于CNN的ADHD诊断模型构建的,用于变换器。其次,在个体空间-时间特征注意力方面,我们将注意力方法变为局部时间注意力和基于ROI排名的遮罩。对于fMRI的时间特征,局部时间注意力仅通过简单的窗口遮罩就能学习局部BOLD信号特征。对于fMRI的空间特征,基于ROI排名的遮罩可以根据注意力分数区分具有高相关性的ROI关系,从而提供更具体的ADHD诊断生物标志物。我们使用了多种类型的变换器模型进行了实验。为了评估这些模型,我们收集了ADHD-200竞赛所提供来自939个个体的数据。通过这种方式,ADHD诊断的空间-时间增强变换器在其他不同类型的变换器变体中的性能表现更优。(准确率77.78、特异性76.60、敏感性79.22、AUC 79.30)