Motor imagery EEG classification algorithm based on improved lightweight feature fusion network

被引:31
作者
Yu, Zihang [1 ]
Chen, Wanzhong [1 ]
Zhang, Tao [1 ]
机构
[1] Jilin Univ, Coll Commun Engn, Changchun 130025, Peoples R China
基金
中国博士后科学基金;
关键词
motor imagery; Data augmentation; Attention mechanism; Tensor decomposition; Deep learning; NEURAL-NETWORK; SIGNALS;
D O I
10.1016/j.bspc.2022.103618
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
When deep learning techniques are introduced for Motor Imagery(MI) EEG signal classification, a multitude of state-of-the-art models, cannot be trained effectively because of the relatively small datasets. Proposing a model specialized for MI EEG signals classification plays a prominent role in promoting the combination of deep learning technology and MI EEG signal classification. In this paper, a novel Lightweight Feature Fusion Network (LFANN) based on an improved attention mechanism and tensor decomposition approach has been introduced. The proposed algorithm has been evaluated on a public benchmark dataset from BCI Competition IV, and the original dataset has been augmented with Enhance-Super-Resolution Generative Adversarial Network(ESRGAN). The experimental results demonstrate that the average accuracy of 91.58% and the average Kappa value of 0.881 can be achieved through the proposed algorithm. Furthermore, the compressed LAFFN, whose parameters have been compressed nearly ten times, creates no significant difference in performance compared to LAFFN. The investigation carried out through this experiment has provided novel insights into the classification research for MI EEG signals.
引用
收藏
页数:9
相关论文
共 49 条
[1]   Multi-channel EEG recordings during a sustained-attention driving task [J].
Cao, Zehong ;
Chuang, Chun-Hsiang ;
King, Jung-Kai ;
Lin, Chin-Teng .
SCIENTIFIC DATA, 2019, 6 (1) :19
[2]  
Du Xiaoyan, 2008, Sheng Wu Yi Xue Gong Cheng Xue Za Zhi, V25, P464
[3]   Towards correlation-based time window selection method for motor imagery BCIs [J].
Feng, Jiankui ;
Yin, Erwei ;
Jin, Jing ;
Saab, Rami ;
Daly, Ian ;
Wang, Xingyu ;
Hu, Dewen ;
Cichocki, Andrzej .
NEURAL NETWORKS, 2018, 102 :87-95
[4]   Simulating quantized inference on convolutional neural networks [J].
Finotti, Vitor ;
Albertini, Bruno .
COMPUTERS & ELECTRICAL ENGINEERING, 2021, 95
[5]   An automatic subject specific channel selection method for enhancing motor imagery classification in EEG-BCI using correlation [J].
Gaur, Pramod ;
McCreadie, Karl ;
Pachori, Ram Bilas ;
Wang, Hui ;
Prasad, Girijesh .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 68
[6]   A classification method for EEG motor imagery signals based on parallel convolutional neural network [J].
Han, Yuexing ;
Wang, Bing ;
Luo, Jie ;
Li, Long ;
Li, Xiaolong .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 71
[7]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[8]   The neural network of motor imagery: An ALE meta-analysis [J].
Hetu, Sebastien ;
Gregoire, Mathieu ;
Saimpont, Arnaud ;
Coll, Michel-Pierre ;
Eugene, Fanny ;
Michon, Pierre-Emmanuel ;
Jackson, Philip L. .
NEUROSCIENCE AND BIOBEHAVIORAL REVIEWS, 2013, 37 (05) :930-949
[9]   Attention Recognition in EEG-Based Affective Learning Research Using CFS plus KNN Algorithm [J].
Hu, Bin ;
Li, Xiaowei ;
Sun, Shuting ;
Ratcliffe, Martyn .
IEEE-ACM TRANSACTIONS ON COMPUTATIONAL BIOLOGY AND BIOINFORMATICS, 2018, 15 (01) :38-45
[10]   Independent component analysis of short-time Fourier transforms for spontaneous EEG/MEG analysis [J].
Hyvarinen, Aapo ;
Ramkumar, Pavan ;
Parkkonen, Lauri ;
Hari, Riitta .
NEUROIMAGE, 2010, 49 (01) :257-271