Segment Origin Prediction: A Self-supervised Learning Method for Electrocardiogram Arrhythmia Classification

Annu Int Conf IEEE Eng Med Biol Soc. 2021 Nov:2021:1132-1135. doi: 10.1109/EMBC46164.2021.9630616.

Abstract

The automatic arrhythmia classification system has made a significant contribution to reducing the mortality rate of cardiovascular diseases. Although the current deep-learning-based models have achieved ideal effects in arrhythmia classification, their performance still needs to be further improved due to the small scale of the dataset. In this paper, we propose a novel self-supervised pre-training method called Segment Origin Prediction (SOP) to improve the model's arrhythmia classification performance. We design a data reorganization module, which allows the model to learn ECG features by predicting whether two segments are from the same original signal without using annotations. Further, by adding a feed-forward layer to the pre-training stage, the model can achieve better performance when using labeled data for arrhythmia classification in the downstream stage. We apply the proposed SOP method to six representative models and evaluate the performances on the PhysioNet Challenge 2017 dataset. After using the SOP pre-training method, all baseline models gain significant improvement. The experimental results verify the effectiveness of the proposed SOP method.

MeSH terms

  • Arrhythmias, Cardiac / diagnosis
  • Cardiovascular Diseases*
  • Electrocardiography
  • Humans
  • Neural Networks, Computer*
  • Supervised Machine Learning