Automatic feature extraction and fusion recognition of motor imagery EEG using multilevel multiscale CNN

被引:37
作者
Li, Ming-ai [1 ,2 ,3 ]
Han, Jian-fu [1 ]
Yang, Jin-fu [1 ,2 ]
机构
[1] Beijing Univ Technol, Fac Informat Technol, Beijing 100124, Peoples R China
[2] Beijing Key Lab Computat Intelligence & Intellige, Beijing 100124, Peoples R China
[3] Minist Educ, Engn Res Ctr Digital Community, Beijing 100124, Peoples R China
基金
中国国家自然科学基金;
关键词
Brain-computer interface; Convolutional neural network; Wavelet transform; MI-EEG imaging method; Machine learning; CONVOLUTIONAL NEURAL-NETWORK; CLASSIFICATION; SIGNALS; TASKS;
D O I
10.1007/s11517-021-02396-w
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
A motor imagery EEG (MI-EEG) signal is often selected as the driving signal in an active brain computer interface (BCI) system, and it has been a popular field to recognize MI-EEG images via convolutional neural network (CNN), which poses a potential problem for maintaining the integrity of the time-frequency-space information in MI-EEG images and exploring the feature fusion mechanism in the CNN. However, information is excessively compressed in the present MI-EEG image, and the sequential CNN is unfavorable for the comprehensive utilization of local features. In this paper, a multidimensional MI-EEG imaging method is proposed, which is based on time-frequency analysis and the Clough-Tocher (CT) interpolation algorithm. The time-frequency matrix of each electrode is generated via continuous wavelet transform (WT), and the relevant section of frequency is extracted and divided into nine submatrices, the longitudinal sums and lengths of which are calculated along the directions of frequency and time successively to produce a 3 x 3 feature matrix for each electrode. Then, feature matrix of each electrode is interpolated to coincide with their corresponding coordinates, thereby yielding a WT-based multidimensional image, called WTMI. Meanwhile, a multilevel and multiscale feature fusion convolutional neural network (MLMSFFCNN) is designed for WTMI, which has dense information, low signal-to-noise ratio, and strong spatial distribution. Extensive experiments are conducted on the BCI Competition IV 2a and 2b datasets, and accuracies of 92.95% and 97.03% are yielded based on 10-fold cross-validation, respectively, which exceed those of the state-of-the-art imaging methods. The kappa values and p values demonstrate that our method has lower class skew and error costs. The experimental results demonstrate that WTMI can fully represent the time-frequency-space features of MI-EEG and that MLMSFFCNN is beneficial for improving the collection of multiscale features and the fusion recognition of general and abstract features for WTMI.
引用
收藏
页码:2037 / 2050
页数:14
相关论文
共 46 条
[11]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[12]   Classification of multi-class motor imagery with a novel hierarchical SVM algorithm for brain-computer interfaces [J].
Dong, Enzeng ;
Li, Changhai ;
Li, Liting ;
Du, Shengzhi ;
Belkacem, Abdelkader Nasreddine ;
Chen, Chao .
MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2017, 55 (10) :1809-1818
[13]   An end-to-end deep learning approach to MI-EEG signal classification for BCIs [J].
Dose, Hauke ;
Moller, Jakob S. ;
Iversen, Helle K. ;
Puthusserypady, Sadasivan .
EXPERT SYSTEMS WITH APPLICATIONS, 2018, 114 :532-542
[14]   Nonstationary Brain Source Separation for Multiclass Motor Imagery [J].
Gouy-Pailler, Cedric ;
Congedo, Marco ;
Brunner, Clemens ;
Jutten, Christian ;
Pfurtscheller, Gert .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2010, 57 (02) :469-478
[15]   Motor Imagery EEG Classification Using Capsule Networks [J].
Ha, Kwon-Woo ;
Jeong, Jin-Woo .
SENSORS, 2019, 19 (13)
[16]  
Karen Simonyan, 2014, ABS14091556 CORR
[17]   Motor Imagery Based EEG Classification by Using Common Spatial Patterns and Convolutional Neural Networks [J].
Korhan, Nuri ;
Dokur, Zumray ;
Olmez, Tamer .
2019 SCIENTIFIC MEETING ON ELECTRICAL-ELECTRONICS & BIOMEDICAL ENGINEERING AND COMPUTER SCIENCE (EBBT), 2019,
[18]  
Lee G.R., 2019, J. Open Source Softw, V4, DOI DOI 10.21105/JOSS.01237
[19]   Brain-computer communication: Motivation, aim, and impact of exploring a virtual apartment [J].
Leeb, Robert ;
Lee, Felix ;
Keinrath, Claudia ;
Scherer, Reinhold ;
Bischof, Horst ;
Pfurtscheller, Gert .
IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2007, 15 (04) :473-482
[20]   MINIMUM CROSS ENTROPY THRESHOLDING [J].
LI, CH ;
LEE, CK .
PATTERN RECOGNITION, 1993, 26 (04) :617-625