Enhancing the performance of premature ventricular contraction detection in unseen datasets through deep learning with denoise and contrast attention module

Premature ventricular contraction (PVC) is a common and harmless cardiac arrhythmia that can be asymptomatic or cause palpitations and chest pain in rare instances. However, frequent PVCs can lead to more serious arrhythmias, such as atrial fibrillation. Several PVC detection models have been proposed to enable early diagnosis of arrhythmias; however, they lack reliability and generalizability due to the variability of electrocardiograms across different settings and noise levels. Such weaknesses are known to aggravate with new data. Therefore, we present a deep learning model with a novel attention mechanism that can detect PVC accurately, even on unseen electrocardiograms with various noise levels. Our method, called the Denoise and Contrast Attention Module (DCAM), is a two-step process that denoises signals with a convolutional neural network (CNN) in the frequency domain and attends to differences. It focuses on differences in the morphologies and intervals of the remaining beats, mimicking how trained clinicians identify PVCs. Using three different encoder types, we evaluated 1D U-Net with DCAM on six external test datasets. The results showed that DCAM significantly improved the F1-score of PVC detection performance on all six external datasets and enhanced the performance of balancing both the sensitivity and precision of the models, demonstrating its robustness and generalization ability regardless of the encoder type. This demonstrates the need for a trainable denoising process before applying the attention mechanism. Our DCAM could contribute to the development of a reliable algorithm for cardiac arrhythmia detection under real clinical electrocardiograms.