Automatic Eyeblink and Muscular Artifact Detection and Removal From EEG Signals Using k-Nearest Neighbor Classifier and Long Short-Term Memory Networks

Electroencephalogram (EEG) is often corrupted with artifacts originating from sources such as eyes and muscles. Hybrid artifact removal methods often require human intervention for the adjustment of different parameters. We propose a robust method that can automatically detect and remove eyeblink and muscular artifacts from EEG using a k-nearest neighbor (kNN) classifier and a long short-term memory (LSTM) network. Our method adopts a sliding window of 0.5 s to detect and remove the artifacts from EEG. Features, such as the variance, peak-to-peak amplitude, and average rectified value, are calculated for each EEG segment to identify corrupted segments using the kNN classifier. The kNN classifier detects the presence of artifacts, after which the corresponding EEG window is forwarded to the LSTM network for artifact removal. The LSTM network is trained with the corrupted segments of 0.5 s as input and clean segments of 0.5 s as output. Our method achieved an accuracy of 97.4% in identifying corrupted EEG segments and an average correlation coefficient, structural similarity, signal-to-artifact ratio, and normalized mean squared error of 0.69, 0.76, 1.52 dB, and 0.0013, respectively, in cleaning the EEG. Our results outperformed other hybrid methods reported in the literature based on a combination of ensemble empirical mode decomposition and canonical correlation analysis, a combination of independent component analysis and wavelet decomposition, and tensor decomposition. The mean absolute error of our method is also better in comparison to other methods. Our method can be applied to single and multiple channels and does not require any tuning of parameters.

Automated summary

We proposed an automated method for detecting and removing eyeblink and muscular artifacts from EEG using a k-nearest neighbor classifier and a long short-term memory network. Our method achieved high accuracy and performance in identifying corrupted EEG segments and cleaning the EEG.