用空间补全思想解决脑电跨域数据分布偏移问题
Spatial Imputation Drives Cross-Domain Alignment for EEG Classification
- 将跨域脑电信号对齐转化为通道依赖的时空补全任务
- 在10个公开数据集上实现跨被试与跨中心最优分类精度
- 适合处理电极布局不一、设备差异大的真实脑电场景
脑电(EEG)信号分类受不同电极配置、采集协议和硬件差异导致的数据分布偏移严重影响。本文提出IMAC框架,将跨域脑电数据对齐建模为时空序列补全问题。通过3D到2D的位置统一映射策略,标准化不同电极布局以建立统一空间表示。不同于传统掩码自监督方法,IMAC引入时空信号对齐机制,构建通道依赖的掩码与重建任务,将其视为从低分辨率到高分辨率的空间补全问题,从而模拟通道缺失与时间不稳定性,使模型在推理时具备鲁棒对齐能力。此外,采用解耦结构分别建模时间与空间信息,降低计算复杂度并提升灵活性。在10个公开脑电数据集上的全面评估表明,IMAC在跨被试与跨中心验证中均达到最先进的分类准确率。尤其在模拟与真实分布偏移下表现出强鲁棒性,完整性评分较基线最高提升35%,且分类性能稳定。
原文摘要 · Abstract (English)
Electroencephalogram (EEG) signal classification faces significant challenges due to data distribution shifts caused by heterogeneous electrode configurations, acquisition protocols, and hardware discrepancies across domains. This paper introduces IMAC, a novel channel-dependent mask and imputation self-supervised framework that formulates the alignment of cross-domain EEG data shifts as a spatial time series imputation task. To address heterogeneous electrode configurations in cross-domain scenarios, IMAC first standardizes different electrode layouts using a 3D-to-2D positional unification mapping strategy, establishing unified spatial representations. Unlike previous mask-based self-supervised representation learning methods, IMAC introduces spatio-temporal signal alignment. This involves constructing a channel-dependent mask and reconstruction task framed as a low-to-high resolution EEG spatial imputation problem. Consequently, this approach simulates cross-domain variations such as channel omissions and temporal instabilities, thus enabling the model to leverage the proposed imputer for robust signal alignment during inference. Furthermore, IMAC incorporates a disentangled structure that separately models the temporal and spatial information of the EEG signals separately, reducing computational complexity while enhancing flexibility and adaptability. Comprehensive evaluations across 10 publicly available EEG datasets demonstrate IMAC's superior performance, achieving state-of-the-art classification accuracy in both cross-subject and cross-center validation scenarios. Notably, IMAC shows strong robustness under both simulated and real-world distribution shifts, surpassing baseline methods by up to $35$\% in integrity scores while maintaining consistent classification accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。