DOI QR코드

DOI QR Code

Random Forest Based Abnormal ECG Dichotomization using Linear and Nonlinear Feature Extraction

선형-비선형 특징추출에 의한 비정상 심전도 신호의 랜덤포레스트 기반 분류

  • Kim, Hye-Jin (Department of Medical Engineering, College of Medicine, Yonsei University) ;
  • Kim, Byeong-Nam (Department of Medical Engineering, College of Medicine, Yonsei University) ;
  • Jang, Won-Seuk (Department of Medical Engineering, College of Medicine, Yonsei University) ;
  • Yoo, Sun-K. (Department of Medical Engineering, College of Medicine, Yonsei University)
  • 김혜진 (연세대학교 의과대학 의학공학교실) ;
  • 김병남 (연세대학교 의과대학 의학공학교실) ;
  • 장원석 (연세대학교 의과대학 의학공학교실) ;
  • 유선국 (연세대학교 의과대학 의학공학교실)
  • Received : 2015.10.15
  • Accepted : 2016.03.23
  • Published : 2016.04.30

Abstract

This paper presented a method for random forest based the arrhythmia classification using both heart rate (HR) and heart rate variability (HRV) features. We analyzed the MIT-BIH arrhythmia database which contains half-hour ECG recorded from 48 subjects. This study included not only the linear features but also non-linear features for the improvement of classification performance. We classified abnormal ECG using mean_NN (mean of heart rate), SD1/SD2 (geometrical feature of poincare HRV plot), SE (spectral entropy), pNN100 (percentage of a heart rate longer than 100 ms) affecting accurate classification among combined of linear and nonlinear features. We compared our proposed method with Neural Networks to evaluate the accuracy of the algorithm. When we used the features extracted from the HRV as an input variable for classifier, random forest used only the most contributed variable for classification unlike the neural networks. The characteristics of random forest enable the dimensionality reduction of the input variables, increase a efficiency of classifier and can be obtained faster, 11.1% higher accuracy than the neural networks.

Keywords

