Publications

Journal Article Epublish 2 citations

PubMed: 39650439 DOI: 10.7717/peerj-cs.2291

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

PeerJ Comput Sci 10:e2291

Authors: Sanjana Singamsetty, Swetha Ghanta, Sujit Biswas, Ashok Pradhan

Affiliations

  • Department of Computer Science and Engineering, School of Engineering and Sciences, SRM University, AP, Guntur, Andhra Pradesh, India.
  • Computer Science Department, Northumbria University, Newcastle, United Kingdom.
  • Computer Science, University of London, London, United Kingdom.

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.

MeSH

Machine Learning, Learning, Artificial Intelligence

Cited by