ARRHYTHMIA DISEASE DIAGNOSIS USING NEURAL NETWORK, SVM, AND GENETIC ALGORITHM-OPTIMIZED k-MEANS CLUSTERING

被引:17
作者
Martis, Roshan Joy [1 ]
Chakraborty, Chandan [1 ]
机构
[1] Indian Inst Technol, Sch Med Sci & Technol, Kharagpur 721302, W Bengal, India
关键词
ECG; MIT-BIH database; arrhythmia; MIT-BIH normal sinus rhythm; PCA; k-means; neural network; support vector machine; genetic algorithm;
D O I
10.1142/S0219519411004101
中图分类号
Q6 [生物物理学];
学科分类号
071011 ;
摘要
This work aims at presenting a methodology for electrocardiogram (ECG)-based arrhythmia disease detection using genetic algorithm (GA)-optimized k-means clustering. The open-source ECG data from MIT-BIH arrhythmia database and MIT-BIH normal sinus rhythm database are subjected to a sequence of steps including segmentation using R-point detection, extraction of features using principal component analysis (PCA), and pattern classification. Here, the classical classifiers viz., k-means clustering, error back propagation neural network (EBPNN), and support vector machine (SVM) have been initially attempted and subsequently m-fold (m = 3) cross validation is used to reduce the bias during training of the classifier. The average classification accuracy is computed as the average over all the three folds. It is observed that EBPNN and SVM with different order polynomial kernel provide significant accuracies in comparison with k-means one. In fact, the parameters (centroids) of k-means algorithm are locally optimized by minimizing its objective function. In order to overcome this limitation, a global optimization technique viz., GA is suggested here and implemented to find more robust parameters of k-means clustering. Finally, it is shown that GA-optimized k-means algorithm enhances its accuracy to those of other classifiers. The results are discussed and compared. It is concluded that the GA-optimized k-means algorithm is an alternate approach for classification whose accuracy will be near to that of supervised (viz., EBPNN and SVM) classifiers.
引用
收藏
页码:897 / 915
页数:19
相关论文
共 26 条
  • [1] [Anonymous], SOFT COMPUTING KNOWL
  • [2] [Anonymous], 2001, Pattern Classification
  • [3] [Anonymous], 1988, Algorithms for Clustering Data
  • [4] Banik S, 2004, P IEEE ENG MED BIOL, V1, P160
  • [5] Bishop CM., 1995, NEURAL NETWORKS PATT
  • [6] Chong E., 2001, INTRO OPTIMIZATION
  • [7] Christianini N., 2000, INTRO SUPPORT VECTOR, P189
  • [8] Deb K., 2001, Multi-objective Optimization Using Evolutionary Algorithms
  • [9] Fauci A., 2008, Harrison's Principles of Internal Medicine
  • [10] Goldberg DavidE., 2008, GENETIC ALGORITHMS S