Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

BrainPro: Towards Large-scale Brain State-aware EEG Representation Learning

About

Electroencephalography (EEG) reflects underlying brain states, whose activities are distributed across brain regions and manifest as spatial patterns on the scalp. Learning these spatially structured, state-related patterns requires consistent spatial representations across datasets. However, existing EEG foundation models are typically based on self-attention, which does not preserve location-specific information and struggles to align signals recorded with different channel configurations. Moreover, brain states contain both shared and state-specific regional activity, suggesting that learning neurophysiologically plausible, state-aware representations can complement the shared representations targeted by current models and improve downstream decoding. To address these limitations, we propose BrainPro, a large EEG model that combines a retrieval-based spatial learning mechanism for cross-layout spatial alignment with a brain state-decoupling module that learns both shared and state-specific representations through parallel encoders and region-aware reconstruction. Pre-trained on a large EEG corpus, BrainPro achieves state-of-the-art performance across nine public BCI datasets spanning emotion, motor, speech, stress, mental disease, and attention tasks. Analyses of spatial filters, channel-drop robustness, and encoder contributions further validate the effectiveness of its spatial alignment and state-aware pathways. These results show that BrainPro achieves improved interpretability of learned spatial patterns and produces representations that benefit diverse EEG decoding tasks.

Yi Ding, Muyun Jiang, Weibang Jiang, Shuailei Zhang, Xinliang Zhou, Chenyu Liu, Shanglin Li, Yong Li, Cuntai Guan• 2025

Related benchmarks

TaskDatasetResultRank
Emotion RecognitionSEED VII
Balanced Accuracy0.3315
28
EEG ClassificationBCIC-IV-2a
Balanced Accuracy56.74
18
EEG ClassificationFACED 9-Class
Balanced Accuracy59.37
13
EEG ClassificationSEED-V 5-Class
Balanced Accuracy (ACC-B)40.78
13
EEG ClassificationSHU-MI 2-Class 11,988 Samples
Balanced Accuracy63.19
12
EEG ClassificationMental Arithmetic 2-Class
Balanced Accuracy80.83
7
EEG ClassificationAttention 2-Class
Accuracy (B)72.22
7
Mental Disorder DiagnosisMajor Depressive Disorder (MDD)
ACC-B91.61
7
Speech ClassificationBCIC2020-3 Speech dataset (test)
ACC-B52.53
7
Showing 9 of 9 rows

Other info

Follow for update