Congestive heart failure detection using random forest classifier

• Published in: Computer methods and programs in biomedicine Vol. 130; pp. 54 - 64
• Main Authors: Masetic, Zerina, Subasi, Abdulhamit
• Format: Journal Article
• Language: English
• Published: Ireland Elsevier Ireland Ltd 01.07.2016
• Subjects: Accuracy; Algorithms; Automation; Autoregressive (AR) modeling; Classification; Classifiers; Congestive heart failure (CHF); Decision trees; Echocardiography; Electrocardiogram (ECG); Electrocardiography; Failure; Feature extraction; Heart Failure - diagnosis; Humans; Internal Medicine; Machine learning; Other; Random forest; Support Vector Machine; Autoregressive (AR) modeling; Random forest; Congestive heart failure (CHF); Electrocardiogram (ECG); Machine learning; Congestive Heart Failure (CHF); Autoregressive (AR) Modeling; Machine Learning; Random Forest
• ISSN: 0169-2607; 1872-7565
• DOI: 10.1016/j.cmpb.2016.03.020

•Heartbeat classification is substantial for diagnosing heart failure.•Machine learning methods classify normal and congestive heart failure (CHF).•The random forest method gives 100% classification accuracy in detecting CHF. Automatic electrocardiogram (ECG) heartbeat classification is substantial for diagnosing heart failure. The aim of this paper is to evaluate the effect of machine learning methods in creating the model which classifies normal and congestive heart failure (CHF) on the long-term ECG time series. The study was performed in two phases: feature extraction and classification phase. In feature extraction phase, autoregressive (AR) Burg method is applied for extracting features. In classification phase, five different classifiers are examined namely, C4.5 decision tree, k-nearest neighbor, support vector machine, artificial neural networks and random forest classifier. The ECG signals were acquired from BIDMC Congestive Heart Failure and PTB Diagnostic ECG databases and classified by applying various experiments. The experimental results are evaluated in several statistical measures (sensitivity, specificity, accuracy, F-measure and ROC curve) and showed that the random forest method gives 100% classification accuracy. Impressive performance of random forest method proves that it plays significant role in detecting congestive heart failure (CHF) and can be valuable in expressing knowledge useful in medicine.