LAtte: Hyperbolic Lorentz Attention for Cross-Subject EEG Classification
Organizations: Data Science Group, Universität Hildesheim · ISMLL, Universität Hildesheim
Abstract
Electroencephalogram (EEG) classification plays a key role in medical diagnosis and brain-computer interfaces, but remains challenging due to low signal-to-noise ratios and high inter-subject variability. As a result, many existing approaches rely on subject-specific models, which fail to exploit shared structure in neural signals and do not generalize to unseen subjects. To address these limitations, we propose LAtte, a framework that combines Lorentz attention with a hyperbolic InceptionTime-based encoder to improve cross-subject generalization in EEG classification. The model explicitly decomposes EEG signals into a learned baseline component and task-relevant deviations, enabling more structured representation learning. To further improve robustness and adaptability, we incorporate subject-specific low-rank adaptation (LoRA) modules at both encoder and decoder levels, augmented with a Lorentz boost-based LoRA mechanism and hyperbolic projection layers to reduce overfitting in geometric representations. We evaluate LAtte with and without finetuning in three settings: subject-specific, subject-conditional, and leave-one-subject-out (LOSO) on five established EEG datasets, achieving a consistent improvement in performance over current state-of-the-art methods for smaller datasets and maintaining performance for larger datasets.
Figures & tables
| Models | BCIC-IV 2a | MAMEM II | BCI Challenge | |
|---|---|---|---|---|
| Subject-Specific | ShallowConvNet | 61.84 6.39 | 56.93 6.97 | 71.86 2.64 |
| EEGNet | 57.43 6.25 | 53.72 7.23 | 74.28 2.47 | |
| SCCNet | 71.95 5.05 | 62.11 7.70 | 70.93 2.31 | |
| EEG-TCNet | 67.09 4.66 | 55.45 7.66 | 77.05 2.46 | |
| TCNet-Fusion | 56.52 3.07 | 45.00 6.45 | 70.46 2.94 | |
| FBCNet | 71.45 4.45 | 53.09 5.67 | 60.47 3.06 |
| Models | BCICIV-2b | HGD |
|---|---|---|
| EEGNet | 83.65 0.46 | 85.59 0.45 |
| ShallowConvNet | 81.45 0.50 | 89.75 0.54 |
| BaseNet | 86.11 0.40 | 93.64 0.86 |
| EEG-TCNet | 86.74 0.12 | 91.83 1.8 |
| TS-SEFFNet | 84.18 0.42 | 92.45 0.85 |
| CTNet | 86.91 0.29 | 94.21 0.52 |
| Models | BCICIV-2a | BCICIV-2b | HGD |
|---|---|---|---|
| EEGNet | 52.03 0.88 | 77.89 0.73 | 57.95 0.51 |
| ShallowConvNet | 47.31 1.06 | 75.58 1.01 | 72.47 0.94 |
| BaseNet | 56.89 0.88 | 78.61 0.57 | 68.55 1.74 |
| EEG-TCNet | 55.99 0.84 | 80.56 0.20 | 60.59 1.98 |
| TS-SEFFNet | 56.74 0.83 | 77.82 0.78 | 69.99 0.57 |
| CTNet | 60.09 0.94 | 80.29 0.42 | 64.6 0.9 |
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Task | Classes | Trials used |
|---|---|---|---|
| MAMEM | SSVEP | 5 | 3,300 |
| BCICha | Error monitoring | 2 | 2,880 |
| BCIC IV 2a | Motor imagery | 4 | 2,592 |
| BCIC IV 2b | Motor imagery | 2 | 3,680 |
| HGD | Motor imagery | 4 | 5,000 |
| Dataset | Baseline | (Real) | (Baseline) | |||
|---|---|---|---|---|---|---|
| MAMEM | 1488 | Shuffle | 0.0038 | 0.0116 | 0.0078 | 11.1 |
| Gaussian | 0.0042 | 0.0222 | 0.0180 | 352.5 | ||
| BCICha | 96 | Shuffle | 0.0238 | 0.0388 | 0.0150 | 26.5 |
| Gaussian | 0.0239 | 0.0565 | 0.0326 | 102.5 | ||
| BCIC IV 2a | 816 | Shuffle | 0.0167 | 0.0193 | 0.0026 | 8.9 |
| Gaussian | 0.0165 | 0.0281 | 0.0116 | 79.5 |
| Dataset | Raw | Encoder | Geometric change | ||
|---|---|---|---|---|---|
| MAMEM | 1,000 | 0.0097 | 1,488 | 0.0078 | Remains hyperbolic |
| BCICha | 8,960 | 0.0112 | 96 | 0.0150 | Euclidean Hyperbolic |
| BCIC IV 2a | 22,000 | 0.0078 | 816 | 0.0026 | Euclidean Hyperbolic |
| BCIC IV 2b | 3,375 | 0.0031 | 960 | 0.0065 | Strengthened |
| HGD | 44,000 | 0.0003 | 816 | 0.0093 | Strengthened |
| Models | MAMEM II | BCI Challenge |
|---|---|---|
| EEGFormer | 29.55 5.40 | 73.30 3.55 |
| ATCNet | 50.31 3.82 | 73.58 4.32 |
| CBraMod | 43.58 2.63 | 66.70 299 |
| TCFormer | 42.18 3.91 | 74.28 4.10 |
| LAtte | 57.86 3.66 | 75.31 3.17 |
| Relative | 15.00% | 1.38% |
| Subject | InceptionTime | MAtt | HyperMAtt | LAtte |
|---|---|---|---|---|
| 1 | ||||
| 2 | ||||
| 3 | ||||
| 4 | ||||
| 5 | ||||
| 6 |
| Models | BCIC IV 2a | MAMEM II | BCI Challenge | |
|---|---|---|---|---|
| CBraMod SS | 45.22 1.26 | 40.45 2.12 | 68.49 1.17 | |
| CBraMod SA | 39.14 1.55 | 49.91 0.00 | 72.90 0.46 | |
| CBraMod + | IncShallow | 40.93 2.41 | 45.47 1.63 | 70.77 0.14 |
| CBraMod❄ + | IncShallow | 30.01 0.06 | 24.00 0.00 | 53.52 0.01 |
| CBraMod + | IncDeep | 41.71 2.23 | 48.09 0.42 | 70.03 0.10 |
| CBraMod❄ + | IncDeep | 32.83 0.00 | 26.87 1.57 | 57.39 9.32 |
| Model Variant | Accuracy | |
|---|---|---|
| LAtte Full | 71.53 1.00 | - |
| LAtte w/o Lorentz | 57.40 6.53 | -19.75% |
| LAtte w/o LoRA | 62.87 0.51 | -12.11% |
| LAtte w/o Proj | 69.34 1.37 | -3.06% |
| LAtte w/o Cut/Fill | 69.55 0.71 | -2.77% |
| Model Variant | BCIC-IV 2a | BCIC-IV 2b | ||
|---|---|---|---|---|
| LAtte Full | 78.85 2.55 | - | 88.87 0.17 | - |
| LAtte w/o Baseline Block | 72.90 1.67 | -7.54% | 84.50 0.47 | -4.9% |
| LAtte w/o Hyperbolic MaxPool | 77.01 1.34 | -2.33% | 87.39 0.48 | -1.66% |
| LAtte w/o LB-LoRA | 77.21 2.13 | -2.1% | 86.29 0.30 | -2.9% |
| Model Variant | Accuracy |
|---|---|
| LAtte + Identity LoRA | 44.22 4.56 |
| LAtte + Embedding Distance | 57.32 1.51 |
| LAtte + SPD Covariance Metric | 56.69 2.22 |
| LAtte + Slot-0 generalization | 65.32 0.71 |
Cross-Subject EEG Decoding
Cross-subject EEG decoding develops methods that classify or predict brain activity for people not seen during training.
See all 77 papers