Clustering MIT–BIH arrhythmias with Ant Colony Optimization using time domain and PCA compressed wavelet coefficients

Abstract

In this paper, Ant Colony Optimization (ACO) based clustering analysis of ECG arrhythmias taken from the MIT-BIH Arrhythmia Database is proposed. Both time domain and discrete wavelet transform (DWT) based frequency domain features are used in the analysis. Since the number of wavelet coefficients are huge amount as compared to the time domain parameters, Principal Component Analysis (PCA) based compression is applied on them in order to decrease their number to the number of time domain features. Then, the reduced numbers of frequency parameters are combined with the time domain features, in order to get the total feature sets. Different types of feature sets are tried and the classification results are compared. These are: time domain feature set, frequency domain feature set and the mixture of them. A neural network algorithm is developed in parallel to verify and measure the ACO classifier's success. Moreover, linear discriminant analysis (LDA) is used to show the effect of clustering on the system's results. The method is tested with MIT-BIH database to classify normal beats and five different critical and having vital importance arrhythmia types. Chosen six classes are normal sinus rhythm, premature ventricular contraction (PVC), atrial premature contraction (APC), right bundle branch block (RBBB), ventricular fusion (F) and fusion (f). Comparison results indicate that the mixture feature set gave a better success for the classification.