Journal Article Epublish 21 citations
PubMed: 38039306 DOI: 10.1371/journal.pone.0295234
Chronic kidney disease prediction using boosting techniques based on clinical parameters.
PLoS One 18(12):e0295234
Authors: Shahid Mohammad Ganie, Pijush Kanti Dutta Pramanik, Saurav Mallik, Zhongming Zhao
Affiliations
- AI Research Centre, School of Business, Woxsen University, Hyderabad, Telangana, India.
- School of Computing Science and Engineering, Galgotias University, Greater Noida, Uttar Pradesh, India.
- Department of Environmental Health, Harvard T. H. Chan School of Public Health, Boston, MA, United States of America.
- Center for Precision Health, School of Biomedical Informatics, The University of Texas Health Science Center at Houston, Houston, TX, United States of America.
Abstract
Chronic kidney disease (CKD) has become a major global health crisis, causing millions of yearly deaths. Predicting the possibility of a person being affected by the disease will allow timely diagnosis and precautionary measures leading to preventive strategies for health. Machine learning techniques have been popularly applied in various disease diagnoses and predictions. Ensemble learning approaches have become useful for predicting many complex diseases. In this paper, we utilise the boosting method, one of the popular ensemble learnings, to achieve a higher prediction accuracy for CKD. Five boosting algorithms are employed: XGBoost, CatBoost, LightGBM, AdaBoost, and gradient boosting. We experimented with the CKD data set from the UCI machine learning repository. Various preprocessing steps are employed to achieve better prediction performance, along with suitable hyperparameter tuning and feature selection. We assessed the degree of importance of each feature in the dataset leading to CKD. The performance of each model was evaluated with accuracy, precision, recall, F1-score, Area under the curve-receiving operator characteristic (AUC-ROC), and runtime. AdaBoost was found to have the overall best performance among the five algorithms, scoring the highest in almost all the performance measures. It attained 100% and 98.47% accuracy for training and testing sets. This model also exhibited better precision, recall, and AUC-ROC curve performance.
MeSH
Kidney Diseases, Mental Recall, Kidney, Algorithms, Diagnosis, Learning
Cited by
- Enhancing machine learning-based forecasting of chronic renal disease with explainable AI. PeerJ Comput Sci, 2024
- Assessment of skin fibrosis in a murine model of systemic sclerosis with multifunctional optical coherence tomography. J Biomed Opt, 2025
- Enhanced and Interpretable Prediction of Multiple Cancer Types Using a Stacking Ensemble Approach with SHAP Analysis. Bioengineering (Basel), 2025
- Ensemble learning with explainable AI for improved heart disease prediction based on multiple datasets. Sci Rep, 2025
- Advancing ensemble learning techniques for residential building electricity consumption forecasting: Insight from explainable artificial intelligence. PLoS One, 2024