MEG-Mamba: A Scalable State-Space Foundation Model for Magnetoencephalography

ABSTRACT

中文摘要

磁共振脑磁图能够以非侵入方式和毫秒级分辨率观察人脑活动。随着 MEG 数据日益丰富，自监督基础模型开始成为可行方向；但现有 MEG 及 EEG 基础模型大多基于 Transformer。作者提出 MEG-Mamba：一个建立在 Mamba 状态空间架构上的神经活动生成基础模型。模型以源重建并按脑区分区的 MEG 为输入，将连续信号离散化为 token，在给定脑区和记录 session 的条件下预测下一个 token。与基于 Transformer 的 MEG-GPT 相比，MEG-Mamba 生成信号的逼真度更高，同时预训练计算从 400 GPU 小时降至 22 GPU 小时，可建模的上下文也从 0.32 秒延长到 4 秒。作者通过生成数据的空间—频谱特性和学习到的嵌入分析模型，并进一步使用轻量 LoRA 刺激条件，使模型能够生成预训练阶段未包含的、逼真的任务诱发反应。结果支持 Mamba 作为扩展神经基础模型的高效主干。