Improving EEG Signal Classification Accuracy Using Wasserstein Generative Adversarial Networks

Joshua Park,Priyanshu Mahey,Ore Adeniyi
2024-02-05
Abstract:Electroencephalography (EEG) plays a vital role in recording brain activities and is integral to the development of brain-computer interface (BCI) technologies. However, the limited availability and high variability of EEG signals present substantial challenges in creating reliable BCIs. To address this issue, we propose a practical solution drawing on the latest developments in deep learning and Wasserstein Generative Adversarial Network (WGAN). The WGAN was trained on the BCI2000 dataset, consisting of around 1500 EEG recordings and 64 channels from 45 individuals. The generated EEG signals were evaluated via three classifiers yielding improved average accuracies. The quality of generated signals measured using Frechet Inception Distance (FID) yielded scores of 1.345 and 11.565 for eyes-open and closed respectively. Even without a spectral or spatial loss term, our WGAN model was able to emulate the spectral and spatial properties of the EEG training data. The WGAN-generated data mirrored the dominant alpha activity during closed-eye resting and high delta waves in the training data in its topographic map and power spectral density (PSD) plot. Our research testifies to the potential of WGANs in addressing the limited EEG data issue for BCI development by enhancing a small dataset to improve classifier generalizability.
Signal Processing,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?
The paper aims to address the issues of limited data volume and high variability in electroencephalogram (EEG) signals within brain-computer interface (BCI) technology. Specifically, the authors propose a method based on the Wasserstein Generative Adversarial Network (WGAN) to generate realistic EEG signals, thereby enhancing existing small-scale datasets and improving the generalization ability of classifiers. By training the WGAN on the BCI2000 dataset and evaluating it using three different classifiers (Fully Connected Neural Network FNN, Convolutional Neural Network CNN, and Recurrent Neural Network RNN), the results show a significant improvement in classification accuracy. Additionally, the quality of the generated signals was validated through the Fréchet Inception Distance (FID) score and Power Spectral Density (PSD) plots, demonstrating a high similarity to real EEG signals. This study showcases the potential of WGAN in expanding and diversifying EEG datasets, thereby enhancing the performance and efficiency of BCI systems.