ECG Classification using Machine Learning

Abstract

Recently, the obvious increasing number of cardiovascular disease, the automatic classification research of Electrocardiogram signals (ECG) has been playing a important part in the clinical diagnosis of cardiovascular disease. Convolution neural network (CNN) based method is proposed to classify ECG signals. The proposed CNN model consists of five layers in addition to the input layer and the output layer, i.e., two convolution layers, two down sampling layers and one full connection layer, extracting the effective features from the original data and classifying the features using wavelet .The classification of ARR (Arrhythmia), CHF (Congestive Heart Failure), and NSR (Normal Sinus Rhythm) signals. The experimental results contains on ARR signals from the MIT-BIH arrhythmia,CHF signals from the BIDMC Congestive Heart Failure and NSR signals from the MIT-BIH Normal Sinus Rhythm Databases show that the proposed method achieves a promising classification accuracy of 90.63%, significantly outperforming several typical ECG classification methods.