The quality of electroencephalogram (EEG) signals directly impacts the performance of brain-computer interface (BCI) tasks. Many methods have been proposed to eliminate noise from EEG signals, but most of these methods focus solely on signal denoising itself, disregarding the impact on subsequent tasks, which deviates from the original intention of EEG denoising. The main objective of this study is to optimize EEG denoising models with a purpose of improving the performance of BCI tasks.
Approach.
To this end, we proposed an innovative Task-Oriented EEG Denoising Generative Adversarial Network (TOED-GAN) method. This network utilizes the generator of GAN to decompose and reconstruct clean signals from the raw EEG signals, and the discriminator to learn to distinguish the generated signals from the true clean signals, resulting in a remarkable increase of the signal-to-noise ratio (SNR) by simultaneously enhancing task-related components and removing task-irrelevant noise from the original contaminated signals.
Main results.
We evaluated the performance of the model on a public dataset and a self-collected dataset respectively, with canonical correlation analysis (CCA) classification tasks of the steady-state visual evoked potential (SSVEP) based BCI. Experimental results demonstrate that TOED-GAN exhibits excellent performance in removing EEG noise and improving performance for SSVEP-BCI, with accuracy improvement rates reaching 18.47% and 21.33% in contrast to the baseline methods of convolutional neural networks, respectively
Significance.
This work proves that the proposed TOED-GAN, as an EEG denoising method tailored for SSVEP tasks, contributes to enhancing the performance of BCIs in practical application scenarios.