Toward calibration-free motor imagery brain-computer interfaces: a VGG-based convolutional neural network and WGAN approach

被引:2
作者
Habashi, A. G. [1 ]
Azab, Ahmed M. [2 ]
Eldawlatly, Seif [1 ,3 ]
Aly, Gamal M. [1 ]
机构
[1] Ain Shams Univ, Fac Engn, Comp & Syst Engn Dept, Cairo, Egypt
[2] Tech Res Ctr, Biomed Engn Dept, Cairo, Egypt
[3] Amer Univ Cairo, Comp Sci & Engn Dept, Cairo, Egypt
关键词
EEG; GAN; motor imagery; short time fourier transform (STFT); BCI; CLASSIFICATION;
D O I
10.1088/1741-2552/ad6598
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Objective. Motor imagery (MI) represents one major paradigm of Brain-computer interfaces (BCIs) in which users rely on their electroencephalogram (EEG) signals to control the movement of objects. However, due to the inter-subject variability, MI BCIs require recording subject-dependent data to train machine learning classifiers that are used to identify the intended motor action. This represents a challenge in developing MI BCIs as it complicates its calibration and hinders the wide adoption of such a technology. Approach. This study focuses on enhancing cross-subject (CS) MI EEG classification using EEG spectrum images. The proposed calibration-free approach employs deep learning techniques for MI classification and Wasserstein Generative Adversarial Networks (WGAN) for data augmentation. The proposed WGAN generates synthetic spectrum images from the recorded MI-EEG to expand the training dataset; aiming to enhance the classifier's performance. The proposed approach eliminates the need for any calibration data from the target subject, making it more suitable for real-world applications. Main results. To assess the robustness and efficacy of the proposed framework, we utilized the BCI competition IV-2B, IV-2 A, and IV-1 benchmark datasets, employing leave one-subject out validation. Our results demonstrate that using the proposed modified VGG-CNN classifier in addition to WGAN-generated data for augmentation leads to an enhancement in CS accuracy outperforming state-of-the-art methods. Significance. This approach could represent one step forward towards developing calibration-free BCI systems and hence broaden their applications.
引用
收藏
页数:15
相关论文
共 47 条
[1]   A comprehensive review of EEG-based brain-computer interface paradigms [J].
Abiri, Reza ;
Borhani, Soheil ;
Sellers, Eric W. ;
Jiang, Yang ;
Zhao, Xiaopeng .
JOURNAL OF NEURAL ENGINEERING, 2019, 16 (01)
[2]   Deep Learning for EEG motor imagery classification based on multi-layer CNNs feature fusion [J].
Amin, Syed Umar ;
Alsulaiman, Mansour ;
Muhammad, Ghulam ;
Mekhtiche, Mohamed Amine ;
Hossain, M. Shamim .
FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2019, 101 :542-554
[3]   Dual Attention Relation Network With Fine-Tuning for Few-Shot EEG Motor Imagery Classification [J].
An, Sion ;
Kim, Soopil ;
Chikontwe, Philip ;
Park, Sang Hyun .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (11) :15479-15493
[4]  
Arjovsky M, 2017, PR MACH LEARN RES, V70
[5]  
Azab AM, 2018, IET CONTR ROBOT SENS, V114, P81, DOI 10.1049/PBCE114E_ch5
[6]   Electro-encephalography [J].
Biasiucci, Andrea ;
Franceschiello, Benedetta ;
Murray, Micah M. .
CURRENT BIOLOGY, 2019, 29 (03) :R80-R85
[7]  
Blankertz B., 2007, BCI competition IV Set 1
[8]  
Brunner C., 2008, Institute for Knowledge Discovery (Laboratory of Brain-Computer Interfaces)
[9]   Classification of Motor Imagery Electroencephalography Signals Based on Image Processing Method [J].
Chen, Zhongye ;
Wang, Yijun ;
Song, Zhongyan .
SENSORS, 2021, 21 (14)
[10]   An Ensemble CNN for Subject-Independent Classification of Motor Imagery-based EEG [J].
Dolzhikova, Irina ;
Abibullaev, Berdakh ;
Sameni, Reza ;
Zollanvari, Amin .
2021 43RD ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE & BIOLOGY SOCIETY (EMBC), 2021, :319-324