Attention-Inception and Long- Short-Term Memory-Based Electroencephalography Classification for Motor Imagery Tasks in Rehabilitation

被引:81
作者
Amin, Syed Umar [1 ,2 ]
Altaheri, Hamdi [1 ,3 ]
Muhammad, Ghulam [1 ,3 ]
Abdul, Wadood [1 ,3 ]
Alsulaiman, Mansour [1 ,3 ]
机构
[1] King Saud Univ, Coll Comp & Informat Sci, Dept Comp Engn, Riyadh 11543, Saudi Arabia
[2] Prince Sultan Univ, Coll Comp & Informat Sci, Dept Comp Sci, Riyadh 11586, Saudi Arabia
[3] King Saud Univ, CCIS, Ctr Smart Robot Res, Riyadh 11543, Saudi Arabia
关键词
Attention network; convolutional neural network (CNN); deep learning (DL); EEG motor imagery (MI) decoding; SMART; NETWORKS;
D O I
10.1109/TII.2021.3132340
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In recent years, the contributions of deep learning have had a phenomenal impact on electroencephalography-based brain-computer interfaces. While the decoding accuracy of electroencephalography signals has continued to increase, the process has caused deep learning models to continuously expand in terms of size and computational resource requirements. However, due to their increased size and computational requirements, it has become difficult to embed, store, and execute deep learning models for artificial intelligence of things, cloud-based, or edge devices used in rehabilitation. Hence, this article proposes a novel deep learning-based lightweight model based on attention-inception convolutional neural network and long- short-term memory. The proposed model achieves excellent accuracy on public competition datasets while requiring few parameters and low computational time. Using the BCI competition IV 2a dataset and the high gamma dataset, the proposed model achieved 82.8% and 97.1% accuracies, respectively.
引用
收藏
页码:5412 / 5421
页数:10
相关论文
共 31 条
[11]  
Hossain M. S., ACM T MULTIM COMPUT, V15, P1
[12]   Improving consumer satisfaction in smart cities using edge computing and caching: A case study of date fruits classification [J].
Hossain, M. Shamim ;
Muhammad, Ghulam ;
Amin, Syed Umar .
FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2018, 88 :333-341
[13]  
Ingolfsson TM, 2020, IEEE SYS MAN CYBERN, P2958
[14]  
Jin Z., 2018, NEURAL COMPUT APPL, V32, P6601
[15]   DEEP-LEARNING-EMPOWERED BREAST CANCER AUXILIARY DIAGNOSIS FOR 5GB REMOTE E-HEALTH [J].
Yu, Keping ;
Tan, Liang ;
Lin, Long ;
Cheng, Xiaofan ;
Yi, Zhang ;
Sato, Takuro .
IEEE WIRELESS COMMUNICATIONS, 2021, 28 (03) :54-61
[16]  
Kumar Shiu, 2016, 2016 3rd Asia-Pacific World Congress on Computer Science and Engineering (APWC on CSE), P34, DOI 10.1109/APWC-on-CSE.2016.017
[17]   EEGNet: a compact convolutional neural network for EEG-based brain-computer interfaces [J].
Lawhern, Vernon J. ;
Solon, Amelia J. ;
Waytowich, Nicholas R. ;
Gordon, Stephen M. ;
Hung, Chou P. ;
Lance, Brent J. .
JOURNAL OF NEURAL ENGINEERING, 2018, 15 (05)
[18]   Densely Feature Fusion Based on Convolutional Neural Networks for Motor Imagery EEG Classification [J].
Li, Donglin ;
Wang, Jianhui ;
Xu, Jiacan ;
Fang, Xiaoke .
IEEE ACCESS, 2019, 7 :132720-132730
[19]   Brain-machine interfaces for rehabilitation in stroke: A review [J].
Lopez-Larraz, E. ;
Sarasola-Sanz, A. ;
Irastorza-Landa, N. ;
Birbaumer, N. ;
Ramos-Murguialday, A. .
NEUROREHABILITATION, 2018, 43 (01) :77-97
[20]   Tree-Based Deep Networks for Edge Devices [J].
Muhammad, Ghulam ;
Hossain, M. Shamim ;
Yassine, Abdulsalam .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2020, 16 (03) :2022-2028