DCAE-SR: Design of a Denoising Convolutional Autoencoder for reconstructing Electrocardiograms signals at Super Resolution

Ugo Lomoio,Pierangelo Veltri,Pietro Hiram Guzzi,Pietro Liò
DOI: https://doi.org/10.1101/2024.04.08.24305091
2024-04-10
Abstract:Electrocardiogram (ECG) signals play a pivotal role in cardiovascular diagnostics, providing essential information on the electrical activity of the heart. However, the inherent noise and limited resolution in ECG recordings can hinder accurate interpretation and diagnosis. In this paper, we propose a novel model for ECG super resolution (SR) that uses a DNAE to enhance temporal and frequency information inside ECG signals. Our approach addresses the limitations of traditional ECG signal processing techniques. Our model takes in input 5-second length ECG windows sampled at 50 Hz (very low resolution) and it is able to reconstruct a denoised super-resolution signal with an x10 upsampling rate (sampled at 500 Hz). We trained the proposed DCAE-SR on public available myocardial infraction ECG signals. Our method demonstrates superior performance in reconstructing high-resolution ECG signals from very low-resolution signals with a sampling rate of 50 Hz. We compared our results with the current deep-learning literature approaches for ECG super-resolution and some non-deep learning reproducible methods that can perform both super-resolution and de-noising. We obtained current state-of-the-art performances in super-resolution of very low resolution ECG signals frequently corrupted by ECG artifacts. We were able to obtain a signal-to-noise ratio of 12.20 dB (outperforms previous 4.68 dB), mean squared error of 0.0044 (outperforms previous 0.0154) and root mean squared error of 4.86% (outperforms previous 12.40%). In conclusion, our DCAE-SR model offers a robust (to artefact presence), versatile and explainable solution to enhance the quality of ECG signals. This advancement holds promise in advancing the field of cardiovascular diagnostics, paving the way for improved patient care and high-quality clinical decisions.
Health Informatics
What problem does this paper attempt to address?
This paper attempts to solve the problems of noise and low resolution in electrocardiogram (ECG) signals in cardiovascular diagnosis. Specifically, the authors propose a new model based on the denoising convolutional auto - encoder (DCAE) - DCAE - SR, which is used to reconstruct high - resolution (500 Hz) signals from low - resolution (50 Hz) electrocardiogram signals and simultaneously remove noise and artifacts. This method aims to improve the quality of electrocardiogram signals, thereby improving the diagnosis and monitoring of heart diseases. ### Main contributions of the paper: 1. **Solve the limitations of current electrocardiogram signal acquisition**: such as noise and very low sampling rates. 2. **For the first time, deal with noise and artifacts in very low - resolution signals with a 50 Hz sampling rate**. 3. **Expand the DCAE architecture to perform electrocardiogram denoising and super - resolution tasks**. 4. **Apply model interpretability to clarify how the model identifies and utilizes relevant features or patterns in LR inputs to generate denoised SR outputs**. 5. **Demonstrate robust behavior and performance in noisy and clean low - resolution electrocardiogram signals**. ### Method overview: - **Dataset**: Use the PTB - XL dataset, which contains 21,799 clinically 12 - lead 10 - second multi - labeled electrocardiogram signals. - **Pre - processing**: Perform high - pass filtering and band - pass filtering on the signals, divide them into 5 - second windows, and down - sample the 100 Hz low - resolution signals to 50 Hz. - **Model architecture**: DCAE - SR includes an encoder and two decoders, where one decoder is used to reconstruct the signal at the original frequency and the other decoder is used to generate a high - resolution signal. - **Training**: Use multi - loss function optimization, including reconstruction error and super - resolution error. - **Evaluation metrics**: Use metrics such as MSE, RMSE, SSIM, SNR, and PSNR to evaluate model performance. ### Experimental results: - **Quantitative results**: On multiple diagnostic categories, the DCAE - SR model shows performance superior to existing methods in super - resolution and denoising tasks. For example, the SNR reaches 12.20 dB, the MSE is 0.0044, and the RMSE is 4.86%. - **Qualitative results**: By visualizing the reconstructed high - resolution signals, the effectiveness of the model in removing noise and artifacts is demonstrated. ### Conclusion: The DCAE - SR model provides a robust, flexible, and interpretable solution that can enhance the quality of electrocardiogram signals. This progress is expected to promote the development of the cardiovascular diagnosis field and provide high - quality support for patient care and clinical decision - making.