CodeBrain: Bridging Decoupled Tokenizer and Multi-Scale Architecture for EEG Foundation Model
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 23 papers in the map (data through 7 Oct 2026). Builds on 50 of its 96 references that are in the map. First posted on arXiv 10 Jun 2025.
Builds on
50 of 96 references.
References · 50
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2023GPT-4 Technical Reportcited by 18,132
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2017Neural Discrete Representation Learningcited by 4,650
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
- 2018SentencePiece: A simple and language independent subword tokenizer and detokenizer for Neural Text Processingcited by 1,973
- 2019Root Mean Square Layer Normalizationcited by 1,425
38 more
- 2022Simplified State Space Layers for Sequence Modelingcited by 776
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 696
- 2016EEGNet: A Compact Convolutional Network for EEG-based Brain-Computer Interfacescited by 616
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 495
- 2022InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutionscited by 479
- 2023Finite Scalar Quantization: VQ-VAE Made Simplecited by 425
- 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 418
- 2024Large Brain Model for Learning Generic Representations with Tremendous EEG Data in BCIcited by 311
- 2021BENDR: using transformers and a contrastive self-supervised learning task to learn from massive amounts of EEG datacited by 194
- 2024CBraMod: A Criss-Cross Brain Foundation Model for EEG Decodingcited by 188
- 2020CLOCS: Contrastive Learning of Cardiac Signals Across Space, Time, and Patientscited by 170
- 2021Transformer-based Spatial-Temporal Feature Learning for EEG Decodingcited by 92
- 2020XSleepNet: Multi-View Sequential Model for Automatic Sleep Stagingcited by 87
- 2022What Makes Convolutional Models Great on Long Sequence Modeling?cited by 86
- 2024NeuroLM: A Universal Multi-task Foundation Model for Bridging the Gap between Language and EEG Signalscited by 83
- 2023Simple Hardware-Efficient Long Convolutions for Sequence Modelingcited by 62
- 2023Online Clustered Codebookcited by 55
- 2024EEG2Rep: Enhancing Self-supervised EEG Representation Through Informative Masked Inputscited by 48
- 2022Deep Latent State Space Models for Time-Series Generationcited by 37
- 2025Reading Your Heart: Learning ECG Words and Sentences via Pre-training ECG Language Modelcited by 36
- 2023BrainNet: Epileptic Wave Detection from SEEG with Hierarchical Graph Diffusion Learningcited by 25
- 2024EEG-Deformer: A Dense Convolutional Transformer for Brain-computer Interfacescited by 25
- 2025Brain Foundation Models: A Survey on Advancements in Neural Signal Processing and Brain Discoverycited by 22
- 2024Has Multimodal Learning Delivered Universal Intelligence in Healthcare? A Comprehensive Surveycited by 20
- 2024EEGMamba: Bidirectional State Space Model with Mixture of Experts for EEG Multi-task Classificationcited by 18
- 2024Brant-X: A Unified Physiological Signal Alignment Frameworkcited by 16
- 2025FEMBA: Efficient and Scalable EEG Analysis with a Bidirectional Mamba Foundation Modelcited by 15
- 2025Tokenizing Single-Channel EEG with Time-Frequency Motif Learningcited by 14
- 2024Exploring Large-Scale Language Models to Evaluate EEG-Based Multimodal Data for Mental Healthcited by 13
- 2024EEG-SSM: Leveraging State-Space Model for Dementia Detectioncited by 11
- 2025Self-supervised Quantized Representation for Seamlessly Integrating Knowledge Graphs with Large Language Modelscited by 11
- 2025Large Cognition Model: Towards Pretrained EEG Foundation Modelcited by 10
- 2024Vector Quantization for Recommender Systems: A Review and Outlookcited by 6
- 2024Long-Term EEG Partitioning for Seizure Onset Detectioncited by 6
- 2024ST-USleepNet: A Spatial-Temporal Coupling Prominence Network for Multi-Channel Sleep Stagingcited by 4
- 2025Development and Validation of a Dynamic Kidney Failure Prediction Model based on Deep Learning: A Real-World Study with External Validationcited by 3
- 2025An Efficient Self-Supervised Framework for Long-Sequence EEG Modelingcited by 2
Built on by
Citing papers per year: 2025: 5; 2026: 18.
23 papers.
Sort:
By area:
- 2025Uni-NTFM: A Unified Foundation Model for EEG Signal Representation Learningcited by 9
- 2026DeeperBrain: A Neuro-Grounded EEG Foundation Model Towards Universal BCIcited by 7
- 2025ECHO: Toward Contextual Seq2Seq Paradigms in Large EEG Modelscited by 6
- 2025DPsurv: Dual-Prototype Evidential Fusion for Uncertainty-Aware and Interpretable Whole-Slide Image Survival Predictioncited by 6
- 2026Brain4FMs: A Benchmark of Foundation Models for Electrical Brain Signalcited by 5
- 2025EEGDM: EEG Representation Learning via Generative Diffusion Modelcited by 4
- 2026NeuroWeaver: An Autonomous Evolutionary Agent for Exploring the Programmatic Space of EEG Analysis Pipelinescited by 4
- 2026BrainRVQ: A High-Fidelity EEG Foundation Model via Dual-Domain Residual Quantization and Hierarchical Autoregressioncited by 4
- 2025SingLEM: Single-Channel Large EEG Modelcited by 3
- 2026SCOPE: Structured Prototype-Guided Adaptation for EEG Foundation Models with Limited Labelscited by 1
- 2026LERD: Latent Event-Relational Dynamics for Neurodegenerative Classificationcited by 1
- 2026TRACE: Temporal Routing with Autoregressive Cross-channel Experts for EEG Representation Learningcited by 1
11 more
- 2026DLink: Distilling Layer-wise and Dominant Knowledge from EEG Foundation Modelscited by 0
- 2026Context-aware tokenization for Cross-subject Emotion Decoding from EEGcited by 0
- 2026EvoBrain: Continual Learning of EEG Foundation Models Across Heterogeneous BCI Taskscited by 0
- 2026Channel-Preserving Representation Alignment for EEG-to-Music Reconstructioncited by 0
- 2026SL-S4Wave: Self-Supervised Learning of Physiological Waveforms with Structured State Space Modelscited by 0
- 2026B[FM]2: Brain Foundation Model via Flow Matching with SplitUNetcited by 0
- 2026ZIPBrain: Can EEG Foundation Models Be Faster, Locally Deployable, but Accurate?cited by 0
- 2026Continuous-Latent Predictive Modeling with Semantic Alignment for EEG-Language Foundation Modelscited by 0
- 2026Taming foundation model with invariance-oriented pre-training for broad-spectrum EEG analysis across signal-level, brain-state, and brain-health taskscited by 0
- 2026Neural State Prediction: Obstructing Shortcut Learning in EEG Foundation Modelscited by 0
- 2026Representational and Functional Robustness to Electrode Montages in EEG Foundation Modelscited by 0
- 2026Representational and Functional Robustness to Electrode Montages in EEG Foundation Models28 Sep 2026
- 2026Neural State Prediction: Obstructing Shortcut Learning in EEG Foundation Models25 Sep 2026
- 2026Taming foundation model with invariance-oriented pre-training for broad-spectrum EEG analysis across signal-level, brain-state, and brain-health tasks25 Aug 2026
- 2026Continuous-Latent Predictive Modeling with Semantic Alignment for EEG-Language Foundation Models12 Aug 2026
- 2026ZIPBrain: Can EEG Foundation Models Be Faster, Locally Deployable, but Accurate?7 Aug 2026
- 2026B[FM]2: Brain Foundation Model via Flow Matching with SplitUNet18 Jun 2026
- 2026SL-S4Wave: Self-Supervised Learning of Physiological Waveforms with Structured State Space Models18 Jun 2026
- 2026Channel-Preserving Representation Alignment for EEG-to-Music Reconstruction2 Jun 2026
- 2026EvoBrain: Continual Learning of EEG Foundation Models Across Heterogeneous BCI Tasks1 Jun 2026
- 2026Context-aware tokenization for Cross-subject Emotion Decoding from EEG30 May 2026
- 2026TRACE: Temporal Routing with Autoregressive Cross-channel Experts for EEG Representation Learning12 May 2026
- 2026DLink: Distilling Layer-wise and Dominant Knowledge from EEG Foundation Models16 Apr 2026
11 more
- 2026LERD: Latent Event-Relational Dynamics for Neurodegenerative Classification20 Feb 2026
- 2026SCOPE: Structured Prototype-Guided Adaptation for EEG Foundation Models with Limited Labels19 Feb 2026
- 2026BrainRVQ: A High-Fidelity EEG Foundation Model via Dual-Domain Residual Quantization and Hierarchical Autoregression18 Feb 2026
- 2026NeuroWeaver: An Autonomous Evolutionary Agent for Exploring the Programmatic Space of EEG Analysis Pipelines13 Feb 2026
- 2026Brain4FMs: A Benchmark of Foundation Models for Electrical Brain Signal12 Feb 2026
- 2026DeeperBrain: A Neuro-Grounded EEG Foundation Model Towards Universal BCI5 Jan 2026
- 2025Uni-NTFM: A Unified Foundation Model for EEG Signal Representation Learning29 Sep 2025
- 2025DPsurv: Dual-Prototype Evidential Fusion for Uncertainty-Aware and Interpretable Whole-Slide Image Survival Prediction28 Sep 2025
- 2025ECHO: Toward Contextual Seq2Seq Paradigms in Large EEG Models26 Sep 2025
- 2025SingLEM: Single-Channel Large EEG Model22 Sep 2025
- 2025EEGDM: EEG Representation Learning via Generative Diffusion Model13 Aug 2025
- 2025Uni-NTFM: A Unified Foundation Model for EEG Signal Representation Learningcited by 9
- 2026DeeperBrain: A Neuro-Grounded EEG Foundation Model Towards Universal BCIcited by 7
- 2025ECHO: Toward Contextual Seq2Seq Paradigms in Large EEG Modelscited by 6
- 2025DPsurv: Dual-Prototype Evidential Fusion for Uncertainty-Aware and Interpretable Whole-Slide Image Survival Predictioncited by 6
- 2026Brain4FMs: A Benchmark of Foundation Models for Electrical Brain Signalcited by 5
- 2026BrainRVQ: A High-Fidelity EEG Foundation Model via Dual-Domain Residual Quantization and Hierarchical Autoregressioncited by 4
- 2025SingLEM: Single-Channel Large EEG Modelcited by 3
- 2026SCOPE: Structured Prototype-Guided Adaptation for EEG Foundation Models with Limited Labelscited by 1
- 2026LERD: Latent Event-Relational Dynamics for Neurodegenerative Classificationcited by 1
- 2026TRACE: Temporal Routing with Autoregressive Cross-channel Experts for EEG Representation Learningcited by 1
- 2026DLink: Distilling Layer-wise and Dominant Knowledge from EEG Foundation Modelscited by 0
- 2026Context-aware tokenization for Cross-subject Emotion Decoding from EEGcited by 0
12 most cited of 19 All 19 (JSON)
- 2026Representational and Functional Robustness to Electrode Montages in EEG Foundation Models28 Sep 2026
- 2026Neural State Prediction: Obstructing Shortcut Learning in EEG Foundation Models25 Sep 2026
- 2026Taming foundation model with invariance-oriented pre-training for broad-spectrum EEG analysis across signal-level, brain-state, and brain-health tasks25 Aug 2026
- 2026Continuous-Latent Predictive Modeling with Semantic Alignment for EEG-Language Foundation Models12 Aug 2026
- 2026ZIPBrain: Can EEG Foundation Models Be Faster, Locally Deployable, but Accurate?7 Aug 2026
- 2026Channel-Preserving Representation Alignment for EEG-to-Music Reconstruction2 Jun 2026
- 2026EvoBrain: Continual Learning of EEG Foundation Models Across Heterogeneous BCI Tasks1 Jun 2026
- 2026Context-aware tokenization for Cross-subject Emotion Decoding from EEG30 May 2026
- 2026TRACE: Temporal Routing with Autoregressive Cross-channel Experts for EEG Representation Learning12 May 2026
- 2026DLink: Distilling Layer-wise and Dominant Knowledge from EEG Foundation Models16 Apr 2026
- 2026LERD: Latent Event-Relational Dynamics for Neurodegenerative Classification20 Feb 2026
- 2026SCOPE: Structured Prototype-Guided Adaptation for EEG Foundation Models with Limited Labels19 Feb 2026
12 newest of 19 All 19 (JSON)