An automated patient-specific ECG beat classification using LSTM-based recurrent neural networks

Abstract

Customized electrocardiogram analysis is inevitable in the cardiac functionality assessment of every single individual patient. In this paper, we firmly introduce a novel framework for patient-specific electrocardiogram (ECG) beat classification characterized by the long-short term memory (LSTM) based recurrent neural networks (RNNs). A novel architecture is developed by governing two LSTM models connected in parallel, and each of them operates on the single-channel ECG input. The proposed model customizes the classification process and imbibes the trait of patient-specific analysis by training the model with the first 5-min segment of heartbeats of the respective patient’s ECG record and tests the same for the rest of the ECG record. The proposed framework undergoes five stages viz; pre-processing, QRS complex detection, segmentation, feature extraction, and LSTM-based RNNs for classification. The efficiency of the proposed framework is deeply examined by the support of various metrics viz; accuracy (Acc), sensitivity (Sen), specificity (Spe), positive predictivity (Ppr), F1 score, and G score and the performance of the proposed method is compared to that of some popular state-of-the-art techniques. The proposed framework is exposed to a highly popular and reliable MIT-BIH arrhythmia database for training and testing purposes. The obtained results are found promising and yield superior classification performance in recognizing ventricular ectopic beats (VEB) and supra-ventricular ectopic beats (SVEB). In addition, the proposed framework is recognized as simple, lightweight yet efficient which in turn makes it suitable for continuous monitoring applications in clinical settings for automated ECG analysis to assist the cardiologist.