Robust deep learning pipeline for PVC beats localization

Technol Health Care. 2021;29(S1):475-486. doi: 10.3233/THC-218045.

Abstract

Background: Premature ventricular contraction (PVC) is among the most frequently occurring types of arrhythmias. Existing approaches for automated PVC identification suffer from a range of disadvantages related to hand-crafted features and benchmarking on datasets with a tiny sample of PVC beats.

Objective: The main objective is to address the drawbacks described above in the proposed framework, which takes a raw ECG signal as an input and localizes R peaks of the PVC beats.

Methods: Our method consists of two neural networks. First, an encoder-decoder architecture trained on PVC-rich dataset localizes the R peak of both Normal and anomalous heartbeats. Provided R peaks positions, our CardioIncNet model does the delineation of healthy versus PVC beats.

Results: We have performed an extensive evaluation of our pipeline with both single- and cross-dataset paradigms on three public datasets. Our approach results in over 0.99 and 0.979 F1-measure on both single- and cross-dataset paradigms for R peaks localization task and above 0.96 and 0.85 F1 score for the PVC beats classification task.

Conclusions: We have shown a method that provides robust performance beyond the beats of Normal nature and clearly outperforms classical algorithms both in the case of a single and cross-dataset evaluation. We provide a Github1 repository for the reproduction of the results.

Keywords: ECG classification; ECG segmentation; Electrocardiography; PVC identification.

MeSH terms

  • Algorithms
  • Deep Learning*
  • Electrocardiography
  • Humans
  • Neural Networks, Computer
  • Signal Processing, Computer-Assisted
  • Ventricular Premature Complexes* / diagnosis