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 条
[21]   Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer Learning [J].
Shin, Hoo-Chang ;
Roth, Holger R. ;
Gao, Mingchen ;
Lu, Le ;
Xu, Ziyue ;
Nogues, Isabella ;
Yao, Jianhua ;
Mollura, Daniel ;
Summers, Ronald M. .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) :1285-1298
[22]   Subject and Class Specific Frequency Bands Selection for Multiclass Motor Imagery Classification [J].
Suk, Heung-Il ;
Lee, Seong-Whan .
INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2011, 21 (02) :123-130
[23]   EEG Motor Imagery Classification With Sparse Spectrotemporal Decomposition and Deep Learning [J].
Sun, Biao ;
Zhao, Xing ;
Zhang, Han ;
Bai, Ruifeng ;
Li, Ting .
IEEE TRANSACTIONS ON AUTOMATION SCIENCE AND ENGINEERING, 2021, 18 (02) :541-551
[24]   Single-trial EEG classification of motor imagery using deep convolutional neural networks [J].
Tang, Zhichuan ;
Li, Chao ;
Sun, Shouqian .
OPTIK, 2017, 130 :11-18
[25]  
Tonin L, 2011, IEEE ENG MED BIO, P4227, DOI 10.1109/IEMBS.2011.6091049
[26]   Deep learning-based BCI for gait decoding from EEG with LSTM recurrent neural network [J].
Tortora, Stefano ;
Ghidoni, Stefano ;
Chisari, Carmelo ;
Micera, Silvestro ;
Artoni, Fiorenzo .
JOURNAL OF NEURAL ENGINEERING, 2020, 17 (04)
[27]  
Wu D., 2020, TRANSFER LEARNING EE
[28]   Deep Fusion Feature Learning Network for MI-EEG Classification [J].
Yang, Jun ;
Yao, Shaowen ;
Wang, Jin .
IEEE ACCESS, 2018, 6 :79050-79059
[29]   Online Classification of Multiple Motor Imagery Tasks Using Filter Bank Based Maximum-a-Posteriori Common Spatial Pattern Filters [J].
Zahid, S. Z. ;
Aqil, M. ;
Tufail, M. ;
Nazir, M. S. .
IRBM, 2020, 41 (03) :141-150
[30]   A novel hybrid deep learning scheme for four-class motor imagery classification [J].
Zhang, Ruilong ;
Zong, Qun ;
Dou, Liqian ;
Zhao, Xinyi .
JOURNAL OF NEURAL ENGINEERING, 2019, 16 (06)