arXiv:2409.00032eess.SPcs.CE2024-09被引 5

无需人工特征提取,直接从原始脑电数据中端到端识别阿尔茨海默病。

ADformer: A Multi-Granularity Spatial-Temporal Transformer for EEG-Based Alzheimer Detection

  • 设计多粒度时空变换器,同时捕捉脑电图的局部与全局特征。
  • 在1713名受试者上测试,对阿尔茨海默病的识别准确率达92.82%。
  • 适用于大规模、多样人群数据,适合临床实际应用。

脑电图(EEG)已成为辅助神经科医生检测阿尔茨海默病(AD)的一种经济高效工具。然而,现有方法大多依赖人工特征工程或数据变换,小规模数据下可能有效,但在大规模数据中常导致信息丢失和失真,限制模型性能。此外,以往研究使用的数据集受试者数量有限且人群多样性不足,难以全面评估模型的鲁棒性与泛化能力,限制了其在真实临床环境中的应用。为此,我们提出ADformer,一种新型多粒度时空变换器,可直接从原始脑电数据中端到端学习时空特征表示。该模型在空间和时间维度均引入多粒度嵌入策略,并采用两阶段内-跨粒度自注意力机制,学习各粒度内的局部模式及粒度间的全局依赖关系。我们在包含总计1713名受试者的4个大规模数据集上进行评估,采用交叉验证的受试者独立设置。实验结果表明,ADformer显著优于现有方法,在4个数据集上的受试者级F1得分分别为92.82%、89.83%、67.99%和83.98%,有效区分阿尔茨海默病患者与健康对照组。

原文摘要 · Abstract (English)

Electroencephalography (EEG) has emerged as a cost-effective and efficient tool to support neurologists in the detection of Alzheimer's Disease (AD). However, most existing approaches rely heavily on manual feature engineering or data transformation. While such techniques may provide benefits when working with small-scale datasets, they often lead to information loss and distortion when applied to large-scale data, ultimately limiting model performance. Moreover, the limited subject scale and demographic diversity of datasets used in prior studies hinder comprehensive evaluation of model robustness and generalizability, thus restricting their applicability in real-world clinical settings. To address these challenges, we propose ADformer, a novel multi-granularity spatial-temporal transformer designed to capture both temporal and spatial features from raw EEG signals, enabling effective end-to-end representation learning. Our model introduces multi-granularity embedding strategies across both spatial and temporal dimensions, leveraging a two-stage intra-inter granularity self-attention mechanism to learn both local patterns within each granularity and global dependencies across granularities. We evaluate ADformer on 4 large-scale datasets comprising a total of 1,713 subjects, representing one of the largest corpora for EEG-based AD detection to date, under a cross-validated, subject-independent setting. Experimental results demonstrate that ADformer consistently outperforms existing methods, achieving subject-level F1 scores of 92.82%, 89.83%, 67.99%, and 83.98% on the 4 datasets, respectively, in distinguishing AD from healthy control (HC) subjects.

阿尔茨海默病脑电图深度学习多粒度

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。