Robust Heartbeat Classification for Wearable Single-Lead ECG via Extreme Gradient Boosting

Abstract

Wearable electrocardiogram (ECG) monitoring devices have enabled everyday ECG collection in our daily lives. However, the condition of ECG signal acquisition using wearable devices varies and wearable ECG signals could be interfered with by severe noises, resulting in great challenges of computer-aided automated ECG analysis, especially for single-lead ECG signals without spare channels as references. There remains room for improvement of the beat-level single-lead ECG diagnosis regarding accuracy and efficiency. In this paper, we propose new morphological features of heartbeats for an extreme gradient boosting-based beat-level ECG analysis method to carry out the five-class heartbeat classification according to the Association for the Advancement of Medical Instrumentation standard. The MIT-BIH Arrhythmia Database (MITDB) and a self-collected wearable single-lead ECG dataset are used for performance evaluation in the static and wearable ECG monitoring conditions, respectively. The results show that our method outperforms other state-of-the-art models with an accuracy of 99.14% on the MITDB and maintains robustness with an accuracy of 98.68% in the wearable single-lead ECG analysis.