Recognition of Motor Imagery EEG Signals Based on Capsule Network

被引:3
作者
Du, Xiuli [1 ,2 ]
Kong, Meiya [1 ,2 ]
Qiu, Shaoming [1 ,2 ]
Guo, Jiangyu [3 ]
Lv, Yana [1 ,2 ]
机构
[1] Dalian Univ, Commun & Network Lab, Dalian, Peoples R China
[2] Dalian Univ, Sch Informat Engn, Dalian 116622, Peoples R China
[3] North Automat Control Technol Inst, Taiyuan, Shanxi, Peoples R China
关键词
Electroencephalography; Feature extraction; Three-dimensional displays; Heuristic algorithms; Routing; Electrodes; Convolution; motor imagery; 3D convolution; capsule network; individual differences; FEATURE-EXTRACTION; WAVELET TRANSFORM;
D O I
10.1109/ACCESS.2023.3262025
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In order to fully extract the temporal and spatial features contained in motor imagery electroencephalography (EEG) signals for effective identification of motor imagery, a three-dimensional capsule network (3D-CapsNet) EEG signal recognition model is proposed, which can integrate the MI-EEG temporal dimension, channel spatial dimension and the intrinsic relationship between features to maximize the feature representation capability. Firstly, a multi-layer 3D convolution module is used to extract features in the time and inter-channel space dimensions as the low-level features. Secondly, advanced spatial features are also obtained through capsule network integration. Finally, dynamic routing connections and squash functions are applied for classification. The experimental analysis is conducted on the BCI competition IV dataset 2a. The proposed model performs well on all the subjects' datasets, such that the average accuracy and average Kappa value of 9 subjects are 84.028% and 0.789, respectively. The experimental results confirm effectiveness of the proposed method. Additionally, accuracy of the four-class classification is improved, and the impact of individual variability is overcome to a certain extent.
引用
收藏
页码:31262 / 31271
页数:10
相关论文
共 24 条
[21]  
Tangermann M., FRONTIERS NEUROSCI
[22]  
Vallabhaneni A, 2005, BIOELECTRIC ENG, P85
[23]   Hybrid deep neural network using transfer learning for EEG motor imagery decoding [J].
Zhang, Ruilong ;
Zong, Qun ;
Dou, Liqian ;
Zhao, Xinyi ;
Tang, Yifan ;
Li, Zhiyu .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 63
[24]   A Multi-Branch 3D Convolutional Neural Network for EEG-Based Motor Imagery Classification [J].
Zhao, Xinqiao ;
Zhang, Hongmiao ;
Zhu, Guilin ;
You, Fengxiang ;
Kuang, Shaolong ;
Sun, Lining .
IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2019, 27 (10) :2164-2177