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 条
  • [31] DeepMI: Deep Learning for Multiclass Motor Imagery Classification
    Abbas, Waseem
    Khan, Nadeem Ahmad
    2018 40TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY (EMBC), 2018, : 219 - 222
  • [32] A Survey on Deep Learning Classification Algorithms for Motor Imagery
    Guragai, Bishal
    AlShorman, Omar
    Masadeh, Mahmoud
    Bin Heyat, Md Belal
    2020 32ND INTERNATIONAL CONFERENCE ON MICROELECTRONICS (ICM), 2020, : 281 - 284
  • [33] Deep Learning Classification of two-class Motor Imagery EEG signals using Transfer Learning
    Shajil, Nijisha
    Sasikala, M.
    Arunnagiri, A. M.
    2020 INTERNATIONAL CONFERENCE ON E-HEALTH AND BIOENGINEERING (EHB), 2020,
  • [34] 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
  • [35] HR-SNN: An End-to-End Spiking Neural Network for Four-Class Classification Motor Imagery Brain-Computer Interface
    Li, Yulin
    Fan, Liangwei
    Shen, Hui
    Hu, Dewen
    IEEE TRANSACTIONS ON COGNITIVE AND DEVELOPMENTAL SYSTEMS, 2024, 16 (06) : 1955 - 1968
  • [36] A trial deep learning-based model for four-class histologic classification of colonic tumor from narrow band imaging
    Takeshi Shimizu
    Yoshihiro Sasaki
    Kei Ito
    Masashi Matsuzaka
    Hirotake Sakuraba
    Shinsaku Fukuda
    Scientific Reports, 13
  • [37] Multi-class Classification of Motor Imagery EEG Signals Using Deep Learning Models
    Khemakhem R.
    Belgacem S.
    Echtioui A.
    Ghorbel M.
    Ben Hamida A.
    Kammoun I.
    SN Computer Science, 5 (5)
  • [38] Tensor-CSPNet: A Novel Geometric Deep Learning Framework for Motor Imagery Classification
    Ju, Ce
    Guan, Cuntai
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (12) : 10955 - 10969
  • [39] Four-Class Classification of Skin Lesions With Task Decomposition Strategy
    Shimizu, Kouhei
    Iyatomi, Hitoshi
    Celebi, M. Emre
    Norton, Kerri-Ann
    Tanaka, Masaru
    IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2015, 62 (01) : 274 - 283
  • [40] A trial deep learning-based model for four-class histologic classification of colonic tumor from narrow band imaging
    Shimizu, Takeshi
    Sasaki, Yoshihiro
    Ito, Kei
    Matsuzaka, Masashi
    Sakuraba, Hirotake
    Fukuda, Shinsaku
    SCIENTIFIC REPORTS, 2023, 13 (01)