Motor imagery EEG classification algorithm based on CNN-LSTM feature fusion network

被引:112
作者
Li, Hongli [1 ]
Ding, Man [1 ]
Zhang, Ronghua [2 ]
Xiu, Chunbo [1 ]
机构
[1] Tiangong Univ, Sch Control Sci & Engn, Tianjin 300387, Peoples R China
[2] Tiangong Univ, Sch Artificial Intelligence, Tianjin 300387, Peoples R China
关键词
Motor Imagery; EEG classification; CNN; LSTM; Feature fusion; CONVOLUTIONAL NEURAL-NETWORKS;
D O I
10.1016/j.bspc.2021.103342
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Motor imagery brain-computer interface (MI-BCI) provides a novel way for human-computer interaction. Traditional neural networks often use serial structure to extract spatial features when dealing with motor imagery EEG signal classification, ignoring temporal information and a large amount of available information in the middle layer, resulting in poor classification performance of MI-BCI. A neural network feature fusion algorithm is proposed by combining the convolutional neural network (CNN) and the long-short-term memory network (LSTM). Specifically, the CNN and LSTM are connected in parallel. The CNN extracts spatial features, the LSTM extracts temporal features and the flatten layer added after the convolutional layer extracts the middle layer features. Then all the features are merged in the fully connected layer to improve the accuracy of classification. The average accuracy and Kappa value of all subjects were 87.68% and 0.8245, respectively. The result shows that the feature fusion neural network proposed in this paper can effectively improve the accuracy of motor imagery EEG, and provides new ideas for the study of feature extraction and classification of motor imagery brain-computer interfaces.
引用
收藏
页数:7
相关论文
共 30 条
[1]   Feature extraction of four-class motor imagery EEG signals based on functional brain network [J].
Ai, Qingsong ;
Chen, Anqi ;
Chen, Kun ;
Liu, Quan ;
Zhou, Tichao ;
Xin, Sijin ;
Ji, Ze .
JOURNAL OF NEURAL ENGINEERING, 2019, 16 (02)
[2]   Deep Learning for EEG motor imagery classification based on multi-layer CNNs feature fusion [J].
Amin, Syed Umar ;
Alsulaiman, Mansour ;
Muhammad, Ghulam ;
Mekhtiche, Mohamed Amine ;
Hossain, M. Shamim .
FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2019, 101 :542-554
[3]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[4]  
Bhattacharjee P., 2012, INT C PATTERN RECOGN
[5]   Deep Temporal-Spatial Feature Learning for Motor Imagery-Based Brain-Computer Interfaces [J].
Chen, Junjian ;
Yu, Zhuliang ;
Gu, Zhenghui ;
Li, Yuanqing .
IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2020, 28 (11) :2356-2366
[6]  
Chu Yaqi, 2021, Sheng Wu Yi Xue Gong Cheng Xue Za Zhi, V38, P1, DOI 10.7507/1001-5515.202007006
[7]  
DUAN F, 2017, IEEE T AUTON MENT DE, P1
[8]   Investigating the effects of visual distractors on the performance of a motor imagery brain-computer interface [J].
Emami, Zahra ;
Chau, Tom .
CLINICAL NEUROPHYSIOLOGY, 2018, 129 (06) :1268-1275
[9]   Applying Deep Learning for Epilepsy Seizure Detection and Brain Mapping Visualization [J].
Hossain, M. Shamim ;
Amin, Syed Umar ;
Alsulaiman, Mansour ;
Muhammad, Ghulam .
ACM TRANSACTIONS ON MULTIMEDIA COMPUTING COMMUNICATIONS AND APPLICATIONS, 2019, 15 (01)
[10]   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)