跳到正文
OpenReview · EEG· Yiyu Gui; Mingzhi Chen; Guibo Luo; Yuchao Yang·· 2024-09-27精选AI 评分74

EEGMamba:结合混合专家的双向状态空间模型用于 EEG 多任务分类

EEGMamba: Bidirectional State Space Model with Mixture of Experts for EEG Multi-task Classification

AI 导读

基于摘要分析。该研究针对 EEG 分类中长序列建模的计算开销,以及不同任务之间信号长度、通道数不一致带来的适配问题,提出 EEGMamba,将 Spatio-Temporal-Adaptive(ST-Adaptive)模块、双向 Mamba 和 Mixture of Experts(MoE)整合为多任务分类框架。 机制上,ST-Adaptive 通过空间自适应卷积对不同长度与通道数的 EEG 进行统一特征提取,并引入 class token 实现时间维度适配;双向 Mamba 用于进一步提取 EEG 特征,旨在兼顾长序列处理的分类表现、推理速度与内存使用;任务感知 MoE 配置通用专家,以建模不同 EEG 任务之间的差异与共性。摘要未提供这些模块的具体结构、专家路由方式、损失函数或多任务训练流程,相关细节尚未验证。 作者报告在八个公开 EEG 数据集上开展评估,覆盖癫痫发作检测、情绪识别、睡眠分期分类和运动想象四类任务,并报告具有性能优势;但摘要未列出数据集名称、被试数、数据划分、对照基线或具体指标,因此尚不能判断优势对应被试内、跨被试、跨会话还是跨数据集协议,也不能将多任务学习直接等同于对未见任务的泛化。作者将其称为首个真正实现 EEG 多任务学习的通用分类网络,该优先性主张尚未检索确认。 复现时需核对不同任务是否联合训练、任务信息如何用于 MoE 路由、通道与长度适配的实现,以及各模块的消融结果;推理速度和内存优势还需结合硬件、输入长度及比较基线核验。实验协议、消融及统计信息尚未验证。材料仅表示代码计划近期发布,不能据此确认代码或预训练权重已公开。

阅读价值

值得阅读的是 EEGMamba 将长度与通道适配、双向状态空间建模及任务感知专家机制组合用于 EEG 多任务分类,但其多任务训练协议、性能优势和计算效率仍需全文核对。

来源:OpenReview · EEG · openreview.net