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
相关论文
共 59 条
[1]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[2]  
[Anonymous], Riemannian geometry
[3]   COMPUTING MEDIANS AND MEANS IN HADAMARD SPACES [J].
Bacak, Miroslav .
SIAM JOURNAL ON OPTIMIZATION, 2014, 24 (03) :1542-1566
[4]   Spatio-Spectral Feature Representation for Motor Imagery Classification Using Convolutional Neural Networks [J].
Bang, Ji-Seon ;
Lee, Min-Ho ;
Fazli, Siamac ;
Guan, Cuntai ;
Lee, Seong-Whan .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2022, 33 (07) :3038-3049
[5]   Multiclass Brain-Computer Interface Classification by Riemannian Geometry [J].
Barachant, Alexandre ;
Bonnet, Stephane ;
Congedo, Marco ;
Jutten, Christian .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2012, 59 (04) :920-928
[7]  
Belkin M, 2002, ADV NEUR IN, V14, P585
[8]  
Bhatia R, 2007, PRINC SER APPL MATH, P1
[9]   Optimizing spatial filters for robust EEG single-trial analysis [J].
Blankertz, Benjamin ;
Tomioka, Ryota ;
Lemm, Steven ;
Kawanabe, Motoaki ;
Mueller, Klaus-Robert .
IEEE SIGNAL PROCESSING MAGAZINE, 2008, 25 (01) :41-56
[10]   Geometric Deep Learning Going beyond Euclidean data [J].
Bronstein, Michael M. ;
Bruna, Joan ;
LeCun, Yann ;
Szlam, Arthur ;
Vandergheynst, Pierre .
IEEE SIGNAL PROCESSING MAGAZINE, 2017, 34 (04) :18-42