跳到正文
原文
arXiv cs.AI· Ioannis Kyprakis, Stefanos Gkikas, Chariklia Chatzaki, Zongyi Jiang, Randy Gomez, Raul Fernandez Rojas·· 8 天前AI 评分25

FOCUS:用于脑电认知负荷分类的轻量级 Mamba-2 模型

FOCUS: A Lightweight Mamba-2 Model for EEG-Based Cognitive Workload Classification

AI 导读

研究者提出基于 Mamba-2 的轻量级模型 FOCUS,参数量仅 0.04M,计算量为 0.11 GFLOPs,专用于脑电(EEG)认知负荷二分类。该模型结合固定 FFT 频段分解、深度可分离卷积与权重共享的 Mamba-2 结构。在 11 名受试者的跨任务留一验证中,FOCUS 取得 74.75% 准确率,GPU 前向延迟仅 0.08 ms。

来源:arXiv cs.AI · arxiv.org