A Multi-feature Fusion Transformer Neural Network for Motor Imagery EEG Signal Classification

被引:0
作者
Hu, Zhangfang [1 ]
He, Lingxiao [2 ]
Wu, Haoze [2 ]
机构
[1] Chongqing Univ Posts & Telecommun, Sch Optoelect Engn, Key Lab Opt Informat Sensing & Technol, Chongqing 400065, Peoples R China
[2] Chongqing Univ Posts & Telecommun, Sch Optoelect Engn, Chongqing 400065, Peoples R China
基金
中国国家自然科学基金;
关键词
motor imagery; brain-computer interface; Transformer; multi-feature fusion; CNN; COMMON SPATIAL-PATTERN;
D O I
暂无
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
recent years, the classification method of motor imagery(MI) electroencephalography(EEG) signals based on deep learning(DL) has become more and more mature in the field of brain-computer interface(BIC). However, most of the studies tend to use a single feature or two associated features when dealing with motor imagery EEG signal classification, while ignoring other features, resulting in poor classification performance. Therefore, this paper proposes a neural network feature fusion algorithm called Multi-feature Fusion Transformer(M-FFT). The network is built based on convolutional neural network (CNN) and Transformer. This method uses CNN and wavelet transform to extract the time frequency and space-time features, and uses the converter to fuse the three feature domains contained in the two features to establish the information interaction between the three feature domains contained in the two features. Then, the global feature pooling is used to output the feature vector, and finally the softmax function is used to classify the feature vector. In the training process, we use cross entropy as the loss function. Finally, on the brain-computer interface competition IV data set 2a, the average classification accuracy is 85.66%, and the average kappa value is 0.833. The experimental results validate the algorithm performance.
引用
收藏
页码:1822 / 1831
页数:10
相关论文
共 30 条
[1]   Deep learning techniques for classification of electroencephalogram (EEG) motor imagery (MI) signals: a review [J].
Altaheri, Hamdi ;
Muhammad, Ghulam ;
Alsulaiman, Mansour ;
Amin, Syed Umar ;
Altuwaijri, Ghadir Ali ;
Abdul, Wadood ;
Bencherif, Mohamed A. ;
Faisal, Mohammed .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (20) :14681-14722
[2]   Motor Imagery Classification for Brain Computer Interface Using Deep Metric Learning [J].
Alwasiti, Haider ;
Yusoff, Mohd Zuki ;
Raza, Kamran .
IEEE ACCESS, 2020, 8 :109949-109963
[3]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[4]  
[Anonymous], 2008, BCI Competition 2008-Graz data set A, V16, P1
[5]  
[Anonymous], 2010, Bmc Neuroscience, DOI DOI 10.1186/1471-2202-11-S1-P127
[6]   Three-Class EEG-Based Motor Imagery Classification Using Phase-Space Reconstruction Technique [J].
Djemal, Ridha ;
Bazyed, Ayad G. ;
Belwafi, Kais ;
Gannouni, Sofien ;
Kaaniche, Walid .
BRAIN SCIENCES, 2016, 6 (03)
[7]   FINDING STRUCTURE IN TIME [J].
ELMAN, JL .
COGNITIVE SCIENCE, 1990, 14 (02) :179-211
[8]   Bilinear neural network with 3-D attention for brain decoding of motor imagery movements from the human EEG [J].
Fan, Chen-Chen ;
Yang, Hongjun ;
Hou, Zeng-Guang ;
Ni, Zhen-Liang ;
Chen, Sheng ;
Fang, Zhijie .
COGNITIVE NEURODYNAMICS, 2021, 15 (01) :181-189
[9]   A novel end-to-end deep learning scheme for classifying multi-class motor imagery electroencephalography signals [J].
Hassanpour, Ahmad ;
Moradikia, Majid ;
Adeli, Hojjat ;
Khayami, Seyed Raouf ;
Shamsinejadbabaki, Pirooz .
EXPERT SYSTEMS, 2019, 36 (06)
[10]   Composite Common Spatial Pattern for Subject-to-Subject Transfer [J].
Kang, Hyohyeong ;
Nam, Yunjun ;
Choi, Seungjin .
IEEE SIGNAL PROCESSING LETTERS, 2009, 16 (08) :683-686