ECG signal de-noising based on deep learning auto encoder and discrete wavelet transform

Abstract

ECG is very important tool for diagnosis of heart disease, this signal is suffered from different types of noises such as baseline wander (BW), muscle artifact (MA) and electrode motion (EM) , which lead to wrong interpretation. In order to prevent or reduce the effect of these noises, different approaches have been applied to enhance the ECG signal. In this paper, we have proposed a new method for ECG signal de-noising based on deep learning Auto encoder (DL-DAE) and wavelet transform named as (WT-DAE). The proposed system (WT-DAE) is constructed from two stages, in the first stage, the wavelet transform is used to isolate the most significant coefficient of the signal (approximation sub-band) from de-tails coefficients (details sub-band). The details coefficients is fed to new proposed threshold method , which is used to evaluate the threshold value according to the feature of ECG signal, this threshold value is used to threshold the detail coefficients, in order to remove the details noise that is contained as high frequencly component , then invers wavelet transform is used to reconstruct the signal . Different wavelet filters and threshold functions are applied in this stage. The second stage of signal de-noising is performed by using DAE method, which is designed for reconstruct the de-noised sig-nal. The proposed DAE model is constructed from 14 layers of convolutional, relu and max_ pooling layer with different parameters. We perform training and testing the model with MIT-BIH ECG database and the performance of the pro-posed system is evaluated by terms of MSE, RMSE, PRD and PSNR. The experimental results are compared with other approaches and show that, the proposed system demonstrated the superiority for de-noising ECG signal.Â