Convolutional block attention autoencoder for denoising electrocardiograms

Wesley Chorney,Haifeng Wang,Lu He,Seunghan Lee,Lir-Wan Fan
DOI: https://doi.org/10.1016/j.bspc.2023.105242
IF: 5.1
2023-07-15
Biomedical Signal Processing and Control
Abstract:Electrocardiograms are commonly used to detect cardiovascular diseases, so it is important that they are of high quality. However, various sources of noise, such as baseline wander, muscle artifact, and electrode motion can obfuscate the signal of the heart recorded by different monitors. In this work, we propose a novel algorithm, convolutional denoising autoencoder with block attention module (CDAE-BAM), for removing noise from electrocardiograms by leveraging attention in a convolutional denoising autoencoder. We propose an attention block including both spatial and channel attention. Spatial attention captures the location of relevant features within channels in a signal, and channel attention captures the most relevant channels in a signal. We validate our proposed algorithm's performance in removing noise from the MIT-BIH Noise Stress Test Database from electrocardiogram signals in the QT Database, the Computing in Cardiology Challenge 2017 Database, and the Medical Information Mart for Intensive Care Database. We show that this method outperforms eight other state-of-the-art methods with respect to sum of squared distances, mean absolute distance, and cosine similarity.
engineering, biomedical
What problem does this paper attempt to address?