| Literature DB >> 34531965 |
Jixiang Zhang1, Chengqin Wu1, Chenzhao Ruan1, Rongxia Zhang1, Zengshun Zhao1, Xiangqian Cheng1.
Abstract
At present, cardiovascular disease is regarded as one of the dangerous diseases that threaten human life. The morbidity and lethality caused by cardiovascular disease are constantly increasing every year. In this paper, we propose a two-stream style operation to handle the electrocardiogram (ECG) classification: one for time-domain features and another for frequency-domain features. For the time-domain features, convolutional neural networks (CNN) are constructed for feature learning and classification of ECG signals. For the frequency-domain features, support vector regression (SVR) machines are designed to perform the regression prediction on each signal. Finally, the D-S evidence theory is adopted to perform the decision fusion strategy on the time-domain and frequency-domain classification results. We confirm a recognition performance of 99.64% from the experiment result for the D-S evidence theory recognition system upon the MIT-BIH arrhythmia database. The analysis of various methods of ECG classification shows that the model delivers superior performance promotion across all these scenarios.Entities:
Mesh:
Year: 2021 PMID: 34531965 PMCID: PMC8440070 DOI: 10.1155/2021/4222881
Source DB: PubMed Journal: J Healthc Eng ISSN: 2040-2295 Impact factor: 2.682
The main highlights and drawbacks of the methods mentioned.
| Author | Approach | Database | ECG beats | Advantages | Disadvantages |
|---|---|---|---|---|---|
| Balasundaram [ | Wavelet analysis | MITDB | Rhythmic ventricular tachycardia WA (VT) | This method performs well in the overlap zone between VT and VF. | The limitation is that it cannot be used as a risk-stratifier, because this method cannot determine the probability of future VF episodes. |
|
| |||||
| Sayantan [ | Gaussian-Bernoulli deep belief network and active learning | SVDB MITDB | AAMI | By using the expert interaction, this method is robust and it can overcome the variance in data distribution in interpatient scenarios. | This method handles intraclass variations poorly. |
|
| |||||
| Javadi [ | ME and NCL | MITDB | Normal (N) | Combined with NCL and ME, it can enable the training algorithm of ME to establish a balance in bias-variance-covariance trade-offs, and it improves the accuracy and generalization of the model. | This work does not provide further insights of the classification boundaries. |
|
| |||||
| Rajpurkar [ | CNN | MITDB | Atrial fibrillation (AFIB) | This method uses a very deep CNN; the model can achieve a high accuracy rate under a big dataset. | If two ECG signals are similar, the model sometimes makes mistakes, such as Wenckebach and AVB_Type2, Supraventricular Tachycardia (SVT), and atrial flutter (AFL). |
|
| |||||
| Li [ | CNN | MITDB | AAMI | This method uses the SMOTE algorithm to balance the classes in dataset. | It only extracted the time-domain features. |
|
| |||||
| Alif [ | 2D CNN | MITDB | AAMI EC57 | This method utilizes CNN to extract features automatically. | It only extracted the time-domain features. |
|
| |||||
| Marinho [ | Feature extraction: Fourier, goertzel, higher order statistics (HOS), and SCM. | MITDB | ANSI/AAMI | This is the first time that SCM has been applied to feature extraction. | It is dependent on QRS's window lengths. |
|
| |||||
| Faziludeen [ | Wavelets and SVM | MITDB | Normal (N) | This work uses one against one (OAO) SVM to classify ECG signals. | This method requires designing features manually, and the classification of ECG signals is less. |
|
| |||||
| Radovan [ | SVM | PhysioNet/CinC | Normal (N) | Combined with SVM and simple threshold-based rules, it can improve performance. | This method requires designing complex features manually. |
|
| |||||
| Mondéjar-Guerra [ | Multiple SVMs | MITDB | Normal (N) | This work trains and integrates specific SVM models for each type of feature; it offers a satisfactory performance. | Data fusion is relatively simple. |
Figure 1The framework of the proposed classification model.
The type of arrhythmia used in this paper.
| Classes (instances) | Size of segments |
|---|---|
| Atrial premature beat (A) | 2545 |
| Left bundle branch block (L) | 8068 |
| Right bundle branch block (R) | 7254 |
| Premature ventricular contraction (V) | 7026 |
| Normal (N) | 74955 |
| Total | 99848 |
Figure 2Waveforms of five arrhythmia types: (a) A, (b) L, (c) R, (d) V, and (e) N.
Figure 3Renderings of low-frequency and high-frequency noise suppression.
Figure 4An individual heart beat waveform of the ECG signal sample.
Figure 5The structure diagram of wavelet packet decomposition [20].
The architecture of 1-D CNN.
| Layers | Type | Activation function | Number of neurons (output layer) | Filter size | Number of filters | Stride |
|---|---|---|---|---|---|---|
| 1 | Input | 250 × 1 | ||||
| 2 | Convolution | ReLU | 246 × 4 | 5 × 1 | 4 | 1 |
| 3 | Max-pooling | 123 × 4 | 2 × 1 | 4 | 2 | |
| 4 | Convolution | ReLU | 120 × 4 | 4 × 1 | 4 | 1 |
| 5 | Max-pooling | 60 × 4 | 2 × 1 | 4 | 2 | |
| 6 | Convolution | ReLU | 56 × 4 | 5 × 1 | 4 | 1 |
| 7 | Max-pooling | 28 × 4 | 2 × 1 | 4 | 2 | |
| 8 | Convolution | ReLU | 24 × 8 | 5 × 1 | 8 | 1 |
| 9 | Max-pooling | 12 × 8 | 2 × 1 | 8 | 2 | |
| 10 | Fully connected | ReLU | 10 | |||
| 11 | Output | SoftMax | 5 |
Results of classification based on 1D CNN.
| Type | Acc (%) | PPV (%) | SE (%) | SP (%) | |
|---|---|---|---|---|---|
| N | 99.34 | 99.74 | 99.14 | 99.54 | 99.79 |
| V | 98.31 | 99.32 | 98.48 | 98.13 | 99.61 |
| R | 99.08 | 99.59 | 98.82 | 99.15 | 99.70 |
| L | 99.65 | 99.86 | 99.49 | 99.81 | 99.88 |
| A | 97.43 | 98.97 | 97.79 | 97.08 | 99.45 |
Results of classification based on SVR.
| Type | Acc (%) | PPV (%) | SE (%) | SP (%) | |
|---|---|---|---|---|---|
| N | 90.04 | 95.86 | 86.23 | 94.24 | 96.27 |
| V | 95.49 | 98.27 | 97.92 | 93.21 | 99.51 |
| R | 97.18 | 98.88 | 97.38 | 97.00 | 99.35 |
| L | 97.35 | 98.92 | 96.27 | 98.46 | 99.04 |
| A | 90.12 | 96.17 | 93.20 | 87.27 | 98.40 |
Results of classification based on the incorporation of outputs by D-S evidence theory.
| Type | Acc (%) | PPV (%) | SE (%) | SP (%) | |
|---|---|---|---|---|---|
| N | 99.49 | 99.80 | 99.49 | 99.49 | 99.88 |
| V | 99.04 | 99.60 | 99.52 | 98.56 | 99.87 |
| R | 99.98 | 99.60 | 99.49 | 98.48 | 99.88 |
| L | 99.76 | 99.90 | 99.51 | 100 | 99.87 |
| A | 98.20 | 99.30 | 97.45 | 98.96 | 99.38 |
Figure 6Confusion matrix of the classified ECG signals.
Results of all classifiers on the ECG test dataset.
| Classifier | Acc | N | V | R | L | A | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| PPV | SE | SP | PPV | SE | SP | PPV | SE | SP | PPV | SE | SP | PPV | SE | SP | ||||
| 1D CNN | 99.5 | 99.1 | 99.5 | 99.7 | 98.5 | 98.1 | 99.6 | 98.8 | 99.2 | 99.7 | 99.4 | 99.8 | 99.9 | 97.8 | 97.1 | 99.5 | ||
| SVR | 96.5 | 86.2 | 94.2 | 96.3 | 97.9 | 93.2 | 99.5 | 97.4 | 97.0 | 99.4 | 96.3 | 98.5 | 99.0 | 93.2 | 87.3 | 98.4 | ||
| D-S | 99.6 | 99.5 | 99.4 | 99.8 | 99.5 | 98.6 | 99.9 | 99.5 | 98.5 | 99.9 | 99.5 | 100 | 99.8 | 97.5 | 99.0 | 99.4 | ||
Figure 7Comparison of accuracy of three models.
Comparison of different classification approaches.
| Author | ECG beats | Approach | Database | Performance (%) |
|---|---|---|---|---|
| Mehrdad Javadi et al. [ | Normal (N) | Mixture of experts (ME) and negatively correlated learning (NCL) | MIT-BIH | SPn = 98.01 |
|
| ||||
| Mondéjar-Guerra V [ | Normal (N) | SVMs | MIT-BIH | SEN = 95.9 |
|
| ||||
| Shi hangrui [ | Left bundle branch block (LBBB) | Learning vector quantization (LVQ) | MIT-BIH | AccAPB = 84.2 |
|
| ||||
| Wang Run [ | N | Radial basis function (RBF) | MIT-BIH | AccN = 89.7 |
|
| ||||
| Yıldırım et al. [ | 13 classes | CNN | MIT-BIH | SE13 = 93.52 |
|
| ||||
| Li et al. [ | AAMI | RF | MIT-BIH | SEN = 94.67 |
|
| ||||
| Amrita Rana et al. [ | N | LSTM | MIT-BIH | Acc = 95.00 |
|
| ||||
| Anika alim et al. [ | N abnormal | SVM and ANN | MIT-BIH | AccSVM = 87 |
|
| ||||
| Sherin M. Mathews et al. [ | SVEB | Restricted Boltzmann machine (RBM) and deep belief networks (DBN) | MIT-BIH | AccSVEB = 93.63 |
|
| ||||
| Ours | N | CNN and SVR by D-S evidence theory | MIT-BIH | SEN = 99.49 |