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