arXiv 人工智能· Ioannis Kyprakis, Stefanos Gkikas, Chariklia Chatzaki, Zongyi Jiang, Randy Gomez, Raul Fernandez Rojas·· 9 小时前AI 评分31
FOCUS:用于脑电认知负荷分类的轻量级 Mamba-2 模型
FOCUS: A Lightweight Mamba-2 Model for EEG-Based Cognitive Workload Classification
AI 导读
研究人员提出基于 Mamba-2 的轻量级模型 FOCUS,用于在受限数据条件下评估脑电信号(EEG)的认知负荷二分类。该模型结合固定 FFT 频带分解、深度可分离卷积与权重共享结构,参数量仅 0.04 million,计算量为 0.11 GFLOPs。在 11 名受试者的留一交叉验证测试中,FOCUS 取得 74.75% 的准确率,GPU 前向推理延迟仅 0.08 ms。
来源:arXiv 人工智能 · arxiv.org