Next Article in Journal
Iterative Diffusion-Based Distributed Cubature Gaussian Mixture Filter for Multisensor Estimation
Next Article in Special Issue
Elimination of Drifts in Long-Duration Monitoring for Apnea-Hypopnea of Human Respiration
Previous Article in Journal
Chemically Roughened Solid Silver: A Simple, Robust and Broadband SERS Substrate
Previous Article in Special Issue
Preliminary Study on Continuous Recognition of Elbow Flexion/Extension Using sEMG Signals for Bilateral Rehabilitation
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Arrhythmia Classification Based on Multi-Domain Feature Extraction for an ECG Recognition System

1
School of Electronics and Information Engineering, Tianjin Polytechnic University, Tianjin 300387, China
2
Tianjin Chest Hospital, Tianjin 300222, China
*
Author to whom correspondence should be addressed.
Sensors 2016, 16(10), 1744; https://doi.org/10.3390/s16101744
Submission received: 30 August 2016 / Revised: 29 September 2016 / Accepted: 14 October 2016 / Published: 20 October 2016
(This article belongs to the Special Issue Wearable Biomedical Sensors)

Abstract

:
Automatic recognition of arrhythmias is particularly important in the diagnosis of heart diseases. This study presents an electrocardiogram (ECG) recognition system based on multi-domain feature extraction to classify ECG beats. An improved wavelet threshold method for ECG signal pre-processing is applied to remove noise interference. A novel multi-domain feature extraction method is proposed; this method employs kernel-independent component analysis in nonlinear feature extraction and uses discrete wavelet transform to extract frequency domain features. The proposed system utilises a support vector machine classifier optimized with a genetic algorithm to recognize different types of heartbeats. An ECG acquisition experimental platform, in which ECG beats are collected as ECG data for classification, is constructed to demonstrate the effectiveness of the system in ECG beat classification. The presented system, when applied to the MIT-BIH arrhythmia database, achieves a high classification accuracy of 98.8%. Experimental results based on the ECG acquisition experimental platform show that the system obtains a satisfactory classification accuracy of 97.3% and is able to classify ECG beats efficiently for the automatic identification of cardiac arrhythmias.

1. Introduction

With the progress of science and technology, automatic analysis and diagnosis systems based on electrocardiogram (ECG) signals have been extensively investigated to detect and diagnose cardiac diseases [1,2,3,4,5]. The use of an ECG automatic analysis and diagnosis system not only reduces the workload of doctors but also improves diagnostic efficiency and accuracy. ECG is a synthesized reflection of heart electrical activity in the human body used to obtain the cardiac rhythm and electrical conduction of physiological and pathological information [6,7]. However, ECG signals are often mixed with various noises and artefacts. As a result, examiners experience difficulty distinguishing normal signals from arrhythmia signals. Pre-processing of ECG signals is necessary to reduce various interferences. Some techniques based on digital filters have been employed to remove noise and power-line interference from biomedical signals [8,9,10]. Other mathematical morphological algorithms are applied in signal pre-processing [11,12]. Wavelet transform (WT) method based on its good time-frequency property for signal denoising have been popularly developed and applied to eliminate noise effectively in ECG identification system [13,14,15,16]. In addition, an adaptive filtering algorithm for ECG denoising was proposed and compared with the denoising method based on wavelet shrinkage [17]. An approach based on empirical mode decomposition and an improved approximate envelope method was also presented for ECG signal processing [18].
Previous studies have demonstrated the successful application of various feature extraction and classification methods. ECG features mainly include time domain, frequency domain, morphological and nonlinear features. Classification methods mainly focus on random forest [19], linear discriminant classification, neural networks (NNs), support vector machine (SVM), etc. As ECG waveforms in time domains are complicated and easily interfered by noise, classic time domain analysis had low accuracy in extracting features [20,21]. Thereafter, the transform and morphological methods are studied to obtain ECG features containing lots of time and frequency information [19,22]. However, these features can not reflect more accurate characteristics of ECG for achieving high classification accuracy. Thus, combining several methods has become popular in ECG feature extraction and classification recently. A wavelet optimization approach based on the combination of the polyphase representation of wavelets, particle swarm optimization for feature extraction and the SVM classifier was employed in the classification of ECG signals [23]. Lin extracted normalized R-R interval and morphological features as the input features to train and test the linear discriminant classifier in the classification of heartbeats [24]. A cross-correlation-based approach was used to extract suitable features, and a least squares SVM (LSSVM) classifier was developed to classify ECG beats [25]. Several independent component analysis (ICA) algorithms were tested and analysed to identify various components with high accuracy in a particular algorithm based on biomedical data for classification [26]. The extracted bispectral features were subjected to principle component analysis (PCA) for dimensionality reduction and then input into a four-layer feed-forward NN and LSSVM for classification [27]. Valenza proposed a personalized probabilistic framework wherein features were derived from instantaneous spectrum and bispectrum; an SVM classifier was applied in heartbeat recognition [28]. In a previous study, multiclass-directed acyclic graph SVM was implemented on feature vectors utilising empirical mode decomposition and singular value decomposition for ECG classification [29]. Kamath studied ECG beats from the energy perspective by extracting features from the nonlinear component in time and frequency domains via the Teager energy operator and used an NN as a classifier to identify the five classes of ECG beats [30].
Although the studies mentioned above can also produced good classification, they only used the combination of time domain, frequency domain and linear features, or used single nonlinear features in ECG classification, which can not represent more complex characteristics of ECG signals. Therefore, the current study proposes an ECG recognition system that extracts multi-domain features through kernel-independent component analysis (KICA) [31] and discrete wavelet transform (DWT). Before feature extraction, ECG signals are pre-processed with an improved threshold method based on wavelet transform for ECG denoising. PCA and linear discriminant analysis (LDA) [32] are applied to reduce feature dimensions. Multi-domain features, which contain abundant characteristic information on ECG, are used as input to train and test an SVM classifier. The SVM classifier is optimized by the genetic algorithm (GA) [33] and applied to categorise ECG beats into five types, namely, normal beat (N), left bundle branch block beat (LBBB), right bundle branch block beat (RBBB), premature ventricular contraction (PVC) and atrial premature beat (APC) [27,34]. The ECG arrhythmia recordings employed in this study are derived from the MIT-BIH arrhythmia database, which is a complete system that began distributing information in 1980 and can be accessed from the PhysioBank. An ECG acquisition experimental platform is also constructed to acquire ECG data from Fluke ProSimTM2 vital sign simulator (Fluke Corp., Everett, WA, USA) to demonstrate the effectiveness of the proposed ECG recognition system. The experimental results based on ECG data collected from the experimental platform show that the proposed system can achieve satisfactory identification results.

