Application of a convolutional neural network for predicting the occurrence of ventricular tachyarrhythmia using heart rate variability features

Abstract

Predicting the occurrence of ventricular tachyarrhythmia (VTA) in advance is a matter of utmost importance for saving the lives of cardiac arrhythmia patients. Machine learning algorithms have been used to predict the occurrence of imminent VTA. In this study, we used a one-dimensional convolutional neural network (1-D CNN) to extract features from heart rate variability (HRV), thereby to predict the onset of VTA. We also compared the prediction performance of our CNN with other machine leaning (ML) algorithms such as an artificial neural network (ANN), a support vector machine (SVM), and a k-nearest neighbor (KNN), which used 11 HRV features extracted using traditional methods. The proposed CNN achieved relatively higher prediction accuracy of 84.6%, while the ANN, SVM, and KNN algorithms obtained prediction accuracies of 73.5%, 67.9%, and 65.9% using 11 HRV features, respectively. Our result showed that the proposed 1-D CNN could improve VTA prediction accuracy by integrating the data cleaning, preprocessing, feature extraction, and prediction.