An Efficient Machine Learning and Deep Learning Framework for Early Prediction and Risk Assessment of Chronic Diseases
DOI:
https://doi.org/10.29070/ykshsf64Keywords:
Chronic Disease Prediction, Deep Learning, Machine Learning, Chronic Kidney Disease , Heart Disease PredictionAbstract
Timely identification of chronic illnesses is imperative in minimizing death rates, enhancing patient care, and facilitating early clinical treatment. This paper presents an effective machine learning and deep learning framework to early-stage predict the Chronic Kidney Disease (CKD) and chronic heart disease using structured clinical data available in the UCI Machine Learning Repository. The main aim of the study is to establish the perfect predictive models combined with the risk evaluation and personalized recommendation systems in the context of intelligent healthcare support.The suggested methodology will involve data preprocessing, missing value imputation, feature scaling, exploratory data analysis, feature selection and stratified data splitting. Several machine learning models were used to predict CKD including Naive Bayes, K-Nearest Neighbors, Decision Tree, Logistic Regression, Support Vector Machine models, and random forest models and Hybrid Ensemble models. Moreover, a more sophisticated CNN-UDRP deep learning network with embedding layers, convolutional neural network, and latent feature extraction via PCA was created. Deep ANN and Advanced DNN with LeakyReLU architectures were applied to predict chronic heart diseases with the help of batch normalization and dropout regularization.Experimental findings prove that the Random Forest model provided the highest performance of CKD prediction with the highest accuracy, precision, recall and F1-score of 100, whereas the Advanced DNN with LeakyReLU provided the highest accuracy of heart disease prediction at 99.93. The suggested framework also incorporates risk stratification and individual health recommendations, thus it can be used in scalable and intelligent healthcare decision-support systems.
Downloads
References
1. A. Ashwini, V. Chirchi, S. Balasubramaniam, and M. A. Shah, “Bio inspired optimization techniques for disease detection in deep learning systems,” Sci. Rep., vol. 15, no. 1, pp. 1–30, 2025, doi: 10.1038/s41598-025-02846-7.
2. A. I. Torre-Bastida, J. Díaz-de-Arcaya, E. Osaba, K. Muhammad, D. Camacho, and J. Del Ser, Bio-inspired computation for big data fusion, storage, processing, learning and visualization: state of the art and future directions, vol. 37, no. 28. 2025. doi: 10.1007/s00521-021-06332-9.
3. T. Banerjee and İ. Paçal, “A systematic review of machine learning in heart disease prediction,” Turkish J. Biol., vol. 49, no. SI-5, pp. 600–634, 2025, doi: 10.55730/1300-0152.2766.
4. R. Islam, A. Sultana, and M. R. Islam, “A comprehensive review for chronic disease prediction using machine learning algorithms,” J. Electr. Syst. Inf. Technol., vol. 11, no. 1, 2024, doi: 10.1186/s43067-024-00150-4.
5. S. Tutun et al., “An AI-based Decision Support System for Predicting Mental Health Disorders,” Inf. Syst. Front., vol. 25, no. 3, pp. 1261–1276, 2023, doi: 10.1007/s10796-022-10282-5.
6. A. A. Almazroi, E. A. Aldhahri, S. Bashir, and S. Ashfaq, “A Clinical Decision Support System for Heart Disease Prediction Using Deep Learning,” IEEE Access, vol. 11, no. June, pp. 61646–61659, 2023, doi: 10.1109/ACCESS.2023.3285247.
7. M. Gholamzadeh, H. Abtahi, and R. Safdari, “The Application of Knowledge-Based Clinical Decision Support Systems to Enhance Adherence to Evidence-Based Medicine in Chronic Disease,” J. Healthc. Eng., vol. 2023, 2023, doi: 10.1155/2023/8550905.
8. M. Brkan, “Artificial Intelligence and Judicial Decision-Making,” Eur. Data Prot. Law Rev., vol. 9, no. 3, pp. 290–295, 2023, doi: 10.21552/edpl/2023/3/5.
9. J. Chung and J. Teo, “Mental Health Prediction Using Machine Learning: Taxonomy, Applications, and Challenges,” Appl. Comput. Intell. Soft Comput., vol. 2022, 2022, doi: 10.1155/2022/9970363.
10. V. Bansla, “A Machine Learning Framework for Early Prediction of Chronic Diseases,” SAMRIDDHI A J. Phys. Sci. Eng. Technol., vol. 17, no. 02, pp. 4–7, 2025, doi: 10.18090/samriddhi.v17i02.02.
11. M. Harishbhai Tilala et al., “Ethical Considerations in the Use of Artificial Intelligence and Machine Learning in Health Care: A Comprehensive Review,” Cureus, vol. 16, no. 6, 2024, doi: 10.7759/cureus.62443.
12. S. A. Marrah et al., “Federated deep learning for privacy-preserving disease detection in IoT-enabled healthcare systems,” Front. Comput. Sci., vol. 8, no. February, 2026, doi: 10.3389/fcomp.2026.1725597.
13. “A Machine Learning – Driven Health Risk Index for Predicting Chronic Disease Burden,” no. April, 2026, doi: 10.14293/PR2199.003245.v2.
14. D. V. S. Kumar, R. Chaurasia, A. Misra, P. K. Misra, and A. Khang, “Heart Disease and Liver Disease Prediction Using Machine Learning,” Data-Centric AI Solut. Emerg. Technol. Healthc. Ecosyst., vol. 14, no. 04, pp. 205–214, 2023, doi: 10.1201/9781003356189-13.
15. R. Alanazi, “Identification and Prediction of Chronic Diseases Using Machine Learning Approach,” J. Healthc. Eng., vol. 2022, 2022, doi: 10.1155/2022/2826127.
16. Z. Xia, P. Xu, Y. Xiong, Y. Lai, and Z. Huang, “Survival Prediction in Patients with Hypertensive Chronic Kidney Disease in Intensive Care Unit: A Retrospective Analysis Based on the MIMIC-III Database,” J. Immunol. Res., vol. 2022, 2022, doi: 10.1155/2022/3377030.
17. D. S. Khafaga, N. Khodadadi, E. Khodadadi, A. Ali Alhussan, M. M. Eid, and E. S. M. El-Kenawy, “Enhanced early chronic kidney disease prediction using hybrid waterwheel plant algorithm for deep neural network optimization,” Sci. Rep., vol. 15, no. 1, pp. 1–23, 2025, doi: 10.1038/s41598-025-26382-6.
18. S. K. Ghosh, N. Widatalla, and A. H. Khandoker, “Machine Learning Framework for Early Detection of Chronic Kidney Disease Stages Using Optimized Estimated Glomerular Filtration Rate,” IEEE Access, vol. 13, no. February, pp. 78057–78072, 2025, doi: 10.1109/ACCESS.2025.3565549.
19. D. Saif, A. M. Sarhan, and N. M. Elshennawy, “Early prediction of chronic kidney disease based on ensemble of deep learning models and optimizers,” J. Electr. Syst. Inf. Technol., vol. 11, no. 1, 2024, doi: 10.1186/s43067-024-00142-4.
20. S. Dey Tusar et al., “Advancing Chronic Kidney Disease Prediction Through Machine Learning and Deep Learning With Feature Analysis,” Front. Heal. Informatics, vol. 13, no. 3, pp. 11338–11348, 2024, [Online]. Available: https://doi.org/10.52783/fhi.vi.1079
21. X. Wang et al., “Securing Federated Learning With Blockchain in the Medical Field: Systematic Literature Review,” J. Med. Internet Res., vol. 28, 2026, doi: 10.2196/79052.
22. S. K. Murarka, N. Jain, V. Sharma, A. Malhosia, and V. K. Chaurasiya, “ML and AI Predictive Modeling and Continuous Patient Surveillance for Chronic Disease Prevention Using Electronic Health Records and IoT-Based Data Streams,” Int. J. Drug Deliv. Technol., vol. 16, no. 8, pp. 138–153, 2026, doi: 10.25258/ijddt.16.8s.17.
23. L. Zhang et al., “A Novel Prediction Model for Multimodal Medical Data Based on Graph Neural Networks,” Mach. Learn. Knowl. Extr., vol. 7, no. 3, pp. 1–17, 2025, doi: 10.3390/make7030092.
24. E. Afrifa-Yamoah et al., “Pathways to chronic disease detection and prediction: Mapping the potential of machine learning to the pathophysiological processes while navigating ethical challenges,” Chronic Dis. Transl. Med., vol. 11, no. 1, pp. 1–21, 2025, doi: 10.1002/cdt3.137.
25. M. J. Sci Ind Res, V. Bhatnagar, S. Sharma, S. Nisha Bhagirath, and D. Sharma, “Mangalayatan Journal of A Machine Learning Framework for Chronic Kidney Disease Analysis Using ORANGE Tool,” vol. 1, no. 1, 2024.
26. D. Saif, A. M. Sarhan, and N. M. Elshennawy, “Deep-kidney: an effective deep learning framework for chronic kidney disease prediction,” Heal. Inf. Sci. Syst., vol. 12, no. 1, pp. 1–22, 2024, doi: 10.1007/s13755-023-00261-8.
27. J. Zhu, Y. Wang, S. Wang, and J. Zhou, “Development and validation of a BMI stratified mortality prediction model for patients with COPD complicated by HF using the MIMIC-IV database,” Sci. Rep., vol. 15, no. 1, pp. 1–15, 2025, doi: 10.1038/s41598-025-09605-8.
28. M. Elhaddad and S. Hamam, “AI-Driven Clinical Decision Support Systems: An Ongoing Pursuit of Potential,” Cureus, vol. 16, no. 4, 2024, doi: 10.7759/cureus.57728.
29. M. S. Al Reshan et al., “An Innovative Ensemble Deep Learning Clinical Decision Support System for Diabetes Prediction,” IEEE Access, vol. 12, no. March, pp. 106193–106210, 2024, doi: 10.1109/ACCESS.2024.3436641.
30. J. L. Cross, M. A. Choma, and J. A. Onofrey, “Bias in medical AI: Implications for clinical decision-making,” PLOS Digit. Heal., vol. 3, no. 11, pp. 1–19, 2024, doi: 10.1371/journal.pdig.0000651.
31. J. N. Saxena, D. V. P. Potturu, and A. Nagraj, “A Hybrid Machine Learning Framework for Early Prediction of Chronic Kidney Disease Progression Using Longitudinal Claims Data : An XGBoost- LSTM Ensemble with Temporal Attention,” 2026.
32. M. Al Ghamdi and S. Alyahyan, “Enhanced Chronic Kidney Disease Detection: A Hybrid Deep Learning Framework Using Clinical Biomarkers and Ensemble Feature Engineering with DeepCKD-Net,” Appl. Sci., vol. 16, no. 6, pp. 1–32, 2026, doi: 10.3390/app16063024.
33. S. T. Miller, K. A. Logan, R. Anderson, P. E. Cowell, C. Busby-Earle, and L. D. Morris, “Multi-perspective machine learning MPML: A high-performance and interpretable ensemble method for heart disease prediction,” Mach. Learn. with Appl., vol. 23, p. 100836, 2026, doi: 10.1016/j.mlwa.2026.100836.
34. R. P. Bhatta and A. Husain, “Analysis of Machine Learning Models for Heart Disease Prediction and Diagnosis NPRC Journal of Multidisciplinary Research,” vol. 3, no. 1, pp. 63–79, 2026, doi: 10.3126/nprcjmr.v3i1.90028.
35. A. K. Parashar, A. Jamliya, S. Nasrat, R. Soni, H. Disease, and P. Achieving, “XGBoost for Heart Disease Prediction Achieving High Accuracy with Robust Machine Learning Techniques,” pp. 185–191, 2025, doi: 10.69968/ijisem.2025v4i3185-191.
36. M. Ahmed and I. Husien, “Heart Disease Prediction Using Hybrid Machine Learning: A Brief Review,” J. Robot. Control, vol. 5, no. 3, pp. 884–892, 2024, doi: 10.18196/jrc.v5i3.21606.
37. G. Manikandan, B. Pragadeesh, V. Manojkumar, A. L. Karthikeyan, R. Manikandan, and A. H. Gandomi, “Classification models combined with Boruta feature selection for heart disease prediction,” Informatics Med. Unlocked, vol. 44, no. January, p. 101442, 2024, doi: 10.1016/j.imu.2023.101442.
38. N. N. Reddy, L. Nipun, U. Baba, and N. Rishindra, “Optimizing heart disease prediction through ensemble and hybrid machine learning techniques,” vol. 14, no. 5, pp. 5744–5754, 2024, doi: 10.11591/ijece.v14i5.pp5744-5754.
39. K. Sumwiza, C. Twizere, G. Rushingabigwi, and P. Bakunzibake, “Informatics in Medicine Unlocked Enhanced cardiovascular disease prediction model using random forest algorithm,” Informatics Med. Unlocked, vol. 41, no. March, p. 101316, 2023, doi: 10.1016/j.imu.2023.101316.
40. L. Kumar, C. Anitha, V. N. Ghodke, N. Nithya, and V. A. Drave, “EAI Endorsed Transactions Deep Learning Based Healthcare Method for Effective Heart Disease Prediction,” vol. 9, 2023, doi: 10.4108/eetpht.9.4283.
