FB-CGANet: filter bank channel group attention network for multi-class motor imagery classification

被引:21
作者
Chen, Jiaming [1 ]
Yi, Weibo [2 ]
Wang, Dan [1 ]
Du, Jinlian [1 ]
Fu, Lihua [1 ]
Li, Tong [1 ]
机构
[1] Beijing Univ Technol, Fac Informat Technol, Beijing 100124, Peoples R China
[2] Beijing Machine & Equipment Inst, Beijing 100854, Peoples R China
关键词
brain-computer interface; motor imagery; deep learning; channel attention; data augmentation; NEURAL-NETWORKS; EEG;
D O I
10.1088/1741-2552/ac4852
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Objective. Motor imagery-based brain-computer interface (MI-BCI) is one of the most important BCI paradigms and can identify the target limb of subjects from the feature of MI-based Electroencephalography signals. Deep learning methods, especially lightweight neural networks, provide an efficient technique for MI decoding, but the performance of lightweight neural networks is still limited and need further improving. This paper aimed to design a novel lightweight neural network for improving the performance of multi-class MI decoding. Approach. A hybrid filter bank structure that can extract information in both time and frequency domain was proposed and combined with a novel channel attention method channel group attention (CGA) to build a lightweight neural network Filter Bank CGA Network (FB-CGANet). Accompanied with FB-CGANet, the band exchange data augmentation method was proposed to generate training data for networks with filter bank structure. Main results. The proposed method can achieve higher 4-class average accuracy (79.4%) than compared methods on the BCI Competition IV IIa dataset in the experiment on the unseen evaluation data. Also, higher average accuracy (93.5%) than compared methods can be obtained in the cross-validation experiment. Significance. This work implies the effectiveness of channel attention and filter bank structure in lightweight neural networks and provides a novel option for multi-class motor imagery classification.
引用
收藏
页数:17
相关论文
共 40 条
[1]   Performance variation in motor imagery brain-computer interface: A brief review [J].
Ahn, Minkyu ;
Jun, Sung Chan .
JOURNAL OF NEUROSCIENCE METHODS, 2015, 243 :103-110
[2]   Filter bank common spatial pattern algorithm on BCI competition IV Datasets 2a and 2b [J].
Ang, Kai Keng ;
Chin, Zheng Yang ;
Wang, Chuanchu ;
Guan, Cuntai ;
Zhang, Haihong .
FRONTIERS IN NEUROSCIENCE, 2012, 6
[3]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[4]   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
[5]   Interpretable and lightweight convolutional neural network for EEG decoding: Application to movement execution and imagination [J].
Borra, Davide ;
Fantozzi, Silvia ;
Magosso, Elisa .
NEURAL NETWORKS, 2020, 129 :55-74
[6]   HS-CNN: a CNN with hybrid convolution scale for EEG motor imagery classification [J].
Dai, Guanghai ;
Zhou, Jun ;
Huang, Jiahui ;
Wang, Ning .
JOURNAL OF NEURAL ENGINEERING, 2020, 17 (01)
[7]  
Demsar J, 2006, J MACH LEARN RES, V7, P1
[8]   Motor-Imagery-Based Brain-Computer Interface Using Signal Derivation and Aggregation Functions [J].
Fumanal-Idocin, Javier ;
Wang, Yu-Kai ;
Lin, Chin-Teng ;
Fernandez, Javier ;
Antonio Sanz, Jose ;
Bustince, Humberto .
IEEE TRANSACTIONS ON CYBERNETICS, 2022, 52 (08) :7944-7955
[9]  
Glorot X., 2010, P 13 INT C ART INT S, P249
[10]  
Graimann B, 2010, FRONT COLLECT, P1, DOI 10.1007/978-3-642-02091-9_1