A Novel Method for Classification of ECG Arrhythmias Using Deep Belief Networks

In this paper, a novel approach based on deep belief networks (DBN) for electrocardiograph (ECG) arrhythmias classification is proposed. The construction process of ECG classification model consists of two steps: features learning for ECG signals and supervised fine-tuning. In order to deeply extract features from continuous ECG signals, two types of restricted Boltzmann machine (RBM) including Gaussian-Bernoulli and Bernoulli-Bernoulli are stacked to form DBN. The parameters of RBM can be learned by two training algorithms such as contrastive divergence and persistent contrastive divergence. A suitable feature representation from the raw ECG data can therefore be extracted in an unsupervised way. In order to enhance the performance of DBN, a fine-tuning process is carried out, which uses backpropagation by adding a softmax regression layer on the top of the resulting hidden representation layer to perform multiclass classification. The method is then validated by experiments on the well-known MIT-BIH arrhythmia database. Considering the real clinical application, the interpatient heartbeat dataset is divided into two sets and grouped into four classes (N, S, V, F) following the recommendations of AAMI. The experiment results show our approach achieves better performance with less feature learning time than traditional hand-designed methods on the classification of ECG arrhythmias.