RunDAE model: Running denoising autoencoder models for denoising ECG signals

Abstract

Background:The denoising autoencoder (DAE) is commonly used to denoise bio-signals such as electrocardiogram (ECG) signals through dimensional reduction. Typically, the DAE model needs to be trained using correlated input segments such as QRS-aligned segments or long ECG segments. However, using long ECG segments as an input can result in a complex deep DAE model that requires many hidden layers to achieve a low-dimensional representation, which is a major drawback. Methods:This work proposes a novel DAE model, called running DAE (RunDAE), for denoising short ECG segments without relying on the R-peak detection algorithm for alignment. The proposed RunDAE model employs a sample-by-sample processing approach, considering the correlation between consecutive, overlapped ECG segments. The performance of both the classical DAE and RunDAE models with convolutional and dense layers, respectively, is evaluated using corrupted QRS-aligned and non-aligned ECG segments with physical noise such as motion artifacts, electrode movement, baseline wander, and simulated noise such as Gaussian white noise. Results:The simulation results indicate that 1. QRS-aligned segments are preferable to non-aligned segments, 2. the RunDAE model outperforms the classical DAE model in denoising ECG signals, especially when using dense layers and QRS-aligned segments, 3. training the RunDAE models with normal and arrhythmic ECG signals enhance model’s properties/capabilities, and 4. the RunDAE is a multistage, non-causal, nonlinear adaptive filter. Conclusion:A shallow learning model, which consists of a couple of hidden layers, could achieve outstanding denoising performance using only the correlation among neighboring samples.