用双曲几何注意力提升脑电跨被试分类效果
LAtte: Hyperbolic Lorentz Attention for Cross-Subject EEG Classification
- 采用双曲空间中的洛伦兹注意力机制建模脑电信号
- 在5个数据集上实现跨被试分类性能超越现有方法
- 适合需要泛化到新被试的脑机接口研究者
脑电图(EEG)分类在医学诊断和脑机接口中至关重要,但受信噪比低和个体间差异大的挑战。现有方法多依赖被试特异性模型,难以捕捉神经信号共享结构且无法泛化至未见被试。为此,我们提出LAtte框架,结合洛伦兹注意力与双曲InceptionTime编码器,显式分解脑电信号为基线成分与任务相关偏差,实现更结构化的表征学习。为增强鲁棒性与适应性,在编码器与解码器层级引入被试特异性的低秩适配(LoRA)模块,并结合洛伦兹提升机制与双曲投影层,降低几何表示过拟合风险。我们在五个成熟脑电数据集上,于被试特异性、被试条件及留一被试出(LOSO)三种设置下评估模型表现,无论是否微调,均在小数据集上持续优于当前最优方法,大样本数据集上保持稳定性能。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。