Mirror contrastive loss based sliding window transformer for subject-independent motor imagery based EEG signal recognition

Jing Luo, Qi Mao, Weiwei Shi, Zhenghao Shi, Xiaofan Wang +2 more
8/29/2024
eess.SPcs.AIcs.LG

Official code: roniusLuo/MCL_SWT (5 stars, Apache-2.0, last commit 2024-07-01).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

roniusLuo/MCL_SWTOfficial100%
Apache-2.0

Jing Luo et.al. Mirror contrastive loss based sliding window transformer for subject-independent motor imagery based EEG signal recognition[C]//Human Brain and Artificial Intelligence: Fourth International Workshop, HBAI 2024, Held in Conjunction with IJCAI 2024, Jeju, South Korea, Aug 3, 2024. Accepted

50PythonJul 1, 20242 years agoApache-2.0

Similar Papers

Cite this paper

@article{luo2024mirror,
  title  = {Mirror contrastive loss based sliding window transformer for subject-independent motor imagery based EEG signal recognition},
  author = {Jing Luo and Qi Mao and Weiwei Shi and Zhenghao Shi and Xiaofan Wang and Xiaofeng Lu and Xinhong Hei},
  year   = {2024},
  eprint = {2409.00130},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.00130v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2409.00130.svg)](https://codeofpaper.com/paper/2409.00130)

Discussion