SVM Based ECG classification using rhythm and morphology features, cluster analysis and multilevel noise estimation

Abstract

Background: Smartphone-based ECG devices comprise great potential in screening for arrhythmias. However, its feasibility is limited by poor signal quality leading to incorrect rhythm classification. In this study, advanced method for automatic classification of normal rhythm (N), atrial fibrillation (A), other rhythm (O), and noisy records (P) is introduced. Methods: Two-step SVM approach followed by simple threshold based rules was used for data classification. In the first step, various features were derived from separate beats to represent particular events (normal as well as pathological and artefacts) in more detail. Output of the first classifier was used to calculate global features describing entire ECG. These features were then used to train the second classification model. Both classifiers were evaluated on training set via cross-validation technique, and additionally on hidden testing set. Results: In the Phase II of challenge, total F1 score of the method is 0.81 and 0.84 within hidden challenge dataset and training set, respectively. Particular F1 scores within hidden challenge dataset are 0.90 (N), 0.81 (A), 0.72 (O), and 0.55 (P). Particular F1 scores within training set are 0.91 (N), 0.85 (A), 0.76 (O), and 0.73 (P).