BINE022/EEGPT
Official code repository for the paper 'EEGPT: Pretrained Transformer for Universal and Reliable Representation of EEG Signals' [NIPS 2024].
解决的问题
EEGPT 解决了从 EEG(脑电图)信号中提取可靠、通用特征的挑战,这些信号通常面临低信噪比(SNR)问题,这会降低传统基于掩码的自监督学习方法提取特征的质量。
工作原理
EEGPT 是一个拥有 1,000 万参数的预训练 Transformer 模型。它采用基于掩码的双重自监督学习方法进行特征提取。与标准方法不同,它引入了时空表示对齐,以高信噪比和丰富语义信息的表示为基础构建自监督任务,而非依赖原始的嘈杂信号。
适用人群
从事脑机接口(BCI)和 EEG 信号处理的研究人员和开发者,他们需要一个稳健的预训练基础模型,用于各种下游神经或生理任务。
主要亮点
- 双重自监督学习:采用专门的掩码策略,提升特征提取效率。
- 时空对齐:专门设计以避免低信噪比信号导致的特征质量下降。
- 预训练权重:提供用于通用 EEG 特征提取的预训练模型。
- 下游任务通用性:支持线性探测和微调,适用于多种下游 EEG 数据集和任务。
相关
- 项目
- 项目
- 项目
- 项目