EEG-Conformer 已入驻 Braindecode:一行代码调用顶级脑电解码模型

EEG-Conformer 已入驻 Braindecode:一行代码调用顶级脑电解码模型 EEG-Conformer 已入驻 Braindecode一行代码调用顶级脑电解码模型【免费下载链接】EEG-Conformer[TNSRE 23] EEG Transformer 2.0. i. Convolutional Transformer for EEG Decoding. ii. Novel visualization - Class Activation Topography.项目地址: https://gitcode.com/gh_mirrors/ee/EEG-Conformer脑电解码EEG Decoding是脑机接口BCI领域的核心任务而 EEG-Conformer 正是近年来这一方向上表现最亮眼的深度学习模型之一。这篇发表于 IEEE TNSRE 2023 的卷积 Transformer 模型如今已正式入驻主流脑电深度学习工具箱 Braindecode——这意味着你只需一行代码就能调用这个顶级脑电解码模型快速完成运动想象、情绪识别等 EEG 分类任务。EEG-Conformer 是什么卷积 Transformer 的脑电解码新范式传统脑电解码模型要么依赖纯卷积网络捕捉局部特征要么堆叠复杂的 Transformer 结构建模全局依赖而 EEG-Conformer 的设计思路非常巧妙它把两者拧在一起形成一套紧凑而强大的统一框架。整个模型的流程可以分为三步卷积模块提取局部特征先用一维时间卷积学习时序上的局部模式再用空间卷积融合多电极通道的信息最后通过平均池化把特征切分成类似 ViT 的Token序列自注意力模块捕捉全局关联将卷积输出的 Token 直接送入 Transformer 自注意力层建模脑电信号中长距离的全局依赖关系全连接分类器输出结果最终由基于全连接层的轻量分类器完成类别预测。上图正是论文中的网络架构图从左侧的原始 EEG 输入channel × sample经过时间卷积、空间卷积与池化生成 Token再进入中间的 N 层自注意力模块最终由右侧的全连接分类器输出类别结果。整个流程一目了然模型核心代码集中在 conformer.py 中。顶级成绩三大脑电数据集上的准确率一览模型好不好数据说了算。EEG-Conformer 在三个公开的经典脑电数据集上都拿到了极具竞争力的结果数据集任务准确率BCI Competition IV 2a四分类运动想象78.66%BCI Competition IV 2b二分类运动想象84.63%SEED情绪识别95.30%其中 2a/2b 数据集采用留出法hold out划分SEED 采用 5 折交叉验证划分方式在 preprocessing/ 目录的 MATLAB 脚本BCIIV2a.m、BCIIV2b.m、seed.m中保持一致方便与其他方法公平对比。一键上手一行代码在 Braindecode 中调用 EEG-Conformer对于大多数普通用户来说最大的好消息是EEG-Conformer 已经正式加入 Braindecode 工具箱。你不再需要手动搭建网络、处理繁琐的数据加载逻辑只需像使用普通 PyTorch 模块一样一行代码即可完成模型创建from braindecode.models import EEGConformer model EEGConformer( n_chans22, # EEG 电极通道数 n_times1000, # 单个 trial 的时间采样点数 n_outputs4, # 分类类别数 )之后就可以直接把model丢进 Braindecode 的训练流程中配合它自带的加载器与训练器使用。无论是快速验证想法还是复现论文结果都非常方便。✨进阶玩法Class Activation Topography 脑地形可视化除了漂亮的分类准确率EEG-Conformer 还有一个很酷的创新——Class Activation TopographyCAT。它借鉴了计算机视觉中的类激活映射CAM思想把模型注意力最集中的区域投影回脑电电极分布图上形成一张直观的脑地形图。这样一来研究者不仅能知道模型分对了没有还能直接看到模型在头皮哪个位置、哪个时间段更关注为脑机接口研究提供了全新的可解释性视角。相关可视化代码全部在 visualization/ 目录下CAT.py类激活地形图CAT主程序topography.py脑地形图绘制工具tSNE.py特征分布可视化。本地复现完整实验的快速指南如果你想从零开始复现论文中的完整实验流程也非常清晰git clone https://gitcode.com/gh_mirrors/ee/EEG-Conformer克隆完成后按照下面的顺序操作即可准备环境项目基于 Python 3.10 与 PyTorch 1.12安装依赖后即可运行预处理数据使用 preprocessing/ 下的 MATLAB 脚本把原始 EEG 数据转成模型需要的格式训练模型不同数据集的训练入口已在仓库中分别提供——BCI 2b 对应 conformer_BCIIV2b.pySEED 情绪数据对应 conformer_seed_1s_5fold.py通用模型定义则在 conformer.py。写在最后EEG-Conformer 用简洁的架构同时收获了强大的性能与优雅的可解释性如今又借由 Braindecode 生态让每一位普通开发者都能轻松上手。如果你正在做脑电解码、脑机接口或情绪识别方向的工作不妨立刻用一行代码把它领回家试试看【免费下载链接】EEG-Conformer[TNSRE 23] EEG Transformer 2.0. i. Convolutional Transformer for EEG Decoding. ii. Novel visualization - Class Activation Topography.项目地址: https://gitcode.com/gh_mirrors/ee/EEG-Conformer创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考