A Novel Multimodule Neural Network for EEG Denoising

Abstract

In this paper, a novel multi-module neural network (MMNN) is proposed to remove ocular artifacts (OAs) and myogenic artifacts (MAs) from noisy single-channel electroencephalogram (EEG) signals. This network is a based on deep learning (DL) architecture consisting of multiple denoising modules connected in parallel. Each denoising module is built using one-dimensional convolutions (Conv1Ds) and fully connected (FC) layers, and it estimates not only clean EEG signals but also artifacts. The proposed MMNN has two main advantages. Frist, the multiple denoising modules can purify noisy input EEG signals by continuously removing artifacts in the forward propagation. Second, the parallel architecture allows the parameters of each denoising module to be updated concurrently in the backpropagation, thereby improving the learning capacity of neural networks. We tested the network denoising performance using a recent public database, namely, EEGdenoiseNet. The results revealed that the proposed network reduced the temporal relative root mean square error (T-RRMSE) and spectral relative root mean square error (S-RRMSE) by at least 6% and enhanced the correlation coefficient (CC) by at least 3% over the state-of-the-art approaches. These significant performance improvements were confirmed by observing the deviation distribution between the denoised and clean signals. Furthermore, the proposed network achieved a similar performance efficiency with only 60% of the training data compared to the existing DL models. Our model can be found at: https://github.com/Zhangzhenkut/Multi-Module-Neural-Network-for-EEG-Denoising

Keywords

• EEG denoising
• multi-module neural network (MMNN)
• deep learning (DL)