LLM2D
基于EEG的阿尔茨海默病检测的大规模基础模型
LEAD: Large Foundation Model for EEG-Based Alzheimer's Disease Detection
作者: Yihe Wang, Nan Huang, Nadia Mammone, Marco Cecchi, Xiang Zhang
发布日期: 2/5/2025
arXiv ID: 2502.01678

摘要

arXiv:2502.01678v1 交叉公告类型:跨学科 摘要:脑电图(EEG)提供了一种无创、高度可访问且成本效益高的解决方案,用于阿尔茨海默病(AD)的检测。然而,现有的方法,无论是基于手动特征提取还是深度学习,面临着两大挑战:缺乏大规模数据集来进行稳健的特征学习和评估,以及由于被试间变化而导致的检测性能较差。为了解决这些挑战,我们整理了一个包含813名被试的EEG-AD语料库,这是我们目前所知的最大的EEG-AD数据集。使用这一独特的数据集,我们提出了LEAD,这是第一个基于EEG的AD检测的大规模基础模型。我们的方法涵盖了从数据选择和预处理到自我监督对比预训练、微调以及被试独立评估和被试级别多数投票的整个管道。我们在11个EEG数据集上预训练模型,并在5个AD数据集上统一微调。我们自我监督的预训练设计包括样本级和被试级对比,以提取有用的通用EEG特征。在5个通道对齐的数据集上进行了微调。我们的骨干编码器结合了时域和通道嵌入,以捕捉跨时空维度的特征。我们的方法展示了卓越的AD检测性能,与最先进的方法相比,在样本级别实现了高达9.86%的F1分数增加,在被试级别实现了高达9.31%的增加。我们模型的结果强烈证实了对比预训练和通道对齐统一微调在解决被试间变化中的有效性。源代码可在https://github.com/DL4mHealth/LEAD。