2. Materials and Methods

2.1. The Proposed System Based on the Multi-Domain Feature Extraction

The entire block diagram of the proposed ECG recognition system for ECG beats classification is shown in Figure 1. An ECG recognition system based on multi-domain feature extraction is proposed to identify cardiac arrhythmias. The proposed system consists of three main parts, namely, ECG pre-processing, feature extraction and classification; its procedure is illustrated by the following steps.
1
The ECG beats are pre-processed to eliminate disturbance and noise by denoising with an improved wavelet threshold method.
2
The original ECG data are optimized through PCA. KICA is applied to reduce data dimensions and obtain the nonlinear features of ECG beats. The DWT method is employed to extract frequency domain features. The maximum, minimum, mean and standard deviation values of each sampling signal wavelet coefficient are calculated. Additionally, LDA is used to optimise the frequency domain features.
3
The multi-domain features are composed of nonlinear and frequency domain features, which are used as input features to train and test the SVM classifier model. GA is employed to optimise the SVM parameters and improve the classifier’s performance. Finally, the five types of ECG beats derived from the MIT-BIH arrhythmia database are classified with the optimised SVM classifier.

2.2. ECG Pre-Processing Based on the Improved Wavelet Threshold Method

ECG signals are weak and often contain noise and interference. Therefore, pre-processing ECG signals before feature extraction and classification by the proposed ECG recognition system is necessary. Given that the energy distributions of ECG and noise are different, we propose an improved threshold method based on wavelet transform for denoising. The improved threshold function of the proposed denoising method is described as follows:
w ^ j , k = { u w + ( 1 u ) s i g n ( w j , k ) ( | w j , k | ( 1 b ) λ ) , | w j , k | λ b s i g n ( w j , k ) ( w j , k 4 λ 3 ) , | w j , k | < λ ,
where u = 1 1 / e ( a ( | w j , k | λ ) 2 ) , wj,k is the wavelet coefficient, w ^ j , k is the wavelet coefficient obtained by wavelet threshold processing, λ is the critical threshold, a and b are regulatory factors and a can be any positive integer, 0 b 0.1 . By adjusting the values of a and b, the continuity of the threshold function is achieved at the critical point, and the attenuation of the reconstructed signal is reduced, thus overcoming the inadequacy of traditional threshold functions. The critical threshold is λ = σ 2 log N , and the noise intensity is σ = ( m e d i a m | w j , k | ) / 0.6745 in the proposed ECG denoising method [35]. We choose sym6 as the mother wavelet and utilise WT based on the Mallat algorithm [36] to decompose ECG signals into five levels, and the decomposition results are shown in Figure 2. The wavelet coefficients are then quantized with the improved wavelet threshold method to remove noise wavelet coefficients. The resulting new wavelet coefficients are utilised to reconstruct ECG signals on the basis of the Mallat algorithm to obtain the denoised ECG signals.
To verify the utility of the presented method in ECG denoising, we obtain a noisy signal for processing with the proposed method by adding white noise as electromyographical interference and 50 Hz power-line interference to a 3000 point normal signal from the MIT-BIH arrhythmia database. The noisy signal is also processed by the hard threshold and soft threshold methods for comparison with the proposed denoising method. Figure 3 and Figure 4 show the denoising results of different threshold functions and the spectra of different signals. For better evaluating the performance of different denoising methods, signal-noise ratio (SNR) and root mean square error (RMSE) are computed based on the following equations:
S N R = 10 log i = 1 n x 2 ( t ) i = 1 n ( x ( t ) x ^ ( t ) ) 2 ,
R M S E = 1 n i = 1 n ( x ( t ) x ^ ( t ) ) 2 ,
where x(t) is the original signal, x ^ ( t ) is the denoised signal and n is the sampling length. While the SNR is bigger and RMSE is smaller, the denoised effect is better. Five ECG signal records from MIT-BIH arrhythmia database are selected to be processed by hard threshold, soft threshold and improved threshold, respectively. The performance indicators of three threshold denoising methods are shown in Table 1, which indicates the proposed denoising method based on the improved wavelet threshold performs well relatively.

2.3. Multi-Domain Feature Extraction

2.3.1. KICA for the Nonlinear Feature Extraction

ECG is mixed with a variety of information, and most of the important information is often included in the nonlinear process [37]. Thus, nonlinear feature extraction helps achieve better recognition of ECG beats. The method of feature extraction based on KICA is presented to extract nonlinear features in the proposed ECG recognition system. Given that high-dimensional data result in a large amount of calculation, we apply PCA [38] to reduce ECG data dimensions prior to nonlinear feature extraction by KICA. The PCA algorithm computes the eigenvalues λi (i = 1, 2, …, k) and eigenvectors p of the covariance matrix C of ECG samples. Here, k is the rank of the covariance. The contribution and cumulative contribution rates are calculated according to the size of eigenvalue λi sequenced in decreasing order. This study selects 99.8% of the cumulative contribution rate. We filter the corresponding eigenvectors of the first 20 maximum eigenvalues (λ1, λ2, …, λ20) of covariance matrix C. A total of 20 dimensions from the original data are obtained as the input data of KICA.
KICA [31] is a kernel method introduced in the ICA algorithm to solve the classification problem of complex nonlinear structures. KICA is applied in reconstruction within kernel Hilbert space (RKHS) nonlinear functions as comparison functions to obtain signals from a low-dimensional space mapped to a high-dimensional space. The radial basis function (RBF) kernel is selected as the kernel function of KICA to realize nonlinear transformation in this study and is defined below:
K ( x i , x ) = exp ( x i x 2 2 σ 2 ) ,
where δ is a positive real number.
The concrete steps of KICA used in nonlinear feature extraction are as follows:
  • Enter ECG data and determine the number of sources p.
  • Initialize the decomposition matrix W.
  • Evaluate the source signals, si = Wxi, and compute the centralized Gram matrix, K1, …, Kp. Here, Ki = k(xi,x) (I = 1, 2, …, p).
  • Compute the minimum eigenvalue, λM (K1, …, Kp), of the generalized eigenvector equation, Kα = λDα.
  • Compute the target function:
    C ( W ) = 0.5 log 2 λ M ( K 1 , , K p ) .
  • Minimize the target function C(W) and output W.
After W is attained by KICA decomposition, feature subspace S is constructed by the set of independent components (si = Wxi). The coefficient of each ECG beat mapped to the feature subspace is obtained as the nonlinear feature vector. The pseudo inverse method is applied to recognize the projection coefficient vector based on the following equation:
A i = x i × S 1 ,
where Ai is the nonlinear feature vector of the recognized ECG beats and S−1 is the pseudo inverse matrix of independent signal vector S. We set the regularization parameter kap = 0.02 and the nuclear width of RBF kernel function δ = 1 in the KICA decomposition. The feature subspace obtained by using KICA is shown in Figure 5.

2.3.2. DWT for Frequency Domain Feature Extraction

The DWT method based on multi-resolution analysis possesses good properties concerning time-frequency localization and is widely used in biomedical signal processing and analysis [22,23,24]. This study employs DWT to extract the frequency domain features of ECG beats in the proposed system. Daubechies 2 (db2) is selected as the mother wavelet because of its good smoothing effect [39]. The ECG beats are broken down into four levels, and the detail coefficients of each level and the approximation coefficient of the 4th level are set as ECG frequency domain features. We select records 100, 109, 118, 106 and 209 from the MIT-BIH arrhythmia database to represent five types of ECG beats, namely, N, LBBB, RBBB, PVC and APC, for decomposition with DWT as shown in Figure 6. The maximum, minimum, mean and standard deviation values of the wavelet coefficients of the ECG signal are calculated as new frequency domain features. We also utilise LDA [32] as a dimension reduction tool to extract more effective frequency domain features in this study.

2.4. Classification Based on SVM Optimized by GA

SVM, which was developed by Vapnik [40], has been widely applied in ECG classification studies [23,25,28,29]. SVM finds a hyperplane in a high-dimensional space by separating the training samples of each class or by maximizing the minimum distance between the hyperplane and training samples. Our proposed ECG recognition system utilises a library for SVM (LIBSVM) [41] as the classifier and adopts the popular RBF function as a kernel function. To improve the performance of the classifier, GA, a type of optimization searching algorithm based on the Darwinian principle of survival of the fittest [42,43], is also employed to optimize the parameters C and δ of the RBF kernel function. The chromosome of GA contains two genes because the chromosome of SVM contains separate values for kernel parameters δ and C. The initial parameters of GA are max iteration = 200, max number of population = 20 and parameter of cross validation = 5. The value range of penalty factor C is 0–100. The value range of RBF kernel width δ is 0–1000. The basic parameter optimization of SVM with GA is described below:
  • Generate an initial population with the binary code.
  • Compute the fitness function and the training set for CV to determine the fitness function value.
  • Filter a new population from the old population on the basis of individual fitness, which is determined by the evaluation function.
  • Employ several genetic operators of mutation and crossover to generate new solutions.
  • Calculate the fitness function of the newly generated individuals.
  • Repeat steps 3–5 until the maximum iteration is reached and then find the optimal parameters.

3. Experimental Design

We construct an ECG acquisition experimental platform to collect ECG data for arrhythmia recognition to verify the validity of the proposed system. Fluke ProSimTM2 vital sign simulator is used as the signal resource to provide different types of ECG signals, and standard I lead ECG is selected in this platform. An ECG acquisition module is a critical part of the experimental platform and is integrated with an ADuCM361 (Analog Devices, Inc., Norwood, MA, USA) micro controller and an A/D converter (Analog Devices, Inc., Norwood, MA, USA) to acquire analog ECG signals and convert them into digital ones. The collected ECG data are transmitted via Bluetooth HC-05 (DX-Smart Technology Co. Ltd., Shenzhen, China) to a PC, and the data are then input into the proposed ECG recognition system for classification. The power module is used to supply electricity to the ECG acquisition module and the Bluetooth module. Figure 7 provides diagrams of the ECG acquisition experimental platform.

4. Results and Discussion

4.1. Results of the Proposed ECG Recognition System

4.1.1. Multi-Domain Feature Extraction Using KICA Combined with DWT

A novel method based on multi-domain feature extraction is proposed to extract many effective features for ECG recognition. ECG data from the MIT-BIH arrhythmia database are sampled, and the data are pre-processed with the improved wavelet threshold method. Given that the amplitude of the R wave is the most outstanding in the ECG signal, the R wave position serves as the benchmark before and after sampling 250 time domain points containing the complete QRS complex of the ECG signal. A total of 1800 samples from the MIT-BIH arrhythmia database are equally divided into training sets. A total of 400 samples of N are derived from records 100, 101, 103 and 105. Similarly, 400 samples of LBBB are derived from records 109, 111, 207 and 214, and 400 samples of RBBB are derived from records 118, 124, 212 and 231. We also derive 400 samples of PVC from records 106, 119, 200 and 203 and 200 samples of APC from records 209 and 222. A total of 1800 samples are used as ECG data after sampling and pre-processing the ECG signals. Each sample contains 250 points for multi-domain feature extraction. A matrix of 1800 × 250 with high dimensions requires extensive calculation for KICA. To obtain effective features from the ECG signals, we reduce the dimensions of ECG data through PCA. The method of PCA dimension reduction is used, and the cumulative contribution rate is set to 99.8%. Twenty dimensions are selected from the original set as KICA input. Thereafter, KICA is applied to extract the nonlinear features of five types of ECG beats for classification with the proposed system. In the algorithm of nonlinear feature extraction using KICA, the regularization parameter is kap = 0.02, and the nuclear width of the RBF kernel function is δ = 1. A set of statistically independent base signals, si (i = 1, …, 20), is obtained, and the feature subspace is constructed with the base signal vector. The coefficient of each projection to the feature subspace is for nonlinear features. Therefore, a feature matrix of 1800 × 20 is obtained as the nonlinear features after KICA.
This study applies DWT to extract the frequency domain features of ECG beats, and db2 is selected as the mother wavelet function to decompose ECG beats into four levels. The details of each level and the approximation of the 4th level as the frequency domain features are computed to obtain the statistical features, which include the maximum, minimum, mean and standard deviation values, of each wavelet coefficient. After 20 dimensions of the frequency domain features are obtained by DWT combined with the statistical method, LDA is utilised to optimise these features reduced to four dimensions. Therefore, a feature vector with a 1800 × 4 matrix is derived as the frequency domain features.

4.1.2. Classification Using SVM Optimized by GA

We obtain a multi-domain feature vector with a 1800 × 24 matrix by KICA combined with DWT. The matrix is used as an input to train and test the SVM classifier. ECG beats are identified by LIBSVM, and RBF is applied to classify multi-domain features. The selection of the kernel function and parameters is extremely important for SVM. The selection of penalty factor C and RBF kernel width δ determines the classifier’s performance, and the optimal parameters can effectively prevent overlearning and lack of a learning state. Therefore, we employ GA to determine the optimal parameters of SVM in the proposed system. The final optimization results show that C = 2.61633 and δ = 4.16832. The fitness curve of GA for finding the optimal parameters of SVM is shown in Figure 8.
Accuracy (ACC), sensitivity (Se), specificity (Sp) and positive predictivity (Pp) are evaluated to investigate the recognition performance of the proposed system. The performance parameters for each class are defined as follows:
A C C = T P + T N T P + T N + F P + F N × 100 % = N T N E N T × 100 %
S e = T P T P + F N × 100 %
S p = T N T N + F P × 100 %
P p = T P T P + F P × 100 %
where TP is true positive, FP is false positive, FN is false negative and TN is true negative. NE and NT represent the total number of classification errors and the total number of the testing sets, respectively. Table 2 and Table 3 present the multi-domain feature classification results obtained with the SVM classifier. Table 2 shows that the five types of ECG beats produce different classification results. N and RBBB samples are correctly classified. Two LBBB samples are incorrectly classified as PVC. Five PVC samples are classified as LBBB. Three APC samples are incorrectly classified as N, and one sample is classified as PVC. Table 3 clearly shows that the five types of ECG beats perform well in classification. RBBB has the best statistical performance indicators of sensitivity, specificity and positive predictive value amongst the five types of ECG beats. APC presents good performance with a specificity of 100% and a positive predictive value of 100%; its sensitivity of 96% is the lowest among the sensitivities of other types. N, LBBB and PVC present similar statistical performance indicators.

4.2. Experimental Results from the ECG Acquisition Platform

The five types of ECG signals acquired by the designed experimental platform (see Figure 9) are entered into the proposed ECG recognition system to be classified. Firstly, noise is removed from the collected ECG signals with the improved wavelet threshold method. Secondly, 1800 ECG beats are sampled and utilised to obtain the nonlinear and frequency domain features of ECG beats with the proposed method. Thirdly, PCA is employed to reduce the dimensions of ECG sample data in the nonlinear feature extraction, and LDA is applied to reduce the dimensions of the frequency domain features. The multi-domain features of the ECG beats are then classified with LIBSVM, and GA is adopted to optimise the penalty factor and the kernel width of the classifier. Lastly, we utilise the proposed ECG recognition system to perform experimental classification. The classification results are shown in Table 4 and Table 5. Table 4 shows that the five types of ECG beats produced different classification results. Two N samples are incorrectly classified as LBBB, and one N sample is classified as PVC. Four LBBB samples are incorrectly classified as N. Two samples of LBBB are classified as PVC. Only one sample of RBBB is incorrectly classified as PVC. Eleven PVC samples are incorrectly classified. One sample of APC is incorrectly classified as N. Table 5 shows that the classification of the five types of ECG beats achieves an overall accuracy of 97.3%. RBBB and APC present the best statistical performance indicators of sensitivity, specificity and positive predictive value amongst the five types of ECG beats. PVC has a good performance with a specificity of 99.43% and a positive predictive value of 97.93%; its sensitivity of 94.50% is the lowest amongst the sensitivities of the other types. N and LBBB possess similar statistical performance indicators.

4.3. Discussion and Comparisons

Table 3 and Table 5 clearly show that the proposed ECG recognition system achieves satisfactory identification results for two types of ECG data. The proposed system using the MIT-BIH arrhythmia database presents excellent performance with an accuracy of 98.80%, average sensitivity of 98.50%, average specificity of 99.69% and average positive specificity of 98.91%. Experimental classification results from the ECG acquisition experimental platform show that the accuracy, average sensitivity, average specificity and average positive specificity are 97.30%, 97.50%, 99.32% and 97.41%, respectively. Although the classification results using ECG data from the ECG acquisition experimental platform are good, the proposed system using ECG data from the MIT-BIH arrhythmia database achieves better results and classification accuracy based on the three statistical performance indicators. The primary reason is that ECG signals from Fluke ProSimTM2 are limited in providing a large number of ECG samples. Another important reason is that several types of ECG signals might require other features to represent their characteristic information. Therefore, further study and evaluation of the algorithm based on feature extraction and classification in the ECG recognition system are necessary.
The results of the proposed ECG recognition system are compared with those of other reported ECG classification methods. The comparison results are shown in Table 6. Martis utilised morphological and time features and used LSSVM to classify five ECG classes (N, RBBB, LBBB, APC and PVC); accuracy of 93.48% was achieved [27]. Kamath derived ECG features based on Teager energy functions, classified five types of beats (N, PVC, Paced beat, RBBB and LBBB) with an NN classifier and obtained a recognition accuracy of 95% [30]. In another study, morphological and time features as effective features for three beat classes (N, PVC and other beats) were used as input for an SVM classifier with a classification accuracy of 97.14% [44]. The SVM classifier was used to identify four types of ECG (N, atrial arrhythmia, conduction block and ventricular arrhythmia) based on the time interval feature extracted via biorthogonal spline wavelet, which achieved an accuracy of 95.65% [20]. Liu presented an algorithm based on SVM to detect and mark ECG signals. The algorithm utilised the self-constructing neural fuzzy interference network (SoNFIN) to recognize five ECG signal types (N, PVC, PAC, LBBB and RBBB) with an accuracy of 96.4% [21]. As shown in Table 6, the proposed ECG recognition system based on multi-domain feature extraction using KICA combined with DWT achieves higher accuracy of 98.8% in comparison with other methods.

5. Conclusions

A novel ECG recognition system based on multi-domain feature extraction using KICA and combined with DWT is proposed to classify five types of ECG heartbeats. A new improved threshold wavelet method for ECG pre-processing is also presented to eliminate the influence of noise. In multi-domain feature extraction, PCA as a feature reduction method is employed to reduce ECG data dimensions and decrease calculation in KICA nonlinear feature extraction. LDA is also adopted to optimise the frequency domain features extracted by DWT. SVM is utilised as the classifier to recognise ECG beats. We apply GA to improve the performance of the SVM classifier by optimizing RBF kernel function parameters δ and C. The proposed system with the MIT-BIH arrhythmia database for ECG beat classification achieves 98.8% accuracy, 98.50% sensitivity, 99.69% specificity and 98.91% positive predictability. We also construct an ECG acquisition experimental platform for ECG data collection to verify the effectiveness of the proposed system. The results of the experimental ECG acquisition platform show that the system obtains excellent results with 97.3% accuracy, 97.5% sensitivity, 99.32% specificity and 97.41% positive predictability. The presented system is able to achieve satisfactory classification results in classifying five types of ECG beats. Therefore, the proposed ECG recognition system can be used to effectively diagnose heart diseases.

Acknowledgments

This study is supported by the National Natural Science Foundation of China (No. 61675154, 61605144, 61177078, 61307094, 61504093, 31271871).

Author Contributions

Hongqiang Li proposed and guided the overall study with contributions from Danyang Yuan and DianYin Cui The algorithm of ECG signals classification was designed by Danyang Yuan, and Youxi Wang and Dianyin Cui performed the experiments. Lu Cao participated in analysis of the ECG data. All authors read and approved the final manuscript.

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. Edla, S.; Kovvali, N.; Papandreou-Suppappola, A. Electrocardiogram signal modeling with adaptive parameter estimation using sequential bayesian methods. IEEE Trans. Signal Process. 2014, 62, 2667–2680. [Google Scholar] [CrossRef]
  2. Wang, J.; She, M.; Nahavandi, S.; Kouzani, A. Human identification from ECG signals via sparse representation of local segments. IEEE Signal Process. Lett. 2013, 20, 937–940. [Google Scholar] [CrossRef]
  3. Ubeyli, E.D. Combining recurrent neural networks with eigenvector methods for classification of ECG beats. Digit. Signal Process. 2009, 19, 320–329. [Google Scholar] [CrossRef]
  4. Jain, S.; Bajaj, V.; Kumar, A. Efficient algorithm for classification of electrocardiogram beats based on artificial bee colony-based least-squares support vector machines classifier. Electron. Lett. 2016, 52, 1198–1200. [Google Scholar] [CrossRef]
  5. Zadeh, A.E.; Khazaee, A. High efficient system for automatic classification of the electrocardiogram beats. Ann. Biomed. Eng. 2011, 39, 996–1011. [Google Scholar] [CrossRef] [PubMed]
  6. Plonsey, R.; Barr, R.C. Bioelectricity: A Quantitative Approach; Springer: New York, NY, USA, 2000; pp. 522–526. [Google Scholar]
  7. Perlman, O.; Katz, A.; Weissman, N.; Amit, G. Atrial electrical activity detection using linear combination of 12-lead ECG signals. IEEE Trans. Biomed. Eng. 2014, 61, 1034–1043. [Google Scholar] [CrossRef] [PubMed]
  8. Afonso, V.X.; Tompkins, W.J.; Nguyen, T.Q.; Luo, S. ECG beat detection using filter banks. IEEE Trans. Biomed. Eng. 1999, 46, 192–202. [Google Scholar] [CrossRef] [PubMed]
  9. Mateo, J.; Torres, A.; Soria, C.; Santos, J.L. A method for removing noise from continuous brain signal recordings. Comput. Electr. Eng. 2013, 39, 1561–1570. [Google Scholar] [CrossRef]
  10. Mateo, J.; Sanchez-Morla, E.M.; Santos, J.L. A new method for removal of powerline interference in ECG and EEG recordings. Comput. Electr. Eng. 2015, 45, 235–248. [Google Scholar] [CrossRef]
  11. Liu, S.; Jilai, L.U.; Hao, L.; Guangshu, H.U. Detection of QRS complex using mathematical morphology and wavelet transform. J. Tsinghua Univ. 2004, 44, 852–855. [Google Scholar]
  12. Vizireanu, D.N. Morphological shape decomposition interframe interpolation method. J. Electron. Imaging. 2008, 17, 1–5. [Google Scholar] [CrossRef]
  13. Szilagyi, S.M.; Benyo, Z.; Szilagyi, L.; David, L. Adaptive wavelet-transform-based ECG waveforms detection. In Proceedings of the 25th Annual International Conference of the IEEE Engineering in Medicine and Biology Society, Cancún, Mexico, 17–21 September 2003; pp. 2412–2415.
  14. Alfaouri, M.; Daqrouq, K. ECG signal denoising by wavelet transform thresholding. Am. J. Appl. Sci. 2008, 5, 276–281. [Google Scholar] [CrossRef]
  15. Zhao, Z.; Yang, L.; Chen, D.; Luo, Y. A human ECG identification system based on ensemble empirical mode decomposition. Sensors 2013, 13, 6832–6864. [Google Scholar] [CrossRef] [PubMed]
  16. Li, H.Q.; Wang, X.F. Detection of electrocardiogram characteristic points using lifting wavelet transform and Hilbert transform. Trans. Inst. Meas. Control. 2013, 35, 574–582. [Google Scholar] [CrossRef]
  17. Gao, J.; Sultan, H.; Hu, J.; Tung, W.W. Denoising nonlinear time series by adaptive filtering and wavelet shrinkage: A comparison. IEEE Signal Process. Lett. 2010, 17, 237–240. [Google Scholar]
  18. Li, H.Q.; Wang, X.F.; Chen, L.; Li, E.B. Denoising and R-Peak detection of electrocardiogram signal based on EMD and improved approximate envelope. Circ. Syst. Signal Process. 2014, 33, 1261–1276. [Google Scholar] [CrossRef]
  19. Ganeshkumar, R.; Yskumaraswamy, D. Investigating cardiac arrhythmia in ECG using random forest classification. Int. J. Comput. Appl. 2012, 37, 31–34. [Google Scholar] [CrossRef]
  20. Fei, S.W. Diagnostic study on arrhythmia cordis based on particle swarm optimization-based support vector machine. Expert Syst. Appl. 2010, 37, 6748–6752. [Google Scholar] [CrossRef]
  21. Liu, S.H.; Cheng, D.C.; Lin, C.M. Arrhythmia identification with two-lead electrocardiograms using artificial neural networks and support vector machines for a portable ECG monitor system. Sensors 2013, 13, 813–828. [Google Scholar] [CrossRef] [PubMed]
  22. Rai, H.M.; Trivedi, A.; Shukla, S. ECG signal processing for abnormalities detection using multi-resolution wavelet transform and artificial neural network classifier. Measurement 2013, 46, 3238–3246. [Google Scholar] [CrossRef]
  23. Daamouche, A.; Hamami, L.; Alajlan, N.; Melgani, F. A wavelet optimization approach for ECG signal classification. Biomed. Signal Process. Control 2012, 7, 342–349. [Google Scholar] [CrossRef]
  24. Lin, C.C.; Yang, C.M. Heartbeat classification using normalized RR intervals and morphological features. Math. Probl. Eng. 2014, 12, 1–11. [Google Scholar] [CrossRef]
  25. Dutta, S.; Chatterjee, A.; Munshi, S. Correlation technique and least square support vector machine combine for frequency domain based ECG beat classification. Med. Eng. Phys. 2010, 32, 1161–1169. [Google Scholar] [CrossRef] [PubMed]
  26. Kasturiwale, H.P.; Ingole, P.V. Component extraction of complex biomedical signals and performance analysis. Int. J. Comput. Sci. Inf. Technol. 2012, 3, 3544–3547. [Google Scholar]
  27. Martis, R.J.; Acharya, U.R.; Mandana, K.M. Cardiac decision making using higher order spectra. Biomed. Signal Process. 2013, 8, 193–203. [Google Scholar] [CrossRef]
  28. Valenza, G.; Citi, L.; Lanata, A.; Scilingo, E.P.; Barbieri, R. Revealing real-time emotional responses: A personalized assessment based on heartbeat dynamics. Sci. Rep. 2014, 4, 1–13. [Google Scholar] [CrossRef] [PubMed]
  29. Saini, I.; Singh, D.; Khosla, A. Electrocardiogram beat classification using empirical mode decomposition and multi-class directed acyclic graph support vector machine. Comput. Electr. Eng. 2014, 40, 1774–1787. [Google Scholar] [CrossRef]
  30. Kamath, C. ECG beat classification using features extracted from Teager energy functions in time and frequency domains. IET Signal Process. 2011, 5, 575–581. [Google Scholar] [CrossRef]
  31. Bach, F.R.; Jordan, M.I. Kernel independent component analysis. J. Mach. Learn. Res. 2003, 3, 1–48. [Google Scholar]
  32. Belhumeur, P.; Hespanha, J.; Kriegman, D. Eigenfaces vs. fisherfaces: Recognition using class specific linear projection. IEEE Trans. Pattern Anal. Mach. Intell. 1993, 19, 711–720. [Google Scholar] [CrossRef]
  33. Goldberg, D.E. Genetic Algorithms in Search, Optimization, and Machine Learning; Addison Wesley: Boston, MA, USA, 1989; pp. 2104–2116. [Google Scholar]
  34. Goldberger, A.L. Clinical Electrocardiography: A Simplified Approach; Elsevier: Amsterdam, The Netherlands, 2006. [Google Scholar]
  35. Donoho, D.L. De-noising by soft-thresholding. IEEE Trans. Inform. Theory 1995, 41, 613–627. [Google Scholar] [CrossRef]
  36. Mallat, S.G. A theory for multiresolution signal decomposition: The wavelet representation. IEEE Trans. Pattern Anal. Mach. Intell. 2010, 11, 674–693. [Google Scholar] [CrossRef]
  37. Jovic, A.; Bogunovic, N. Electrocardiogram analysis using a combination of statistical, geometric, and nonlinear heart rate variability features. Artif. Intell. Med. 2011, 51, 175–186. [Google Scholar] [CrossRef] [PubMed]
  38. Duda, R.O.; Hart, P.E.; Stork, D.G. Pattern Classification; Wiley: NewYork, NY, USA, 2001; pp. 119–131. [Google Scholar]
  39. Ubeyli, E.D. ECG beats classification using multiclass support vector machines with error correcting output codes. Digit. Signal Process. 2007, 17, 675–684. [Google Scholar] [CrossRef]
  40. Vapnik, V. Statistical Learning Theory; Wiley: NewYork, NY, USA, 1998. [Google Scholar]
  41. Chang, C.C.; Lin, C.J. LIBSVM: A library for support vector machines. ACM Trans. Intell. Syst. Technol. 2011, 2, 27. [Google Scholar] [CrossRef]
  42. Gamarra, A.; Quintero, M. Using genetic algorithm feature selection in neural classification systems for image pattern recognition. Ing. Investig. 2013, 33, 52–58. [Google Scholar]
  43. Huang, C.J. Using genetic algorithm optimization SVM to construction of investment model. Int. J. Digit. Content Technol. Appl. 2011, 5, 123–132. [Google Scholar]
  44. Zadeh, A.E.; Khazaee, A.; Ranaee, V. Classification of the electrocardiogram signals using supervised classifiers and efficient features. Comput. Meth. Program. Biomed. 2010, 99, 179–194. [Google Scholar] [CrossRef] [PubMed]
Figure 1. Block scheme of the proposed electrocardiogram (ECG) recognition system for ECG beats classification. The presented system is composed of ECG pre-processing, feature extraction and classification. ECG pre-processing removes noise and interference from original ECG beats. Feature extraction derives multi-domain features through kernel-independent component analysis (KICA) and discrete wavelet transform (DWT). The support vector machine (SVM) classifier, optimized with genetic algorithm (GA), divides ECG beats into five categories: normal beat (N), left bundle branch block beat (LBBB), right bundle branch block beat (RBBB), premature ventricular contraction (PVC) and atrial premature beat (APC).
Figure 1. Block scheme of the proposed electrocardiogram (ECG) recognition system for ECG beats classification. The presented system is composed of ECG pre-processing, feature extraction and classification. ECG pre-processing removes noise and interference from original ECG beats. Feature extraction derives multi-domain features through kernel-independent component analysis (KICA) and discrete wavelet transform (DWT). The support vector machine (SVM) classifier, optimized with genetic algorithm (GA), divides ECG beats into five categories: normal beat (N), left bundle branch block beat (LBBB), right bundle branch block beat (RBBB), premature ventricular contraction (PVC) and atrial premature beat (APC).
Sensors 16 01744 g001
Figure 2. Results of wavelet transform (WT) for ECG denoising. (a1a5) present the approximation coefficients of WT, and (d1d5) present the detail coefficients of WT.
Figure 2. Results of wavelet transform (WT) for ECG denoising. (a1a5) present the approximation coefficients of WT, and (d1d5) present the detail coefficients of WT.
Sensors 16 01744 g002
Figure 3. Denoising results of different threshold functions. (a) Original signal; (b) Noisy signal; (c) Signal denoised by the soft threshold function; (d) Signal denoised by the hard threshold function; and (e) Signal denoised by the improved threshold function.
Figure 3. Denoising results of different threshold functions. (a) Original signal; (b) Noisy signal; (c) Signal denoised by the soft threshold function; (d) Signal denoised by the hard threshold function; and (e) Signal denoised by the improved threshold function.
Sensors 16 01744 g003
Figure 4. Spectrum of different signals. (a) Spectrum of the original signal; (b) Spectrum of the noisy signal; (c) Spectrum of the signal denoised by the soft threshold function; (d) Spectrum of the signal denoised by the hard threshold function; (e) Spectrum of the signal denoised by the improved threshold function.
Figure 4. Spectrum of different signals. (a) Spectrum of the original signal; (b) Spectrum of the noisy signal; (c) Spectrum of the signal denoised by the soft threshold function; (d) Spectrum of the signal denoised by the hard threshold function; (e) Spectrum of the signal denoised by the improved threshold function.
Sensors 16 01744 g004
Figure 5. Feature subspace obtained through kernel-independent component analysis (KICA). (s1s20) are 20 independent base signals, x-axis represents sample points of the ECG signal and y-axis is the amplitude.
Figure 5. Feature subspace obtained through kernel-independent component analysis (KICA). (s1s20) are 20 independent base signals, x-axis represents sample points of the ECG signal and y-axis is the amplitude.
Sensors 16 01744 g005
Figure 6. Frequency domain features of five types of ECG beats obtained through DWT. s0 is the original ECG beat, ca4 is the approximation of the 4th level and cd1 to cd4 are the details of each level. (a) ECG of record 100 is used to represent N; (b) ECG of record 109 represents LBBB; (c) ECG of record 118 represents RBBB; (d) ECG of record 106 represents PVC; and (e) ECG of record 209 represents APC.
Figure 6. Frequency domain features of five types of ECG beats obtained through DWT. s0 is the original ECG beat, ca4 is the approximation of the 4th level and cd1 to cd4 are the details of each level. (a) ECG of record 100 is used to represent N; (b) ECG of record 109 represents LBBB; (c) ECG of record 118 represents RBBB; (d) ECG of record 106 represents PVC; and (e) ECG of record 209 represents APC.
Sensors 16 01744 g006
Figure 7. Diagrams of the ECG acquisition experimental platform. (a) Schematic of the experimental platform; and (b) Construction of the experimental platform.
Figure 7. Diagrams of the ECG acquisition experimental platform. (a) Schematic of the experimental platform; and (b) Construction of the experimental platform.
Sensors 16 01744 g007
Figure 8. Fitness curve of GA for finding the optimal parameters of SVM. Average fitness and best fitness are gradually increased via a series of iterations. When the evolution algebra is 200, average fitness and best fitness reach the maximum value, namely, the final optimization parameters of SVM are obtained.
Figure 8. Fitness curve of GA for finding the optimal parameters of SVM. Average fitness and best fitness are gradually increased via a series of iterations. When the evolution algebra is 200, average fitness and best fitness reach the maximum value, namely, the final optimization parameters of SVM are obtained.
Sensors 16 01744 g008
Figure 9. Five types of ECG signals acquired by the experimental platform. The ECG signals in the red dashed boxes are the five types of the acquired beats. (a) N; (b) LBBB; (c) RBBB; (d) PVC; and (e) APC.
Figure 9. Five types of ECG signals acquired by the experimental platform. The ECG signals in the red dashed boxes are the five types of the acquired beats. (a) N; (b) LBBB; (c) RBBB; (d) PVC; and (e) APC.
Sensors 16 01744 g009
Table 1. Performance indicators of three threshold denoising method.
Table 1. Performance indicators of three threshold denoising method.
MethodsPerformance IndicatorsECG Signal Records
103102118232231
Hard thresholdsignal-noise ratio (SNR)21.276920.432021.895718.109822.6279
root mean square error (RMSE)0.02550.02520.02570.03190.0221
Soft thresholdSNR21.745420.226822.103218.005722.5574
RMSE0.02410.02580.02510.03280.0223
Improved thresholdSNR24.062623.486924.012820.868025.2851
RMSE0.01850.01780.01790.02360.0163
Table 2. Classification results of the support vector machine (SVM) classifier.
Table 2. Classification results of the support vector machine (SVM) classifier.
TypeNLBBBRBBBPVCAPC
normal beat (N) 2000000
left bundle branch block beat (LBBB)0198020
right bundle branch block beat (RBBB)0020000
premature ventricular contraction (PVC)0501950
atrial premature beat (APC)300196
Table 3. Statistical performance indicators of the SVM classifier: sensitivity (Se), specificity (Sp), and positive predictability (Pp).
Table 3. Statistical performance indicators of the SVM classifier: sensitivity (Se), specificity (Sp), and positive predictability (Pp).
TypeSensitivity (Se)Specificity (Sp)Positive Predictability (Pp)
normal beat (N) 100%99.57%98.52%
LBBB99%99.29%97.54%
RBBB100%100%100%
PVC97.50%99.57%98.48%
APC96%100%100%
Average98.50%99.69%98.91%
Accuracy(Acc) 98.80%
Table 4. The classification results based on electrocardiogram (ECG) acquisition experiment platform.
Table 4. The classification results based on electrocardiogram (ECG) acquisition experiment platform.
TypeNLBBBRBBBPVCAPC
N1952012
LBBB4194020
RBBB0019910
PVC1821890
APC100099
Table 5. The performance statistical indicators of the experiment results: sensitivity (Se) and specificity (Sp), and positive predictability (Pp).
Table 5. The performance statistical indicators of the experiment results: sensitivity (Se) and specificity (Sp), and positive predictability (Pp).
TypeSeSpPp
N97.5%99.14%97.01%
LBBB97%98.57%95.10%
RBBB99.50%99.71%99%
PVC94.50%99.43%97.93%
APC99%99.75%98.02%
Average97.50%99.32%97.41%
Acc 97.30%
Table 6. Comparison results of the proposed system with other literatures.
Table 6. Comparison results of the proposed system with other literatures.
MethodsClassifierClassesAccuracyReference
Principal components of bispectrum featuresleast squares support vector machine (LSSVM)593.48%Martis et al.
Teager energy function featuresneural network (NN)595%Kamath
Morphological and time features support vector machine (SVM)397.14%Zadeh et al.
Time intervalsSVM595.65%Fei
R-R intervalsself-constructing neural fuzzy interference network (SoNFIN)596.4%Liu
The multi-domain featuresa library for SVM (LIBSVM)598.8%Proposed method

Share and Cite

MDPI and ACS Style

Li, H.; Yuan, D.; Wang, Y.; Cui, D.; Cao, L. Arrhythmia Classification Based on Multi-Domain Feature Extraction for an ECG Recognition System. Sensors 2016, 16, 1744. https://doi.org/10.3390/s16101744

AMA Style

Li H, Yuan D, Wang Y, Cui D, Cao L. Arrhythmia Classification Based on Multi-Domain Feature Extraction for an ECG Recognition System. Sensors. 2016; 16(10):1744. https://doi.org/10.3390/s16101744

Chicago/Turabian Style

Li, Hongqiang, Danyang Yuan, Youxi Wang, Dianyin Cui, and Lu Cao. 2016. "Arrhythmia Classification Based on Multi-Domain Feature Extraction for an ECG Recognition System" Sensors 16, no. 10: 1744. https://doi.org/10.3390/s16101744

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop