Interpretable Machine Learning for Chronic Kidney Disease Diagnosis: A Gaussian Processes Approach
DOI:
https://doi.org/10.60084/ijcr.v2i1.204Keywords:
Artificial intelligence, Medical diagnostics, Clinical decision support, SHAPAbstract
Chronic Kidney Disease (CKD) is a global health issue impacting over 800 million people, characterized by a gradual loss of kidney function leading to severe complications. Traditional diagnostic methods, relying on laboratory tests and clinical assessments, have limitations in sensitivity and are prone to human error, particularly in the early stages of CKD. Recent advances in machine learning (ML) offer promising tools for disease diagnosis, but a lack of interpretability often hinders their adoption in clinical practice. Gaussian Processes (GP) provide a flexible ML model capable of delivering predictions and uncertainty estimates, essential for high-stakes medical applications. However, the integration of GP with interpretable methods remains underexplored. We developed an interpretable CKD classification model to address this knowledge gap by combining GP with Shapley Additive Explanations (SHAP). We assessed the model's performance using three GP kernels (Radial Basis Function, Matern, and Rational Quadratic). The results show that the Rational Quadratic kernel outperforms the other kernels, achieving an accuracy of 98.75%, precision of 100%, sensitivity of 97.87%, specificity of 100%, and an F1-score of 98.51%. SHAP values indicate that haemoglobin and specific gravity are the most influential features. The results demonstrate that the Rational Quadratic kernel enhances predictive accuracy and provides robust uncertainty estimates and interpretable explanations. This combination of accuracy and interpretability supports clinicians in making informed decisions and improving patient management and outcomes in CKD. Our study connects advanced ML techniques with practical medical applications, leading to more effective and reliable ML-driven healthcare solutions.
Downloads
References
- Kovesdy, C. P. (2022). Epidemiology of Chronic Kidney Disease: An Update 2022, Kidney International Supplements, Vol. 12, No. 1, 7–11. doi:10.1016/j.kisu.2021.11.003.
- Levey, A. S., and Coresh, J. (2012). Chronic Kidney Disease, The Lancet, Vol. 379, No. 9811, 165–180. doi:10.1016/S0140-6736(11)60178-5.
- Chen, T. K., Knicely, D. H., and Grams, M. E. (2019). Chronic Kidney Disease Diagnosis and Management, JAMA, Vol. 322, No. 13, 1294. doi:10.1001/jama.2019.14745.
- Levey, A., and Inker, L. (2017). Assessment of Glomerular Filtration Rate in Health and Disease: A State of the Art Review, Clinical Pharmacology & Therapeutics, Vol. 102, No. 3, 405–419. doi:10.1002/cpt.729.
- Joo, Y. S., Rim, T. H., Koh, H. B., Yi, J., Kim, H., Lee, G., Kim, Y. A., Kang, S.-W., Kim, S. S., and Park, J. T. (2023). Non-invasive Chronic Kidney Disease Risk Stratification Tool Derived from Retina-Based Deep Learning and Clinical Factors, Npj Digital Medicine, Vol. 6, No. 1, 114. doi:10.1038/s41746-023-00860-5.
- Chouhan, A. S., Kaple, M., and Hingway, S. (2023). A Brief Review of Diagnostic Techniques and Clinical Management in Chronic Kidney Disease, Cureus. doi:10.7759/cureus.49030.
- Arif, M. S., Mukheimer, A., and Asif, D. (2023). Enhancing the Early Detection of Chronic Kidney Disease: A Robust Machine Learning Model, Big Data and Cognitive Computing, Vol. 7, No. 3, 144. doi:10.3390/bdcc7030144.
- Akazawa, M., Hashimoto, K., Noda, K., and Yoshida, K. (2021). The Application of Machine Learning for Predicting Recurrence in Patients with Early-Stage Endometrial Cancer: A Pilot Study, Obstetrics & Gynecology Science, Vol. 64, No. 3, 266–273. doi:10.5468/ogs.20248.
- Noviandy, T. R., Nainggolan, S. I., Raihan, R., Firmansyah, I., and Idroes, R. (2023). Maternal Health Risk Detection Using Light Gradient Boosting Machine Approach, Infolitika Journal of Data Science, Vol. 1, No. 2, 48–55. doi:10.60084/ijds.v1i2.123.
- Solomon, D. D., Khan, S., Garg, S., Gupta, G., Almjally, A., Alabduallah, B. I., Alsagri, H. S., Ibrahim, M. M., and Abdallah, A. M. A. (2023). Hybrid Majority Voting: Prediction and Classification Model for Obesity, Diagnostics, Vol. 13, No. 15, 2610. doi:10.3390/diagnostics13152610.
- Maulana, A., Faisal, F. R., Noviandy, T. R., Rizkia, T., Idroes, G. M., Tallei, T. E., El-Shazly, M., and Idroes, R. (2023). Machine Learning Approach for Diabetes Detection Using Fine-Tuned XGBoost Algorithm, Infolitika Journal of Data Science, Vol. 1, No. 1, 1–7. doi:10.60084/ijds.v1i1.72.
- Ali, L., Niamat, A., Khan, J. A., Golilarz, N. A., Xingzhong, X., Noor, A., Nour, R., and Bukhari, S. A. C. (2019). An Optimized Stacked Support Vector Machines Based Expert System for the Effective Prediction of Heart Failure, IEEE Access, Vol. 7, 54007–54014. doi:10.1109/ACCESS.2019.2909969.
- Ahmad, N., Ul-Saufie, A. Z., Shaziayani, W. N., Abidin, A. W. Z., Zulazmi, N. E. S., and Harb, S. M. (2022). Evaluating the Performance of Random Forest and Multiple Linear Regression for Higher Observed PM10Concentrations, Israa University Journal of Applied Science, Vol. 6, No. 1, 72–90. doi:10.52865/WHPM9019.
- Noviandy, T. R., Maulana, A., Idroes, G. M., Emran, T. B., Tallei, T. E., Helwani, Z., and Idroes, R. (2023). Ensemble Machine Learning Approach for Quantitative Structure Activity Relationship Based Drug Discovery: A Review, Infolitika Journal of Data Science, Vol. 1, No. 1, 32–41. doi:10.60084/ijds.v1i1.91.
- Qin, J., Chen, L., Liu, Y., Liu, C., Feng, C., and Chen, B. (2020). A Machine Learning Methodology for Diagnosing Chronic Kidney Disease, IEEE Access, Vol. 8, 20991–21002. doi:10.1109/ACCESS.2019.2963053.
- Bai, Q., Su, C., Tang, W., and Li, Y. (2022). Machine Learning to Predict End Stage Kidney Disease in Chronic Kidney Disease, Scientific Reports, Vol. 12, No. 1, 8377. doi:10.1038/s41598-022-12316-z.
- Mustafizur Rahman, M., Al-Amin, M., and Hossain, J. (2024). Machine Learning Models for Chronic Kidney Disease Diagnosis and Prediction, Biomedical Signal Processing and Control, Vol. 87, 105368. doi:10.1016/j.bspc.2023.105368.
- Williams, C. K. I., and Barber, D. (1998). Bayesian Classification with Gaussian Processes, IEEE Transactions on Pattern Analysis and Machine Intelligence, Vol. 20, No. 12, 1342–1351.
- Liu, M., Chowdhary, G., Da Silva, B. C., Liu, S.-Y., and How, J. P. (2018). Gaussian Processes for Learning and Control: A Tutorial with Examples, IEEE Control Systems Magazine, Vol. 38, No. 5, 53–86.
- Stiglic, G., Kocbek, P., Fijacko, N., Zitnik, M., Verbert, K., and Cilar, L. (2020). Interpretability of Machine Learning-Based Prediction Models in Healthcare, WIREs Data Mining and Knowledge Discovery, Vol. 10, No. 5. doi:10.1002/widm.1379.
- Vellido, A. (2020). The Importance of Interpretability and Visualization in Machine Learning for Applications in Medicine and Health Care, Neural Computing and Applications, Vol. 32, No. 24, 18069–18083. doi:10.1007/s00521-019-04051-w.
- Lundberg, S. M., and Lee, S.-I. (2017). A Unified Approach to Interpreting Model Predictions, Advances in Neural Information Processing Systems, Vol. 30.
- Yang, R., Xiong, X., Wang, H., and Li, W. (2022). Explainable Machine Learning Model to Prediction EGFR Mutation in Lung Cancer, Frontiers in Oncology, Vol. 12. doi:10.3389/fonc.2022.924144.
- Prendin, F., Pavan, J., Cappon, G., Del Favero, S., Sparacino, G., and Facchinetti, A. (2023). The Importance of Interpreting Machine Learning Models for Blood Glucose Prediction in Diabetes: An Analysis Using SHAP, Scientific Reports, Vol. 13, No. 1, 16865. doi:10.1038/s41598-023-44155-x.
- Lu, S., Chen, R., Wei, W., Belovsky, M., and Lu, X. (2021). Understanding Heart Failure Patients EHR Clinical Features via SHAP Interpretation of Tree-Based Machine Learning Model Predictions., AMIA ... Annual Symposium Proceedings. AMIA Symposium, Vol. 2021, 813–822.
- Rubini, L., Soundarapandian, P., and Eswaran, P. (2015). Chronic Kidney Disease, UCI Machine Learning Repository, from https://archive.ics.uci.edu/dataset/336/chronic+kidney+disease.
- Noviandy, T. R., Nisa, K., Idroes, G. M., Hardi, I., and Sasmita, N. R. (2024). Classifying Beta-Secretase 1 Inhibitor Activity for Alzheimer’s Drug Discovery with LightGBM, Journal of Computing Theories and Applications, Vol. 2, No. 2, 138–147. doi:10.62411/jcta.10129.
- Noviandy, T. R., Idroes, G. M., and Hardi, I. (2024). Enhancing Loan Approval Decision-Making: An Interpretable Machine Learning Approach Using LightGBM for Digital Economy Development, Malaysian Journal of Computing (MJOC), Vol. 9, No. 1, 1734–1745. doi:10.24191/mjoc.v9i1.25691.
- Rasmussen, C. E., and Nickisch, H. (2010). Gaussian Processes for Machine Learning (GPML) Toolbox, The Journal of Machine Learning Research, Vol. 11, 3011–3015.
- Corbière, C., Thome, N., Bar-Hen, A., Cord, M., and Pérez, P. (2019). Addressing Failure Prediction by Learning Model Confidence, Advances in Neural Information Processing Systems, Vol. 32.
- Moncada-Torres, A., van Maaren, M. C., Hendriks, M. P., Siesling, S., and Geleijnse, G. (2021). Explainable Machine Learning Can Outperform Cox Regression Predictions and Provide Insights in Breast Cancer Survival, Scientific Reports, Vol. 11, No. 1, 6968. doi:10.1038/s41598-021-86327-7.
- Idroes, G. M., Noviandy, T. R., Maulana, A., Zahriah, Z., Suhendrayatna, S., Suhartono, E., Khairan, K., Kusumo, F., Helwani, Z., and Abd Rahman, S. (2023). Urban Air Quality Classification Using Machine Learning Approach to Enhance Environmental Monitoring, Leuser Journal of Environmental Studies, Vol. 1, No. 2, 62–68. doi:10.60084/ljes.v1i2.99.
- Noviandy, T. R., Idroes, G. M., and Hardi, I. (2024). Machine Learning Approach to Predict AXL Kinase Inhibitor Activity for Cancer Drug Discovery Using XGBoost and Bayesian Optimization, Journal of Soft Computing and Data Mining, Vol. 5, No. 1, 46–56.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 Teuku Rizky Noviandy, Ghifari Maulana Idroes, Maimun Syukri, Rinaldi Idroes
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.