A machine learning approach for the classification of cardiac arrhythmia

Rapid advancements in technology have facilitated early diagnosis of diseases in the medical sector. One of the most prevalent medical conditions that demands early diagnosis is cardiac arrhythmia. ECG signals can be used to classify and detect the type of cardiac arrhythmia. This paper introduces a novel approach to classify the ECG data into one of the sixteen types of arrhythmia using Machine Learning. The proposed method uses the UCI Machine Learning Repository [1] dataset of cardiac arrhythmia to train the system on 279 different attributes. In order to increase the accuracy, the method uses Principal Component Analysis for dimensionality reduction, Bag of Visual Words approach for clustering and compares different classification algorithms like Support Vector Machine, Random Forest, Logistic Regression and K-Nearest Neighbor algorithms, thus choosing the most accurate algorithm, Support Vector Machine.

[1]  Priya Ranjan,et al.  Comparative analysis of dimensionality reduction algorithms, case study: PCA , 2017, 2017 11th International Conference on Intelligent Systems and Control (ISCO).

[2]  Roshan Joy Martis,et al.  Machine intelligent diagnosis of ECG for arrhythmia classification using DWT, ICA and SVM techniques , 2015, 2015 Annual IEEE India Conference (INDICON).

[3]  Yaron Kinar,et al.  Using machine learning to detect problems in ECG data collection , 2011, 2011 Computing in Cardiology.

[4]  Tito Waluyo Purboyo,et al.  Stomach disorder detection through the Iris Image using Backpropagation Neural Network , 2016, 2016 International Conference on Informatics and Computing (ICIC).

[5]  Zhen Fang,et al.  An ECG arrhythmia classification and heart rate variability analysis system based on android platform , 2015, 2015 2nd International Symposium on Future Information and Communication Technologies for Ubiquitous HealthCare (Ubi-HealthTech).

[6]  Xiangyun Hu,et al.  Bag-of-Words and Object-Based Classification for Cloud Extraction From Satellite Imagery , 2015, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[7]  Hermann Ney,et al.  Bag-of-visual-words models for adult image classification and filtering , 2008, 2008 19th International Conference on Pattern Recognition.

[8]  Yedukondalu Kamatham,et al.  Performance Analysis of Artificial Neural Networks for Cardiac Arrhythmia Detection , 2016, 2016 IEEE 6th International Conference on Advanced Computing (IACC).

[9]  Shahidan M. Abdullah,et al.  Advantage and drawback of support vector machine functionality , 2014, 2014 International Conference on Computer, Communications, and Control Technology (I4CT).

[10]  Héctor Pomares,et al.  Comparison of different computational intelligent classifier to autonomously detect cardiac pathologies diagnosed by ECG , 2013, 2013 13th International Conference on Intellient Systems Design and Applications.

[11]  Yong Zhang,et al.  Study on K-means method based on Data-Mining , 2015, 2015 Chinese Automation Congress (CAC).

[12]  Arslan Shaukat,et al.  Identifying best feature subset for cardiac arrhythmia classification , 2015, 2015 Science and Information Conference (SAI).