English

CodeBrain: Bridging Decoupled Tokenizer and Multi-Scale Architecture for EEG Foundation Model

Machine Learning 2026-05-12 v4

Abstract

Electroencephalography (EEG) provides real-time insights into brain activity and supports diverse applications in neuroscience. While EEG foundation models (EFMs) have emerged to address the scalability issues of task-specific models, current approaches still yield clinically uninterpretable and weakly discriminative representations, inefficiently capturing global dependencies and neglecting important local neural events. We present CodeBrain, a two-stage EFM designed to fill this gap. In the first stage, we introduce the TFDual-Tokenizer, which decouples heterogeneous temporal and frequency EEG signals into discrete tokens, quadratically expanding the representation space to enhance discriminative power and offering domain-specific representation-level interpretability by suggesting potential links to neural events and spectral rhythms. In the second stage, we propose the multi-scale EEGSSM architecture, which combines structured global convolution with sliding window attention to efficiently capture both sparse long-range and local dependencies, reflecting the brain's small-world topology. Pretrained on the largest public EEG corpus, CodeBrain achieves strong generalization across eight downstream tasks and ten datasets under distribution shifts, supported by comprehensive ablations, scaling-law analyzes, and interpretability evaluations. The code and the pretrained weights are available at https://github.com/jingyingma01/CodeBrain.

Keywords

Cite

@article{arxiv.2506.09110,
  title  = {CodeBrain: Bridging Decoupled Tokenizer and Multi-Scale Architecture for EEG Foundation Model},
  author = {Jingying Ma and Feng Wu and Qika Lin and Yucheng Xing and Chenyu Liu and Ziyu Jia and Mengling Feng},
  journal= {arXiv preprint arXiv:2506.09110},
  year   = {2026}
}

Comments

Published as a conference paper at the International Conference on Learning Representations (ICLR 2026)

R2 v1 2026-07-01T03:09:42.274Z