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 条
[1]   A Deep Learning Framework for Decoding Motor Imagery Tasks of the Same Hand Using EEG Signals [J].
Alazrai, Rami ;
Abuhijleh, Motaz ;
Alwanni, Hisham ;
Daoud, Mohammad, I .
IEEE ACCESS, 2019, 7 :109612-109627
[2]  
Alfeld P., 1984, Computer-Aided Geometric Design, V1, P169, DOI 10.1016/0167-8396(84)90029-3
[3]  
Alom M.Z., 2018, ARXIV180206955
[4]   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
[5]   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
[6]  
[Anonymous], 2009, International Journal of Digital Content Technology and its Applications, DOI DOI 10.4156/JDCTA.VOL3.ISSUE2.HU
[7]   Spatial filtering and selection of optimized components in four class motor imagery EEG data using independent components analysis [J].
Brunner, Clemens ;
Naeem, Muhammad ;
Leeb, Robert ;
Graimann, Bernhard ;
Pfurtscheller, Gert .
PATTERN RECOGNITION LETTERS, 2007, 28 (08) :957-964
[8]   Meditation states and traits: EEG, ERP, and neuroimaging studies [J].
Cahn, BR ;
Polich, J .
PSYCHOLOGICAL BULLETIN, 2006, 132 (02) :180-211
[9]   Convolutional Neural Network Based Approach Towards Motor Imagery Tasks EEG Signals Classification [J].
Chaudhary, Shalu ;
Taran, Sachin ;
Bajaj, Varun ;
Sengur, Abdulkadir .
IEEE SENSORS JOURNAL, 2019, 19 (12) :4494-4500
[10]   EEG Classification of Motor Imagery Using a Novel Deep Learning Framework [J].
Dai, Mengxi ;
Zheng, Dezhi ;
Na, Rui ;
Wang, Shuai ;
Zhang, Shuailei .
SENSORS, 2019, 19 (03)