EEG artifact removal using sub-space decomposition, nonlinear dynamics, stationary wavelet transform and machine learning algorithms

Abstract

Blind source separation (BSS) methods have received a great deal of attention in electroencephalogram (EEG) artifact elimination as they are routine and standard signal processing tools to remove artifacts and reserve desired neural information. On the other hand, a classifier should follow BSS methods to automatically identify artifactual sources and remove them in the following steps. In addition, removing all detected artifactual components leads to loss of information since some desired information related to neural activity leaks to these sources. So, an approach should be employed to detect and suppress the artifacts and reserve neural activity. This study introduces a novel method based on EEG and Poincare planes in the phase space to detect artifactual components estimated by second-order blind identification (SOBI). Artifacts are detected using a mixture of well-known conventional classifiers and were removed employing stationary wavelet transform (SWT) to reserve neural information. The proposed method is a combination of signal processing techniques and machine learning algorithms, including multi-layer perceptron (MLP), K-nearest neighbor (KNN), naïve Bayes, and support vector machine (SVM) which have significant results while applying our proposed method to different scenarios. Simulated, semi-simulated, and real EEG signals are employed to evaluate the proposed method, and several evaluation criteria are calculated. We achieved acceptable results, for example, 98% average accuracy and 97% average sensitivity in artifactual EEG component detection or about 2% as mean square error in EEG reconstruction after artifact removal. Results showed that the proposed method is effective and can be used in future studies as we have considered different real-world scenarios to evaluate it.