1. AAC-WGAN: A Novel Attention-Enhanced GAN Framework for SSVEP Augmentation and Classification
- Author
-
Junjie Liu, Jun Xie, Qing Tao, Huanqing Zhang, Hu Wang, and Bo Hu
- Subjects
Brain-computer interface (BCI) ,generative adversarial network (GAN) ,data augmentation ,global attention mechanism ,auxiliary classifier ,steady-state visual evoked potential (SSVEP) ,Electrical engineering. Electronics. Nuclear engineering ,TK1-9971 - Abstract
The rapid advancement of brain-computer interface (BCI) technology has created interest in steady-state visual evoked potential (SSVEP)-based BCIs, which are valued for their high information transfer rates and ability to manage multiple targets. Nonetheless, the efficacy of SSVEP decoding is often constrained by the volume and duration of user calibration data, limiting its practical application. Generative adversarial networks (GANs) have shown promise in synthesizing SSVEP electroencephalogram (EEG) data. However, they face challenges, such as low signal-to-noise ratio and capturing both temporal and spatial features. To address the need for high-quality data generation, the Attention-Aided Classifier Wasserstein GAN (AAC-WGAN) is proposed, which is a novel GAN model that combines an Attention Mechanism and an Auxiliary Classifier to improve data quality and classification performance. Our experiments on the Direction and Dial datasets reveal that our model performs obviously better, particularly with a training set ratio of 25% synthetic data and 75% real data. It achieves a classification accuracy of 91.65% on the Direction dataset, which is a significant improvement over the baseline accuracy of 84.32% (p =0.008) and outperforms other comparable models (p
- Published
- 2024
- Full Text
- View/download PDF