Multimodal Learning 相关度: 7/10

SpecMoE: Spectral Mixture-of-Experts Foundation Model for Cross-Species EEG Decoding

D. Darankoum, C. Habermacher, J. Volle, S. Grudinin
arXiv: 2603.16739v1 发布: 2026-03-17 更新: 2026-03-17

AI 摘要

SpecMoE模型利用频谱信息进行跨物种脑电解码,性能优于现有方法。

主要贡献

  • 提出了基于STFT图和高斯平滑掩码的预训练方法
  • 设计了SpecHi-Net模型,用于高效信号重构
  • 提出了SpecMoE模型,利用频谱门控机制融合多个专家模型

方法论

采用自监督预训练,通过高斯掩码STFT图,训练SpecHi-Net重构信号,再用SpecMoE融合专家模型。

原文摘要

Decoding the orchestration of neural activity in electroencephalography (EEG) signals is a central challenge in bridging neuroscience with artificial intelligence. Foundation models have made strides in generalized EEG decoding, yet many existing frameworks primarily relying on separate temporal and spectral masking of raw signals during self-supervised pretraining. Such strategies often tend to bias learning toward high-frequency oscillations, as low-frequency rhythmic patterns can be easily inferred from the unmasked signal. We introduce a foundation model that utilizes a novel Gaussian-smoothed masking scheme applied to short-time Fourier transform (STFT) maps. By jointly applying time, frequency, and time-frequency Gaussian masks, we make the reconstruction task much more challenging, forcing the model to learn intricate neural patterns across both high- and low-frequency domains. To effectively recover signals under this aggressive masking strategy, we design SpecHi-Net, a U-shaped hierarchical architecture with multiple encoding and decoding stages. To accelerate large-scale pretraining, we partition the data into three subsets, each used to train an independent expert model. We then combine these models through SpecMoE, a mixture of experts framework guided by a learned spectral gating mechanism. SpecMoE achieves state-of-the-art performance across a diverse set of EEG decoding tasks, including sleep staging, emotion recognition, motor imagery classification, abnormal signal detection, and drug effect prediction. Importantly, the model demonstrates strong cross-species and cross-subject generalization, maintaining high accuracy on both human and murine EEG datasets.

标签

脑电信号 深度学习 自监督学习 混合专家模型 跨物种

arXiv 分类

cs.LG cs.AI cs.HC