A deep learning approach for inter-patient classification of premature ventricular contraction from electrocardiogram

Ziqiang Wang,Kun Wang,Xiaozhong Chen,Yefeng Zheng,Xian Wu
DOI: https://doi.org/10.1016/j.bspc.2024.106265
IF: 5.1
2024-04-11
Biomedical Signal Processing and Control
Abstract:Premature ventricular contractions (PVCs) are a common type of arrhythmia and can be life-threatening if deteriorated. There are several challenges remain to be resolved to the inter-patient paradigm of PVCs detection task by deep learning approaches. This research aims to overcome data imbalance, inaccurate ECG segments labeling and finding a generalized end-to-end model. In this paper, we present a deep learning network, PVCNet, to detect PVCs based on single-lead ECG. All ECG records are firstly split into fixed length segments based on the R-peak locations. To obtain more contextual information and improve the accuracy of segments labeling, we innovatively label the segment with sequence heartbeats label in the single heartbeat classification task, which was widely labeled with single heartbeat label in previous practices. Meanwhile, a loss function of redesigned weights is introduced to encourage the model to focus on detecting the specific heartbeat in the input segment. Additionally, a novel data augmentation method is conducted to alleviate the data imbalance problem. The proposed model is trained and validated on the MIT-BIH Arrhythmia Database (MITDB) with the inter-patient paradigm and tested on the St.Petersburg INCART Database (INCARTDB). It achieves a sensitivity of 89.91%, positive prediction value of 96.72%, F1-score of 93.19% and an overall accuracy of 98.50% on the testing database and outperforms existing approaches. The experimental results validate the effectiveness of the proposed approach and potentially can be applied to long-term ECG monitoring scenarios for the PVC detection task.
engineering, biomedical
What problem does this paper attempt to address?