A Multi-scale Residual Network Based on the Multi-head Attention Mechanism for Motor Imagery EEG Decoding

被引:0
作者
Li, Ketong [1 ]
Liu, Xiaodong [1 ]
Chen, Qian [1 ]
Chen, Peng [1 ]
机构
[1] Southwest Jiaotong Univ, Sch Mech Engn, Chengdu, Sichuan, Peoples R China
来源
2024 4TH INTERNATIONAL CONFERENCE ON INDUSTRIAL AUTOMATION, ROBOTICS AND CONTROL ENGINEERING, IARCE | 2024年
关键词
motor imagery; muti-head attention; residual network; convolutional neural network; electroencephalography; CLASSIFICATION;
D O I
10.1109/IARCE64300.2024.00052
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Recent advancements in deep learning have led to increased application in brain-computer interfaces (BCIs) for decoding motor imagery electroencephalogram (EEG) signals. Nevertheless, current methods still present considerable challenges in effectively decoding these signals. Convolutional neural networks (CNNs), most used in this domain, are constrained by limitations associated with their kernel size and inherently limited receptive fields. This restricts their capacity to fully capture the global spatial and temporal dependencies present in EEG signals, which is crucial for accurate decoding. In response to these limitations, we propose a novel approach: a multi-scale residual network based on the multi-head attention mechanism. This model is designed to overcome the constraints of traditional CNNs by incorporating a multi-head attention layer, four convolutional layers, and a pooling layer, which enable the extraction of global and local features. To verify the effectiveness of our model, we conducted an experiment involving left- and right-hand fist imagery. We evaluate the performance of our model against several state-of-the-art models currently used in the field. The experimental results indicate that our model performs significantly better than the baseline model, with superior decoding accuracy, and our model is faster and more stable in the convergence of the loss function. It achieves an optimal balance between model complexity and accuracy, which makes it more advantageous in practical applications.
引用
收藏
页码:241 / 244
页数:4
相关论文
共 13 条
[1]   Motor imagery EEG signal classification using image processing technique over GoogLeNet deep learning algorithm for controlling the robot manipulator [J].
Ak, Ayca ;
Topuz, Vedat ;
Midi, Ipek .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 72
[2]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[3]   THE WAVELET TRANSFORM, TIME-FREQUENCY LOCALIZATION AND SIGNAL ANALYSIS [J].
DAUBECHIES, I .
IEEE TRANSACTIONS ON INFORMATION THEORY, 1990, 36 (05) :961-1005
[4]   EEG-Based Volitional Control of Prosthetic Legs for Walking in Different Terrains [J].
Gao, Hongbo ;
Luo, Ling ;
Pi, Ming ;
Li, Zhijun ;
Li, Qinjian ;
Zhao, Kuankuan ;
Huang, Junliang .
IEEE TRANSACTIONS ON AUTOMATION SCIENCE AND ENGINEERING, 2021, 18 (02) :530-540
[5]   HSI-BERT: Hyperspectral Image Classification Using the Bidirectional Encoder Representation From Transformers [J].
He, Ji ;
Zhao, Lina ;
Yang, Hongwei ;
Zhang, Mengmeng ;
Li, Wei .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (01) :165-178
[6]   EEGNet: a compact convolutional neural network for EEG-based brain-computer interfaces [J].
Lawhern, Vernon J. ;
Solon, Amelia J. ;
Waytowich, Nicholas R. ;
Gordon, Stephen M. ;
Hung, Chou P. ;
Lance, Brent J. .
JOURNAL OF NEURAL ENGINEERING, 2018, 15 (05)
[7]   An EEG-based stereoscopic research of the PSD differences in pre and post 2D&3D movies watching [J].
Manshouri, Negin ;
Maleki, Masoud ;
Kayikcioglu, Temel .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2020, 55
[8]   Designing optimal spatial filters for single-trial EEG classification in a movement task [J].
Müller-Gerking, J ;
Pfurtscheller, G ;
Flyvbjerg, H .
CLINICAL NEUROPHYSIOLOGY, 1999, 110 (05) :787-798
[9]  
Pfurtscheller G, 1998, IEEE Trans Rehabil Eng, V6, P316, DOI 10.1109/86.712230
[10]   Motor imagery and direct brain-computer communication [J].
Pfurtscheller, G ;
Neuper, C .
PROCEEDINGS OF THE IEEE, 2001, 89 (07) :1123-1134