跳到正文
arXiv:cs.LG· Peng Xie, Yequan Bie, Jianda Mao, Kani Chen·· 4 小时前AI 评分35

SpecBraM:EEG 基础模型应该预测什么?掩码频带功率预测与波形重建对比

SpecBraM: What Should an EEG Foundation Model Predict? Masked Band-Power Prediction versus Waveform Reconstruction

AI 导读

研究提出掩码频带功率预测(MBP),对掩码通道-时间片段预测固定窄带对数谱能量,替代相位敏感的波形重建和学习码本。

正文

View PDF HTML (experimental)

Abstract:Self-supervised EEG models often reconstruct masked waveforms or predict discrete codes. We study a task-aligned alternative: masked band-power prediction (MBP), which predicts fixed narrow-band log spectral energy for masked channel-time patches. This target retains rhythm power relevant to sleep staging while avoiding phase-sensitive waveform reconstruction and a learned codebook. Across three pretraining seeds, we compare band-power and waveform targets with matched backbones, pretraining data (2,388 hours), and training steps, including a 2x2 tokenizer-by-target design. On ISRUC and HMC sleep staging, MBP exceeds raw- and band-waveform reconstruction by 1.6-2.8 balanced-accuracy points with all labels and 4.7-7.3 points with 1% of labels under a strict linear probe; the target effect exceeds the tokenizer effect. Its frozen features reach 0.7916/0.7425 balanced accuracy, versus 0.7636/0.7227 for a matched rich handcrafted spectral baseline, although the gap is about one point with 1% of labels. Full fine-tuning reaches 0.8107/0.7669. The gains do not extend to every task with spectral cues, including motor imagery, depression screening, and vigilance regression. These results support choosing pretraining targets to match the physical quantities and spatial and temporal scales relevant to downstream labels.
Comments: 12 pages, 3 figures, 9 tables; includes an appendix
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:2610.07484 [cs.LG]
  (or arXiv:2610.07484v1 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2610.07484

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Peng Xie [view email]
[v1] Mon, 5 Oct 2026 22:49:16 UTC (399 KB)

来源:arXiv:cs.LG · arxiv.org