English

LAtte: Hyperbolic Lorentz Attention for Cross-Subject EEG Classification

Machine Learning 2026-05-18 v2

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.

Keywords

Cite

@article{arxiv.2603.10881,
  title  = {LAtte: Hyperbolic Lorentz Attention for Cross-Subject EEG Classification},
  author = {Ahmad Bdeir and Johannes Burchert and Tom Hanika and Lars Schmidt-Thieme and Niels Landwehr},
  journal= {arXiv preprint arXiv:2603.10881},
  year   = {2026}
}
R2 v1 2026-07-01T11:14:50.609Z