LAtte: Hyperbolic Lorentz Attention for Cross-Subject EEG Classification
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Cross-subject Binary Classification | HighGamma MI EEG cross-subject | Accuracy0.4231 | 29 | |
| EEG Classification | SSVEP (test) | Accuracy73.75 | 17 | |
| EEG Classification | ERN (test) | AUC81.7 | 17 | |
| EEG Classification | MI (test) | Accuracy73.77 | 17 | |
| Cross-subject EEG classification | SSVEP (leave-one-subject-out) | Accuracy57.86 | 3 | |
| Cross-subject EEG classification | ERN (LOSO) | AUC75.31 | 3 |