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

被引:128
作者
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] 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
  • [2] Convolutional Neural Networks for Four-Class Motor Imagery Data Classification
    Uktveris, Tomas
    Jusas, Vacius
    INTELLIGENT DISTRIBUTED COMPUTING XI, 2018, 737 : 185 - 197
  • [3] 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
  • [4] Increase performance of four-class classification for Motor-Imagery based Brain-Computer Interface
    Le Quoc Thang
    Temiyasathit, Chivalai
    2014 INTERNATIONAL CONFERENCE ON COMPUTER, INFORMATION AND TELECOMMUNICATION SYSTEMS (CITS), 2014,
  • [5] Classification of Four-Class Motor Imagery Employing Single-Channel Electroencephalography
    Ge, Sheng
    Wang, Ruimin
    Yu, Dongchuan
    PLOS ONE, 2014, 9 (06):
  • [6] A novel deep learning approach for classification of EEG motor imagery signals
    Tabar, Yousef Rezaei
    Halici, Ugur
    JOURNAL OF NEURAL ENGINEERING, 2017, 14 (01)
  • [7] 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
  • [8] CLTNet: A Hybrid Deep Learning Model for Motor Imagery Classification
    Gu, He
    Chen, Tingwei
    Ma, Xiao
    Zhang, Mengyuan
    Sun, Yan
    Zhao, Jian
    BRAIN SCIENCES, 2025, 15 (02)
  • [9] Classification of the four-class motor imagery signals using continuous wavelet transform filter bank-based two-dimensional images
    Mahamune, Rupesh
    Laskar, Shahedul H.
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2021, 31 (04) : 2237 - 2248
  • [10] EEG Classification of Motor Imagery Using a Novel Deep Learning Framework
    Dai, Mengxi
    Zheng, Dezhi
    Na, Rui
    Wang, Shuai
    Zhang, Shuailei
    SENSORS, 2019, 19 (03)