arXiv:2601.03322cs.LGcs.AI2026-01中稿 · ICLR被引 1

用双曲空间建模脑电数据层次结构,提升跨被试泛化能力

HEEGNet: Hyperbolic Embeddings for EEG

  • 融合欧式与双曲编码器,分阶段适应不同受试者数据分布
  • 在多个公开脑电数据集上达到当前最优性能,跨被试准确率提升显著
  • 适合需要跨被试鲁棒性的人机交互、神经解码研究者

基于脑电图(EEG)的脑机接口可实现人与计算机的直接通信,具有广阔应用前景。但其实际应用受限于跨被试(如受试者间)分布偏移导致的解码泛化能力差。学习能捕捉任务相关本质信息的鲁棒表征可缓解此问题。已有研究表明,视觉等认知过程具有层次性,可能编码于EEG信号中。尽管多数方法仍使用欧氏嵌入,近年开始探索双曲几何在EEG中的应用。双曲空间作为树状结构的连续类比,天然适合表示层次数据。本研究首次实证表明EEG数据具有双曲性,并证明双曲嵌入能提升泛化性能。据此提出HEEGNet——一种混合双曲网络架构,用于捕捉EEG的层次结构并学习域不变的双曲嵌入。该模型结合欧氏与双曲编码器,并采用新颖的粗到细域自适应策略。在包含视觉诱发电位、情绪识别和颅内脑电的多个公开数据集上的大量实验表明,HEEGNet性能达到当前最优水平。代码已开源。

原文摘要 · Abstract (English)

Electroencephalography (EEG)-based brain-computer interfaces facilitate direct communication with a computer, enabling promising applications in human-computer interactions. However, their utility is currently limited because EEG decoding often suffers from poor generalization due to distribution shifts across domains (e.g., subjects). Learning robust representations that capture underlying task-relevant information would mitigate these shifts and improve generalization. One promising approach is to exploit the underlying hierarchical structure in EEG, as recent studies suggest that hierarchical cognitive processes, such as visual processing, can be encoded in EEG. While many decoding methods still rely on Euclidean embeddings, recent work has begun exploring hyperbolic geometry for EEG. Hyperbolic spaces, regarded as the continuous analogue of tree structures, provide a natural geometry for representing hierarchical data. In this study, we first empirically demonstrate that EEG data exhibit hyperbolicity and show that hyperbolic embeddings improve generalization. Motivated by these findings, we propose HEEGNet, a hybrid hyperbolic network architecture to capture the hierarchical structure in EEG and learn domain-invariant hyperbolic embeddings. To this end, HEEGNet combines both Euclidean and hyperbolic encoders and employs a novel coarse-to-fine domain adaptation strategy. Extensive experiments on multiple public EEG datasets, covering visual evoked potentials, emotion recognition, and intracranial EEG, demonstrate that HEEGNet achieves state-of-the-art performance. The code is available at https://github.com/fightlesliefigt/HEEGNet

脑电分析双曲嵌入域泛化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。