Tensor-CSPNet: A Novel Geometric Deep Learning Framework for Motor Imagery Classification

被引:34
|
作者
Ju, Ce [1 ]
Guan, Cuntai [1 ]
机构
[1] Nanyang Technol Univ, Sch Comp Sci & Engn, Singapore 639798, Singapore
关键词
Electroencephalography; Manifolds; Tensors; Task analysis; Deep learning; Covariance matrices; Visualization; Electroencephalography (EEG)-based brain-computer interfaces (BCIs); geometric deep learning (DL); motor imagery (MI) classification; symmetric positive definite (SPD) manifolds; BRAIN-COMPUTER INTERFACES; COMMON SPATIAL-PATTERN; RIEMANNIAN GEOMETRY; EEG CLASSIFICATION; EEG/MEG; FILTERS;
D O I
10.1109/TNNLS.2022.3172108
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning (DL) has been widely investigated in a vast majority of applications in electroencephalography (EEG)-based brain-computer interfaces (BCIs), especially for motor imagery (MI) classification in the past five years. The mainstream DL methodology for the MI-EEG classification exploits the temporospatial patterns of EEG signals using convolutional neural networks (CNNs), which have been particularly successful in visual images. However, since the statistical characteristics of visual images depart radically from EEG signals, a natural question arises whether an alternative network architecture exists apart from CNNs. To address this question, we propose a novel geometric DL (GDL) framework called Tensor-CSPNet, which characterizes spatial covariance matrices derived from EEG signals on symmetric positive definite (SPD) manifolds and fully captures the temporospatiofrequency patterns using existing deep neural networks on SPD manifolds, integrating with experiences from many successful MI-EEG classifiers to optimize the framework. In the experiments, Tensor-CSPNet attains or slightly outperforms the current state-of-the-art performance on the cross-validation and holdout scenarios in two commonly used MI-EEG datasets. Moreover, the visualization and interpretability analyses also exhibit the validity of Tensor-CSPNet for the MI-EEG classification. To conclude, in this study, we provide a feasible answer to the question by generalizing the DL methodologies on SPD manifolds, which indicates the start of a specific GDL methodology for the MI-EEG classification.
引用
收藏
页码:10955 / 10969
页数:15
相关论文
共 50 条
  • [1] 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)
  • [2] A novel deep learning approach for classification of EEG motor imagery signals
    Tabar, Yousef Rezaei
    Halici, Ugur
    JOURNAL OF NEURAL ENGINEERING, 2017, 14 (01)
  • [3] Motor imagery electroencephalogram classification based on Riemannian geometry and deep learning framework
    Lu, Bo
    Han, Shengjie
    Ding, Xiaotong
    Li, Guang
    PROCEEDINGS OF 2023 4TH INTERNATIONAL SYMPOSIUM ON ARTIFICIAL INTELLIGENCE FOR MEDICINE SCIENCE, ISAIMS 2023, 2023, : 59 - 62
  • [4] 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
  • [5] 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
  • [6] A novel hybrid deep learning scheme for four-class motor imagery classification
    Zhang, Ruilong
    Zong, Qun
    Dou, Liqian
    Zhao, Xinyi
    JOURNAL OF NEURAL ENGINEERING, 2019, 16 (06)
  • [7] Motor Imagery Classification Emphasizing Corresponding Frequency Domain Method based on Deep Learning Framework
    Kwon, Byoung-Hee
    Lee, Byeong-Hoo
    Jeong, Ji-Hoon
    2021 9TH IEEE INTERNATIONAL WINTER CONFERENCE ON BRAIN-COMPUTER INTERFACE (BCI), 2021, : 268 - 271
  • [8] Deep Learning Solutions for Motor Imagery Classification: A Comparison Study
    Lu, Na
    Yin, Tao
    Jing, Xue
    2020 8TH INTERNATIONAL WINTER CONFERENCE ON BRAIN-COMPUTER INTERFACE (BCI), 2020, : 201 - 206
  • [9] 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)
  • [10] Classification of Motor Imagery EEG Signals with Deep Learning Models
    Shen, Yurun
    Lu, Hongtao
    Jia, Jie
    INTELLIGENCE SCIENCE AND BIG DATA ENGINEERING, ISCIDE 2017, 2017, 10559 : 181 - 190