Inter-patient arrhythmia classification with improved deep residual convolutional neural network
Li, Y.; Qian, R.; Li, K.
Computer Methods and Programs in Biomedicine 214: 106582
ISSN/ISBN: 1872-7565 PMID: 34933228 Accession: 079527106
Early detection of arrhythmias has become critical due to the increased mortality from cardiovascular disease, and ECG is an effective tool for diagnosing cardiovascular disease and detecting arrhythmias. Classification based on ECG signal segments is more suitable for clinical application. An improved deep residual convolutional neural network is proposed to classify arrhythmias automatically. Firstly, the overlapping segmentation method is used to segment the ECG signals in the MIT-BIH database into segments of 5 seconds in length to overcome the imbalance between classes, and these segments of the ECG signals are re-labeled. Then the discrete wavelet transform (DWT) is used to denoise these segments and the improved deep residual convolutional neural network is used for arrhythmia classification. In addition, the focal loss function is used to overcome the imbalanced classification difficulty between classes. The proposed method gives 94.54% sensitivity, 93.33% positive predictivity, and 80.80% specificity for normal segments. For the supraventricular ectopic segment, the proposed method gives 35.22% sensitivity, 65.88% positive predictivity, and 98.83% specificity. For the ventricular ectopic segment, the proposed method gives 88.35% sensitivity, 79.86% positive predictivity, and 94.92% specificity. The results of this study indicate that the proposed improved deep residual convolutional neural network model trained by the training set obtained by using the overlapping segmentation method is comparable to a classical method and three state-of-art methods. In addition, the classification performance of the network model trained by focal loss as the loss function is further improved.