Early classification and recognition algorithm for sudden cardiac arrest based on limited electrocardiogram data trained with a two-stages convolutional neural network

Abstract

Sudden cardiac arrest (SCA) is a lethal cardiac arrhythmia that poses a serious threat to human life and health. However, clinical records of sudden cardiac death (SCD) electrocardiogram (ECG) data are extremely limited. This paper proposes an early prediction and classification algorithm for SCA based on deep transfer learning. With limited ECG data, it extracts heart rate variability features before the onset of SCA and utilizes a lightweight convolutional neural network model for pre-training and fine-tuning in two stages of deep transfer learning. This achieves early classification, recognition and prediction of high-risk ECG signals for SCA by neural network models. Based on 16 788 30-second heart rate feature segments from 20 SCA patients and 18 sinus rhythm patients in the international publicly available ECG database, the algorithm performance evaluation through ten-fold cross-validation shows that the average accuracy (Acc), sensitivity (Sen), and specificity (Spe) for predicting the onset of SCA in the 30 minutes prior to the event are 91.79%, 87.00%, and 96.63%, respectively. The average estimation accuracy for different patients reaches 96.58%. Compared to traditional machine learning algorithms reported in existing literatures, the method proposed in this paper helps address the requirement of large training datasets for deep learning models and enables early and accurate detection and identification of high-risk ECG signs before the onset of SCA.