Auto-Denoising for EEG Signals Using Generative Adversarial Network

Abstract

The brain–computer interface (BCI) has many applications in various fields. In EEG-based research, an essential step is signal denoising. In this paper, a generative adversarial network (GAN)-based denoising method is proposed to denoise the multichannel EEG signal automatically. A new loss function is defined to ensure that the filtered signal can retain as much effective original information and energy as possible. This model can imitate and integrate artificial denoising methods, which reduces processing time; hence it can be used for a large amount of data processing. Compared to other neural network denoising models, the proposed model has one more discriminator, which always judges whether the noise is filtered out. The generator is constantly changing the denoising way. To ensure the GAN model generates EEG signals stably, a new normalization method called sample entropy threshold and energy threshold-based (SETET) normalization is proposed to check the abnormal signals and limit the range of EEG signals. After the denoising system is established, although the denoising model uses the different subjects’ data for training, it can still apply to the new subjects’ data denoising. The experiments discussed in this paper employ the HaLT public dataset. Correlation and root mean square error (RMSE) are used as evaluation criteria. Results reveal that the proposed automatic GAN denoising network achieves the same performance as the manual hybrid artificial denoising method. Moreover, the GAN network makes the denoising process automatic, representing a significant reduction in time.

A Comprehensive Survey of EEG Preprocessing Methods for Cognitive Load Assessment

“…Recently, it has been effectively employed for denoising [72]. It is less time-consuming and can handle much data [73]. Moreover, it is flexible once trained and can cope with non-linear and non-stationary artifacts [68].…”

Anomaly Detection in Medical Time Series with Generative Adversarial Networks: A Selective Review

“…Super-resolution and noise reduction are types of image-to-image translation that involve converting low-resolution images into high-resolution images by imputing data. This is especially interesting in cases where the imaging modality is intrinsically low-resolution, such as positron emission tomography (PET) [94][95][96][97][98][99].…”

Motor Imagery Multi-Tasks Classification for BCIs Using the NVIDIA Jetson TX2 Board and the EEGNet Network

“…Their work achieved an average accuracy of 90.2 ± 4.34% with the EEGNet network converting EEG signals from C3, Cz, and C4 channels into spectrum images by using the variational mode decomposition (VMD) and the short-time Fourier transform (STFT). Likewise, a generative adversarial network (GAN) was proposed by An et al [43] to denoise MI-EEG signals using the same dataset. Lately, the EEGNet network has been implemented to classify MI-EEG signal-based BCI utilizing HaLT's benchmark [44].…”