融合时频域的EEG表示学习,提升认知负荷分类准确率
Multi-Domain EEG Representation Learning with Orthogonal Mapping and Attention-based Fusion for Cognitive Load Classification
- 通过时频双路编码获取脑电信号表征
- 多域注意力机制增强跨域关系建模,准确率达89.3%
- 抗噪能力强,适合真实场景下的脑机接口应用
我们提出一种基于脑电图(EEG)的认知负荷分类新表示学习方法。首先将原始EEG信号输入卷积编码器,获得时域表征;接着计算五种频率带的功率谱密度(PSD),生成通道功率值的二维图像——多光谱拓扑图,并通过独立编码器提取频域表征。采用多域注意力模块将这些领域特定嵌入映射到共享嵌入空间,强化重要跨域关系以增强表征。此外,在训练中引入正交投影约束,有效增大类间距离并改善类内聚类。在两个公开数据集CL-Drive和CLARE上进行大量实验验证,结果表明该方法优于传统单域技术。消融实验与敏感性分析评估了各组件影响,不同噪声水平下的鲁棒性实验也证明其稳定性优于现有先进方法。
原文摘要 · Abstract (English)
We propose a new representation learning solution for the classification of cognitive load based on Electroencephalogram (EEG). Our method integrates both time and frequency domains by first passing the raw EEG signals through the convolutional encoder to obtain the time domain representations. Next, we measure the Power Spectral Density (PSD) for all five EEG frequency bands and generate the channel power values as 2D images referred to as multi-spectral topography maps. These multi-spectral topography maps are then fed to a separate encoder to obtain the representations in frequency domain. Our solution employs a multi-domain attention module that maps these domain-specific embeddings onto a shared embedding space to emphasize more on important inter-domain relationships to enhance the representations for cognitive load classification. Additionally, we incorporate an orthogonal projection constraint during the training of our method to effectively increase the inter-class distances while improving intra-class clustering. This enhancement allows efficient discrimination between different cognitive states and aids in better grouping of similar states within the feature space. We validate the effectiveness of our model through extensive experiments on two public EEG datasets, CL-Drive and CLARE for cognitive load classification. Our results demonstrate the superiority of our multi-domain approach over the traditional single-domain techniques. Moreover, we conduct ablation and sensitivity analyses to assess the impact of various components of our method. Finally, robustness experiments on different amounts of added noise demonstrate the stability of our method compared to other state-of-the-art solutions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。