Design of an artificial neural network and feature extraction to identify arrhythmias from ECG

Abstract

This paper presents a design of an artificial neural network (ANN) and feature extraction methods to identify two types of arrhythmias in datasets obtained through electrocardiography (ECG) signals, namely arrhythmia dataset (AD) and supraventricular arrhythmia dataset (SAD). No special ANN toolkit was used; instead, each neuron and necessary calculus were modeled and individually programmed. Thus, four temporal-based features are used: heart rate (HR), R-peaks root mean square (R-RMS), RR-peaks variance (RR-VAR), and QSR-complex standard deviation (QSR-SD). The network architecture presents four neurons in the input layer, eight in hidden layer and an output layer with two neurons. The proposed classification method uses the MIT-BIH Dataset (Massachusetts Institute of Technology-Beth Israel Hospital) for training, validation and execution or test phases. Preliminary results show the high efficiency of the proposed ANN design and its classification method, reaching accuracies between 98.76% and 98.91%, when in the identification of NSRD and arrhythmic ECG; and accuracies of 86.37% (AD) and 76.35% (SAD), when analyzing only classifications between both arrhythmias.