References

  1. U.R. ACHARYA, K.P. JOSEPH, N. KANNATHAL, C.M. LIM and J.S. SURI, "Heart rate variability: a review", Medical and Biological Engineering and Computing, vol. 44, no. 12, pp. 1031-1051, 2006. https://doi.org/10.1007/s11517-006-0119-0
  2. E. ZELLMER, F. SHANG and H. ZHANG, "Highly accurate ECG beat classification based on continuous wavelet transformation and multiple support vector machine classifiers", In Proceeding of the 2009 2th International Conference on Biomedical Engineering and Informatics (BMEI 2009), 2009, pp. 1-5.
  3. M. ENGIN, "ECG beat classification using neuro-fuzzy network", Pattern Recognition Letters, vol. 25, no. 15, pp. 1715-1722, 2004. https://doi.org/10.1016/j.patrec.2004.06.014
  4. Y.-C. YEH, W.-J. WANG and C.W. CHIOU, "Cardiac arrhythmia diagnosis method using linear discriminant analysis on ECG signals", Measurement, vol. 42, no. 5, pp. 778-789, 2009. https://doi.org/10.1016/j.measurement.2009.01.004
  5. I. CHRISTOV, G. G MEZ-HERRERO, V. KRASTEVA, I. JEKOVA, A. GOTCHEV and K. EGIAZARIAN, "Comparative study of morphological and time-frequency ECG descriptors for heartbeat classification", Medical Engineering & Physics, vol. 28, no. 9, pp. 876-887, 2006. https://doi.org/10.1016/j.medengphy.2005.12.010
  6. B. ANURADHA, K.S. KUMAR and V.V. REDDY, "Classification of cardiac signals using time domain methods", ARPN Journal of Engineering and Applied Sciences, vol. 3, no. 3, pp. 7-12, 2008.
  7. R. ACHARYA, A. KUMAR, P. BHAT, C. LIM, S. LYENGAR, N. KANNATHAL and S. KRISHNAN, "Classification of cardiac abnormalities using heart rate signals", Medical and Biological Engineering and Computing, vol. 42, no. 3, pp. 288-293, 2004. https://doi.org/10.1007/BF02344702
  8. M. OWIS, A.H. ABOU-ZIED, A.-B.M. YOUSSEF and Y.M. KADAH, "Study of features based on nonlinear dynamical modeling in ECG arrhythmia detection and classification", IEEE Transactions on Biomedical Engineering, vol. 49, no. 7, pp. 733-736, 2002. https://doi.org/10.1109/TBME.2002.1010858
  9. E. YlLMAZ, "An expert system based on Fisher score and LS-SVM for cardiac arrhythmia diagnosis", Computational and Mathematical Methods in Medicine, vol. 2013, 2013.
  10. J.V. TU, "Advantages and disadvantages of using artificial neural networks versus logistic regression for predicting medical outcomes", Journal of Clinical Epidemiology, vol. 49, no. 11, pp. 1225-1231, 1996. https://doi.org/10.1016/S0895-4356(96)00002-9
  11. L. BREIMAN, "Random forests", Machine Learning, vol. 45, no. 1, pp. 5-32, 2001. https://doi.org/10.1023/A:1010933404324
  12. G.B. MOODY and R.G. MARK, "The impact of the MITBIH arrhythmia database", IEEE Engineering in Medicine and Biology Magazine, vol. 20, no. 3, pp. 45-50, 2001. https://doi.org/10.1109/51.932724
  13. J. PAN and W.J. TOMPKINS, "A real-time QRS detection algorithm", IEEE Transactions on Biomedical Engineering, no. 3, pp. 230-236, 1985.
  14. G.C. CASOLO, P. STRODER, C. SIGNORINI, F. CALZOLARI, M. ZUCCHINI, E. BALLI, A. SULLA and S. LAZZERINI, "Heart rate variability during the acute phase of myocardial infarction", Circulation, vol. 85, no. 6, pp. 2073-2079, 1992. https://doi.org/10.1161/01.CIR.85.6.2073
  15. T.F.O.T.E.S.O. CARDIOLOGY, "Heart rate variability standards of measurement, physiological interpretation, and clinical use", European Heart Journal, vol. 17, pp. 354-381, 1996. https://doi.org/10.1093/oxfordjournals.eurheartj.a014868
  16. J. MIETUS, C. PENG, I. HENRY, R. GOLDSMITH and A. GOLDBERGER, "The pNNx files: re-examining a widely used heart rate variability measure", Heart, vol. 88, no. 4, pp. 378-380, 2002. https://doi.org/10.1136/heart.88.4.378
  17. R. ACHARYA, S.M. KRISHNAN, J.A. SPAAN and J.S. SURI, Advances in cardiac signal processing, Springer, 2007.
  18. C. KAMATH, "Quantification of electrocardiogram rhythmicity to detect life threatening cardiac arrhythmias using spectral entropy", Journal of Engineering Science and Technology, vol. 8, pp. 588-602, 2013.
  19. J. PISKORSKI and P. GUZIK, "Filtering poincare plots", Computational Methods in Science and Technology, vol. 11, no. 1, pp. 39-48, 2005. https://doi.org/10.12921/cmst.2005.11.01.39-48
  20. M. BRENNAN, M. PALANISWAMI and P. KAMEN, "Do existing measures of Poincare plot geometry reflect nonlinear features of heart rate variability?", IEEE Transactions on Biomedical Engineering, vol. 48, no. 11, pp. 1342-1347, 2001. https://doi.org/10.1109/10.959330
  21. M.G. TSIPOURAS, D.I. FOTIADIS and D. SIDERIS, "An arrhythmia classification system based on the RR-interval signal", Artificial Intelligence in Medicine, vol. 33, no. 3, pp. 237-250, 2005. https://doi.org/10.1016/j.artmed.2004.03.007
  22. M.G. TSIPOURAS and D.I. FOTIADIS, "Automatic arrhythmia detection based on time and time-frequency analysis of heart rate variability", Computer Methods and Programs in Biomedicine, vol. 74, no. 2, pp. 95-108, 2004. https://doi.org/10.1016/S0169-2607(03)00079-8