A novel hybrid deep learning scheme for four-class motor imagery classification

被引:135
作者
Zhang, Ruilong [1 ]
Zong, Qun [1 ]
Dou, Liqian [1 ]
Zhao, Xinyi [1 ]
机构
[1] Tianjin Univ, Sch Elect & Informat Engn, Tianjin, Peoples R China
基金
中国国家自然科学基金;
关键词
brain-computer interface; four-class motor imagery; OVR-FBCSP; convolutional neural network; long short-term memory; BRAIN-COMPUTER-INTERFACE; NEURAL-NETWORKS; EEG; BCI;
D O I
10.1088/1741-2552/ab3471
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Objective. Learning the structures and unknown correlations of a motor imagery electroencephalogram (MI-EEG) signal is important for its classification. It is also a major challenge to obtain good classification accuracy from the increased number of classes and increased variability from different people. In this study, a four-class MI task is investigated. Approach. An end-to-end novel hybrid deep learning scheme is developed to decode the MI task from EEG data. The proposed algorithm consists of two parts: a. A one-versus-rest filter bank common spatial pattern is adopted to preprocess and pre-extract the features of the four-class MI signal. b. A hybrid deep network based on the convolutional neural network and long-term short-term memory network is proposed to extract and learn the spatial and temporal features of the MI signal simultaneously. Main results. The main contribution of this paper is to propose a hybrid deep network framework to improve the classification accuracy of the four-class MI-EEG signal. The hybrid deep network is a subject-independent shared neural network, which means it can be trained by using the training data from all subjects to form one model. Significance. The classification performance obtained by the proposed algorithm on brain-computer interface (BCI) competition IV dataset 2a in terms of accuracy is 83% and Cohen's kappa value is 0.80. Finally, the shared hybrid deep network is evaluated by every subject respectively, and the experimental results illustrate that the shared neural network has satisfactory accuracy. Thus, the proposed algorithm could be of great interest for real-life BCIs.
引用
收藏
页数:11
相关论文
共 52 条
[1]  
Ahmed S, 2013, IEEE GLOB CONF SIG, P33, DOI 10.1109/GlobalSIP.2013.6736804
[2]   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)
[3]   Multilevel Weighted Feature Fusion Using Convolutional Neural Networks for EEG Motor Imagery Classification [J].
Amin, Syed Umar ;
Alsulaiman, Mansour ;
Muhammad, Ghulam ;
Bencherif, Mohamed A. ;
Hossain, M. Shamim .
IEEE ACCESS, 2019, 7 :18940-18950
[4]   A Deep Learning Method for Classification of EEG Data Based on Motor Imagery [J].
An, Xiu ;
Kuang, Deping ;
Guo, Xiaojiao ;
Zhao, Yilu ;
He, Lianghua .
INTELLIGENT COMPUTING IN BIOINFORMATICS, 2014, 8590 :203-210
[5]   Filter bank common spatial pattern algorithm on BCI competition IV Datasets 2a and 2b [J].
Ang, Kai Keng ;
Chin, Zheng Yang ;
Wang, Chuanchu ;
Guan, Cuntai ;
Zhang, Haihong .
FRONTIERS IN NEUROSCIENCE, 2012, 6
[6]  
[Anonymous], 2015, Nature, DOI [10.1038/nature14539, DOI 10.1038/NATURE14539]
[7]  
[Anonymous], 2006, INT C P IEEE ENG MED
[8]  
[Anonymous], IEEE ACCESS
[9]  
[Anonymous], 2017, NEUROCOMPUTING
[10]  
[Anonymous], 2014, 2 INT C LEARN REPR