Efficient prediction of early-stage diabetes using XGBoost classifier with random forest feature selection technique

• Published in: Multimedia tools and applications Vol. 82; no. 22; pp. 34163 - 34181
• Main Author: Gündoğdu, Serdar
• Format: Journal Article
• Language: English
• Published: New York Springer US 01.09.2023; Springer Nature B.V
• Subjects: Accuracy; Algorithms; Classifiers; Computer Communication Networks; Computer Science; Data Structures and Information Theory; Diabetes; Diagnosis; Feature selection; Multimedia Information Systems; Special Purpose and Application-Based Systems; COVID-19; Feature selection; MLR; Random forest; Diabetes; XGBoost
• ISSN: 1380-7501; 1573-7721
• DOI: 10.1007/s11042-023-15165-8

Diabetes is one of the most common and serious diseases affecting human health. Early diagnosis and treatment are vital to prevent or delay complications related to diabetes. An automated diabetes detection system assists physicians in the early diagnosis of the disease and reduces complications by providing fast and precise results. This study aims to introduce a technique based on a combination of multiple linear regression (MLR), random forest (RF), and XGBoost (XG) to diagnose diabetes from questionnaire data. MLR-RF algorithm is used for feature selection, and XG is used for classification in the proposed system. The dataset is the diabetic hospital data in Sylhet, Bangladesh. It contains 520 instances, including 320 diabetics and 200 control instances. The performance of the classifiers is measured concerning accuracy (ACC), precision (PPV), recall (SEN, sensitivity), F1 score (F1), and the area under the receiver-operating-characteristic curve (AUC). The results show that the proposed system achieves an accuracy of 99.2%, an AUC of 99.3%, and a prediction time of 0.04825 seconds. The feature selection method improves the prediction time, although it does not affect the accuracy of the four compared classifiers. The results of this study are quite reasonable and successful when compared with other studies. The proposed method can be used as an auxiliary tool in diagnosing diabetes.