LLM2D
LEAD: 基于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: oai:arXiv.org:2502.01678v1

摘要

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