首页 > AI前沿 > SpecBraM: What Should an EEG Foundation Model Predict? Masked Band-Power Prediction versus Waveform Reconstruction

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

arXiv机器学习 2026-10-06 06:49 4 阅读 查看原文

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.

Experiments

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.

Results on ISRUC and HMC Sleep Staging

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.

Limitations

The gains do not extend to every task with spectral cues, including motor imagery, depression screening, and vigilance regression.

Conclusion

These results support choosing pretraining targets to match the physical quantities and spatial and temporal scales relevant to downstream labels.