Enhancing machine learning-based forecasting of chronic renal disease with explainable AI

Sanjana Singamsetty, Swetha Ghanta, Sujit Biswas*, Ashok Pradhan*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract


Chronic renal disease (CRD) is a significant concern in the field of healthcare, highlighting the crucial need of early and accurate prediction in order to provide prompt treatments and enhance patient outcomes. This article presents an end-to-end predictive model for the binary classification of CRD in healthcare, addressing the crucial need for early and accurate predictions to enhance patient outcomes. Through hyperparameter optimization using GridSearchCV, we significantly improve model performance. Leveraging a range of machine learning (ML) techniques, our approach achieves a high predictive accuracy of 99.07% for random forest, extra trees classifier, logistic regression with L2 penalty, and artificial neural networks (ANN). Through rigorous evaluation, the logistic regression with L2 penalty emerges as the top performer, demonstrating consistent performance. Moreover, integration of Explainable Artificial Intelligence (XAI) techniques, such as Local Interpretable Model-agnostic Explanations (LIME) and SHapley Additive exPlanations (SHAP), enhances interpretability and reveals insights into model decision-making. By emphasizing an end-to-end model development process, from data collection to deployment, our system enables real-time predictions and informed healthcare decisions. This comprehensive approach underscores the potential of predictive modeling in healthcare to optimize clinical decision-making and improve patient care outcomes.
Original languageEnglish
Article numbere2291
Number of pages27
JournalPeerJ Computer Science
Volume10
DOIs
Publication statusPublished - 26 Sept 2024
Externally publishedYes

Keywords

  • chronic renal disease
  • machine learning
  • explainable AI
  • GridSearchCV
  • chronic kidney disease

Cite this