Arrhythmia classification detection based on multiple electrocardiograms databases

Abstract

According to the World Health Organization, cardiovascular diseases are the leading cause of deaths globally. Electrocardiogram (ECG) is a non-invasive approach for detecting heart diseases and reducing the risk of heart disease-related death. However, there are limited numbers of ECG samples and imbalance distribution for existing ECG databases. It is difficult to train practical and efficient neural networks. Based on the analysis and research of many existing ECG databases, this paper conduct an in-depth study on three fine-labeled ECG databases, to extract heartbeats, unify the sampling frequency, and propose a self-processing method of heartbeats, and finally form a unified ECG arrhythmia classification database, noted as Hercules-3. It is separated into training sets (80%) and testing sets (the remaining 20%). In order to verify its capabilities, we have trained a 16-classification fully connected neural network based on Hercules-3 and it achieves an accuracy rate of up to 98.67%. Compared with other data processing, our proposed method improves classification recall by at least 6%, classification accuracy by at least 4%, and F1-score by at least 7%.