Comparison the Performance of Classification Methods for Diagnosis of Heart Disease and Chronic Conditions

Jiarui Si,Haohan Zou,Chuanyi Huang,Huan Feng,Honglin Liu,Guangyu Li,Shuaijun Hu,Hong Zhang,Xin Wang
DOI: https://doi.org/10.1007/978-3-030-91560-5_10
2021-01-01
Abstract:In this paper, we aim to identify the best ratio of training set, and to evaluate the diagnostic performance of eight classical machine learning methods for chronic diseases. The five categories of chronic disease datasets were collected from UCI and GitHub database include heart disease, breast cancer, diabetic retinopathy, Parkinson's disease and diabetes. Machine learning (ML) methods including six individual learners (logistic regression (LR), BP neural network (BP), learning vector quantization (LVQ), extreme learning machine(ELM), support vector machine (SVM), decision trees (DTs)), and two ensemble learning methods were implemented using BP_Adaboost (Ada) and random forest (RF). We used five indicators, including AUC value, accuracy, sensitivity, specificity and running time to compare the behaviors of each algorithm. The ensemble learning methods were the most suited for the diagnosis of chronic diseases and led to a significant enhancement of performance compared to individual learners.
What problem does this paper attempt to address?