A Novel Algorithmic Structure of EEG Channel Attention Combined With Swin Transformer for Motor Patterns Classification

Abstract

With the development of brain-computer interfaces (BCI) technologies, EEG-based BCI applications have been deployed for medical purposes. Motor imagery (MI), applied to promote neural rehabilitation for stroke patients, is among the most common BCI paradigms that. The Electroencephalogram (EEG) signals, encompassing an extensive range of channels, render the training dataset a high-dimensional construct. This high dimensionality, inherent in such a dataset, tends to challenge traditional deep learning approaches, causing them to potentially disregard the intrinsic correlations amongst these channels. Such an oversight often culminates in erroneous data classification, presenting a significant drawback of these conventional methodologies. In our study, we propose a novel algorithmic structure of EEG channel-attention combined with Swin Transformer for motor pattern recognition in BCI rehabilitation. Effectively, the self-attention module from transformer architecture could captures temporal-spectral-spatial features hidden in EEG data. The experimental results verify that our proposed methods outperformed other state-of-art approaches with the average accuracy of 87.67%. It is implied that our method can extract high-level and latent connections among temporal-spectral features in contrast to traditional deep learning methods. This paper demonstrates that channel-attention combined with Swin Transformer methods has great potential for implementing high-performance motor pattern-based BCI systems.

Keywords

• EEG classification
• attention
• transformer
• deep learning
• stroke