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

被引:125
|
作者
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
相关论文
共 50 条
  • [1] Feature extraction and classification of four-class motor imagery EEG data
    Shi, Jin-He
    Shen, Ji-Zhong
    Wang, Pan
    Zhejiang Daxue Xuebao (Gongxue Ban)/Journal of Zhejiang University (Engineering Science), 2012, 46 (02): : 338 - 344
  • [2] Convolutional Neural Networks for Four-Class Motor Imagery Data Classification
    Uktveris, Tomas
    Jusas, Vacius
    INTELLIGENT DISTRIBUTED COMPUTING XI, 2018, 737 : 185 - 197
  • [3] Classification of Four-Class Motor Imagery Employing Single-Channel Electroencephalography
    Ge, Sheng
    Wang, Ruimin
    Yu, Dongchuan
    PLOS ONE, 2014, 9 (06):
  • [4] Application of Convolutional Neural Networks to Four-Class Motor Imagery Classification Problem
    Uktveris, Tomas
    Jusas, Vacius
    INFORMATION TECHNOLOGY AND CONTROL, 2017, 46 (02): : 260 - 273
  • [5] Research on four-class motor imagery EEG classification method based on ITD and PLV
    Jiang G.
    Chen W.
    Ma D.
    Wu J.
    Yi Qi Yi Biao Xue Bao/Chinese Journal of Scientific Instrument, 2019, 40 (05): : 195 - 202
  • [6] Deep CNN model based on serial-parallel structure optimization for four-class motor imagery EEG classification
    Zhao, Xuefei
    Liu, Dong
    Ma, Li
    Liu, Quan
    Chen, Kun
    Xie, Shane
    Ai, Qingsong
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 72
  • [7] Continuous detection of motor imagery in a four-class asynchronous BCI
    Sadeghian, E. B.
    Moradi, A. H.
    2007 ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY, VOLS 1-16, 2007, : 3241 - 3244
  • [8] A lightweight and accurate double-branch neural network for four-class motor imagery classification
    Ma, Weifeng
    Gong, Yifei
    Xue, Haojie
    Liu, Yang
    Lin, Xuefen
    Zhou, Gongxue
    Li, Yaru
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 75
  • [9] A new phase-based feature extraction method for four-class motor imagery classification
    Tosun, Mustafa
    Cetin, Osman
    SIGNAL IMAGE AND VIDEO PROCESSING, 2022, 16 (01) : 283 - 290
  • [10] A new phase-based feature extraction method for four-class motor imagery classification
    Mustafa Tosun
    Osman Çetin
    Signal, Image and Video Processing, 2022, 16 : 283 - 290