A novel method for ECG signal classification via one-dimensional convolutional neural network
Xuan Hua,Jungang Han,Chen Zhao,Haipeng Tang,Zhuo He,Qinghui Chen,Shaojie Tang,Jinshan Tang,Weihua Zhou
DOI: https://doi.org/10.1007/s00530-020-00713-1
IF: 3.9
2020-11-11
Multimedia Systems
Abstract:This paper develops an end-to-end ECG signal classification algorithm based on a novel segmentation strategy and 1D Convolutional Neural Networks (CNN) to aid the classification of ECG signals and alleviate the workload of physicians. The ECG segmentation strategy named R-R-R strategy (i.e., retaining ECG data between the R peaks just before and after the current R peak) is used for segmenting the original ECG data into segments to train and test the 1D CNN models. The novel strategy mimics physicians in scanning ECG to a greater extent, and maximizes the inherent information of ECG segments for diagnosis. The performance of the proposed end to end ECG signal classification algorithm was verified with the ECG signals from 48 records in the MIT-BIH arrhythmia database. When the heartbeat types were divided into the five classes recommended by clinicians, i.e., normal beat, left bundle branch block beat, right bundle branch block beat, premature ventricular contraction, and paced beat, the classification accuracy, the area under the curve (AUC), the sensitivity, and the F1-score achieved by the proposed model were 0.9924, 0.9994, 0.99 and 0.99, respectively. When the heartbeat types were divided into six classes recommended by clinicians, i.e., normal beat, left bundle branch block beat, right bundle branch block beat, premature ventricular contraction, paced beat and other beats, the beat classification accuracy, the AUC, the sensitivity, and the F1-score achieved by the model reached 0.9702, 0.9966, 0.97, and 0.97, respectively. When the heartbeat types were divided into five classes recommended by the Association for Advancement of Medical Instrumentation (AAMI), i.e., normal beat, supraventricular ectopic beat, ventricular ectopic beat, fusion beat, and unknown beat, the beat classification accuracy, the sensitivity, and the F1-score were 0.9745, 0.97, and 0.97, respectively. Experimental results show that the proposed method achieves better performance than the state-of-the-art methods.
computer science, information systems, theory & methods