NOMAD-EEG: Native Omni-Montage Adaptive Diffusion

ABSTRACT

中文摘要

EEG 的蒙太奇、通道数量和记录长度差异很大，但多数 EEG 基础模型仍绑定于训练时使用的通道集合和输入长度。作者旨在构建一个不依赖电极数量、身份、排列顺序或输入长度的生成式 EEG 基础模型。NOMAD-EEG 是一个 1,200 万参数的扩散模型：模型沿时间方向交替使用不分 patch 的状态空间层，并在电极之间执行注意力；每个电极仅由其三维头皮坐标标识。模型在单一语料的 147 小时无标签 EEG 上预训练，并以冻结线性探针在十个含 1–64 个电极的数据集上与七个基础模型按统一协议比较，同时测试两个未见人群。尽管规模较小、预训练只来自一个语料，NOMAD-EEG 在 EEGMMIDB 运动想象、Siena 癫痫检测和 Bonn 癫痫检测上优于全部七个比较模型，在 TUEP 癫痫诊断上取得比较模型中的最高准确率；微调后，它还在 TUEV 事件分类上取得最高的 κ 和加权 F1。在一个未见的八电极自闭症人群 P300 检测任务上，它在五个模型中表现最佳；把 19 个电极减至 5 个时，平均 AUROC 仅从 0.942 降至 0.891。结果表明，以空间位置标识电极并连续建模时间，可以让小型单语料预训练模型在多项任务上超过更大的 EEG 基础模型。