Real-time Automatic Arrhythmia Detection System based on Extreme Gradient Boosting and Neural Network Algorithm

Abstract

Arrhythmia and other diseases are puzzling more and more people. Accurate detection is the key to realizing intelligent diagnosis of electrocardiogram(ECG) monitoring systems. It can prevent heart disease and effectively reduce mortality. An efficient and accurate arrhythmia detection method is urgent. In this work, a real-time automatic arrhythmia detection technology based on extreme gradient boosting (XGboost) and convolutional neural network (CNN) algorithm were developed. First, ECG signals in the MIT-BIH Arrhythmia database are preprocessed: 1) EMG interference filtering; 2) Power frequency interference suppression; 3) Baseline drift correction. Secondly, We use the cyclic singular spectrum (CISSA) algorithm to decompose the ECG after pretreatment. From the original ECG and the 7 simple signals obtained from decomposition, 23 features about the time domain, frequency domain, nonlinear dynamics and statistics are extracted respectively. Finally, XGboost and CNN algorithms are used to build a classification model, and the extracted features are classified, trained and recognized to achieve automatic detection of arrhythmia. The experimental results show that XGboost and CNN algorithms can automatically detect 98.40%, 95.65% and 97.60%, 95.12% of Category 2 and Category 4 arrhythmias, respectively.