EAD让不同设备采集的脑电数据统一分类,准确率超99%。
EAD: An EEG Adapter for Automated Classification
- 用适配器架构融合多设备脑电信号,支持任意通道数输入。
- 在两个数据集上达到99.33%和92.31%的分类准确率。
- 支持零样本分类,适合跨设备、多任务脑电分析场景。
尽管脑电图(EEG)是神经解码的常用模态,但其数据采集常依赖特定任务与设备,导致难以构建统一的嵌入学习管道。传统方法对信号预处理和深度学习模型依赖性强,且受每样本电极数影响,不同设备采集的数据无法直接复用。为此,本文提出一种通用框架EEG Adapter(EAD),基于近期的脑电基础模型进行适配,可从不同设备获取的脑电信号中学习鲁棒表征。我们在两个公开数据集上验证了EAD,分别在EEG-ImageNet和BrainLat上取得99.33%和92.31%的准确率。该结果表明该框架能有效应对两类感知任务:刺激诱发与静息态脑电。此外,我们在EEG-ImageNet任务上实现了零样本分类,验证了其强大的泛化能力。
原文摘要 · Abstract (English)
While electroencephalography (EEG) has been a popular modality for neural decoding, it often involves task specific acquisition of the EEG data. This poses challenges for the development of a unified pipeline to learn embeddings for various EEG signal classification, which is often involved in various decoding tasks. Traditionally, EEG classification involves the step of signal preprocessing and the use of deep learning techniques, which are highly dependent on the number of EEG channels in each sample. However, the same pipeline cannot be applied even if the EEG data is collected for the same experiment but with different acquisition devices. This necessitates the development of a framework for learning EEG embeddings, which could be highly beneficial for tasks involving multiple EEG samples for the same task but with varying numbers of EEG channels. In this work, we propose EEG Adapter (EAD), a flexible framework compatible with any signal acquisition device. More specifically, we leverage a recent EEG foundational model with significant adaptations to learn robust representations from the EEG data for the classification task. We evaluate EAD on two publicly available datasets achieving state-of-the-art accuracies 99.33% and 92.31% on EEG-ImageNet and BrainLat respectively. This illustrates the effectiveness of the proposed framework across diverse EEG datasets containing two different perception tasks: stimulus and resting-state EEG signals. We also perform zero-shot EEG classification on EEG-ImageNet task to demonstrate the generalization capability of the proposed approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。