提出EEG-PRIME模型,实现跨被试、跨数据集的高效脑电解码。
EEG-PRIME: Prototype-Aligned Representation Learning with Multi-Level Conditioning for EEG Decoding

- 分两阶段训练:先掩码重建预训练,再用原型对齐指令微调。
- 在16个数据集上跨被试性能超越现有基线,零样本迁移表现接近有校准模型。
- 适合需要少标注、强泛化的脑机接口应用开发者。
脑电图(EEG)解码模型常因采集协议和个体神经生理差异导致域偏移,泛化能力差。本文提出EEG-PRIME,一种两阶段的脑电基础模型,用于跨数据集多任务解码。预训练阶段,通过频率截断谱增强的掩码重建,使EEG编码器学习可迁移表示;指令微调阶段,引入任务语义、数据集特异性和被试无关性条件信号,通过逐层查询调制调控Q-Former,并利用类别标签冻结文本嵌入作为原型,基于余弦相似度实现异构标签空间预测。在涵盖运动想象、情绪识别、注意力缺陷、隐性言语和心理负荷等16个数据集上的实验表明,该模型在跨被试设置下持续优于当前最优基线及已有脑电基础模型。在两个额外保留数据集上,无需目标域优化、校准或线性探针即可达到与会话内校准模型相当的平衡准确率,展现出优异的零样本迁移能力。
原文摘要 · Abstract (English)
Electroencephalography (EEG) decoding models often generalize poorly across datasets and subjects due to domain shifts in acquisition protocols and individual neurophysiology. We propose EEG-PRIME, a two-stage EEG foundation model for cross-dataset multi-task decoding. EEG-PRIME combines masked pretraining with prototype-aligned instruction tuning to enable instruction-aware and subject-invariant decoding across diverse BCI paradigms. During pretraining, an EEG encoder learns transferable representations through masked reconstruction with frequency-cutoff spectral augmentation. During instruction tuning, EEG-PRIME incorporates task-semantic, dataset-specific, and subject-invariant conditioning. The resulting conditioning signal modulates the Q-Former through Layer-wise Query Modulation, while frozen text embeddings of class labels serve as prototypes for cosine-similarity-based prediction across heterogeneous label spaces. Experiments on sixteen datasets covering motor imagery, emotion recognition, ADHD detection, covert speech, and mental workload show consistent improvements over state-of-the-art baselines and prior EEG foundation models under cross-subject settings. On two additional held-out datasets, EEG-PRIME achieves balanced accuracy comparable to within-session calibration models without target-domain optimization, calibration, or linear probing, demonstrating promising zero-shot transfer capability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。