EEGGAN-Net: enhancing EEG signal classification through data augmentation

• Published in: Frontiers in human neuroscience Vol. 18; p. 1430086
• Main Authors: Song, Jiuxiang, Zhai, Qiang, Wang, Chuang, Liu, Jizhong
• Format: Journal Article
• Language: English
• Published: Switzerland Frontiers Media S.A 21.06.2024
• Subjects: brain-computer interface; Conditional Generative Adversarial Network; cropped training; electroencephalography; Neuroscience; Squeeze-and-Excitation attention; electroencephalography; Conditional Generative Adversarial Network; Squeeze-and-Excitation attention; brain-computer interface; cropped training
• ISSN: 1662-5161; 1662-5161
• DOI: 10.3389/fnhum.2024.1430086

Emerging brain-computer interface (BCI) technology holds promising potential to enhance the quality of life for individuals with disabilities. Nevertheless, the constrained accuracy of electroencephalography (EEG) signal classification poses numerous hurdles in real-world applications. In response to this predicament, we introduce a novel EEG signal classification model termed EEGGAN-Net, leveraging a data augmentation framework. By incorporating Conditional Generative Adversarial Network (CGAN) data augmentation, a cropped training strategy and a Squeeze-and-Excitation (SE) attention mechanism, EEGGAN-Net adeptly assimilates crucial features from the data, consequently enhancing classification efficacy across diverse BCI tasks. The EEGGAN-Net model exhibits notable performance metrics on the BCI Competition IV-2a and IV-2b datasets. Specifically, it achieves a classification accuracy of 81.3% with a kappa value of 0.751 on the IV-2a dataset, and a classification accuracy of 90.3% with a kappa value of 0.79 on the IV-2b dataset. Remarkably, these results surpass those of four other CNN-based decoding models. In conclusion, the amalgamation of data augmentation and attention mechanisms proves instrumental in acquiring generalized features from EEG signals, ultimately elevating the overall proficiency of EEG signal classification.