HyBRA-EEG: Hyperbolic Global-Local Brain-Region Aggregation for Self-Supervised EEG Representation Learning
Abstract
Electroencephalography (EEG) representation learning is challenged by noisy signals, inter-subject variability, and heterogeneous electrode montages in different datasets and recording settings. Recent self-supervised EEG pretraining methods improve transferability by converting heterogeneous recordings into unified token or sequence representations, but channel-aware designs may still encode montage- or dataset-specific regularities. We propose HyBRA, a self-supervised framework based on hyperbolic global-local brain-region aggregation. HyBRA maps heterogeneous channel observations into a fixed regional token space by combining electrode coordinates, region priors, and feature-dependent hyperbolic distances. A channel-evidence injection branch preserves fine-grained discriminative cues during aggregation. The resulting tokens are processed by a Region-Time Transformer and pretrained with masked reconstruction of latent features and region-level signals. HyBRA outperforms the compared methods in average balanced accuracy on six benchmark datasets. Under a controlled six-fold evaluation on OpenBMI-MI, HyBRA matches CSBrain in balanced accuracy, improves AUROC by 1.84 percentage points and AUPRC by 5.23 percentage points, and uses 96.8% fewer parameters. Matched comparisons on TUAB and TUEV show higher balanced accuracy for hyperbolic matching than for Euclidean distance or cosine similarity.
Figures & tables
| Phase | Dataset | Type | # Ch. | # Samp. | Dur. |
|---|---|---|---|---|---|
| Pre-train | BETA (2s) | SSVEP | 60 | 2,400 | 2s |
| BETA (3s) | SSVEP | 60 | 8,800 | 3s | |
| Tsinghua SSVEP | SSVEP | 60 | 8,400 | 5s | |
| Cho2017 | MI | 64 | 10,520 | 3s | |
| PhysioNet-MI | MI | 64 | 9,270 | 4s | |
| FACED | EMO | 30 | 33,880 | 3s |
| Method | TUAB | TUEV | ||||
|---|---|---|---|---|---|---|
| Balanced Acc. | AUPRC | AUROC | Balanced Acc. | Cohen’s Kappa | Weighted-F1 | |
| EEGNet | 0.7642 0.0036 | 0.8299 0.0043 | 0.8412 0.0031 | 0.3876 0.0143 | 0.3577 0.0155 | 0.6539 0.0120 |
| EEGConformer | 0.7758 0.0049 | 0.8427 0.0054 | 0.8445 0.0038 | 0.4074 0.0164 | 0.3967 0.0195 | 0.6983 0.0152 |
| SPaRCNet | 0.7896 0.0018 | 0.8414 0.0018 | 0.8676 0.0012 | 0.4161 0.0262 | 0.4233 0.0181 | 0.7024 0.0104 |
| ContraWR | 0.7746 0.0041 | 0.8421 0.0104 | 0.8456 0.0074 | 0.4384 0.0349 | 0.3912 0.0237 | 0.6893 0.0136 |
| CNN-Transformer | 0.7777 0.0022 | 0.8433 0.0039 | 0.8461 0.0013 | 0.4087 0.0161 | 0.3815 0.0134 | 0.6854 0.0293 |
| OpenBMI-MI | MentalArithmetic | |||
|---|---|---|---|---|
| Method | BAcc | Params | BAcc | Params |
| HyBRA | 81.17 4.36 | 1.45 (1.0 ) | 76.63 5.01 | 1.45 (1.0 ) |
| CSBrain | 81.04 3.51 | 44.86 (30.9 ) | 65.38 9.64 | 28.06 (19.3 ) |
| LaBraM | 70.13 2.84 | 5.82 (4.0 ) | 69.64 5.08 | 5.82 (4.0 ) |
| BIOT | 69.08 3.18 | 3.19 (2.2 ) | 64.16 7.03 | 3.19 (2.2 ) |
| EEGPT | 68.10 3.54 | 25.32 (17.4 ) | 55.12 3.68 | 25.32 (17.4 ) |
| Number of brain regions | ||||
|---|---|---|---|---|
| Dataset | Metric | 5 | 10 | 11 |
| TUAB | BAcc (%) | 81.82 0.57 | 81.17 0.74 | 82.24 0.21 |
| Kappa (%) | 63.59 1.17 | 62.34 1.51 | 64.38 0.45 | |
| OpenBMI-MI | BAcc (%) | 72.25 4.82 | 80.90 4.32 | 81.17 4.36 |
| Kappa (%) | 44.51 9.64 | 61.81 8.64 | 62.33 8.71 | |
| Module | Setup | TUAB | TUEV |
|---|---|---|---|
| Coordinate-Aware Patch Encoding | w/o coordinate encoding | 0.8057 0.0051 | 0.6555 0.0340 |
| Hyperbolic Region Mixer | Euclidean distance-based matching | 0.8134 0.0078 | 0.6509 0.0214 |
| Cosine similarity-based matching | 0.8164 0.0035 | 0.5989 0.0206 | |
| w/o brain-region tokenization | 0.7980 0.0073 | 0.6166 0.0175 | |
| w/o channel evidence injection | 0.8123 0.0049 | 0.6468 0.0428 | |
| Region-Time Transformer | region-only attention | 0.8137 0.0028 | 0.6814 0.0089 |
Electroencephalography
Electroencephalography (EEG) studies brain-generated electrical activity to understand cognition and support diagnosis, monitoring, and brain-computer interfaces.
See all 183 papers