通过镜像对比损失提升脑电信号识别准确率
Mirror contrastive loss based sliding window transformer for subject-independent motor imagery based EEG signal recognition
- 用镜像对比损失增强对脑区事件相关去同步的敏感性
- 在两个数据集上分别达到66.48%和75.62%准确率
- 适合关注脑机接口可泛化性的研究者
尽管深度学习模型已广泛应用于基于运动想象的脑电信号识别,但其常被视为黑箱。受神经学发现启发——左/右肢想象会引发对侧运动感觉区的事件相关去同步(ERD),我们提出一种基于镜像对比损失的滑动窗口变换器(MCL-SWT),以提升跨被试运动想象脑电信号识别性能。具体而言,所提出的镜像对比损失通过对比原始脑电信号与其镜像信号(左右半球通道互换生成)来增强对ERD空间位置的敏感性。此外,引入时间滑动窗口变换器,从高时间分辨率特征中计算自注意力分数,从而在可控计算复杂度下提升模型表现。我们在跨被试运动想象脑电信号识别任务上评估MCL-SWT,实验结果表明其准确率分别达到66.48%和75.62%,较当前最优模型提升2.82%和2.17%。消融实验证明了镜像对比损失的有效性。MCL-SWT代码演示见https://github.com/roniusLuo/MCL_SWT。
原文摘要 · Abstract (English)
While deep learning models have been extensively utilized in motor imagery based EEG signal recognition, they often operate as black boxes. Motivated by neurological findings indicating that the mental imagery of left or right-hand movement induces event-related desynchronization (ERD) in the contralateral sensorimotor area of the brain, we propose a Mirror Contrastive Loss based Sliding Window Transformer (MCL-SWT) to enhance subject-independent motor imagery-based EEG signal recognition. Specifically, our proposed mirror contrastive loss enhances sensitivity to the spatial location of ERD by contrasting the original EEG signals with their mirror counterparts-mirror EEG signals generated by interchanging the channels of the left and right hemispheres of the EEG signals. Moreover, we introduce a temporal sliding window transformer that computes self-attention scores from high temporal resolution features, thereby improving model performance with manageable computational complexity. We evaluate the performance of MCL-SWT on subject-independent motor imagery EEG signal recognition tasks, and our experimental results demonstrate that MCL-SWT achieved accuracies of 66.48% and 75.62%, surpassing the state-of-the-art (SOTA) model by 2.82% and 2.17%, respectively. Furthermore, ablation experiments confirm the effectiveness of the proposed mirror contrastive loss. A code demo of MCL-SWT is available at https://github.com/roniusLuo/MCL_SWT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。