Data-Driven Predictive Modeling for Early Diagnosis and Risk Stratification of Chronic Diseases

Main Article Content

Md Jubayar Hossain
Mohammed Majid Bakhsh
Muhammad Adnan
Evha Rozario
Sraboni Clara Mohonta
Tushar Roy
Md Imtiaz Faruk
Emran Hossain

Abstract

Chronic kidney disease (CKD) often remains undetected until advanced stages, limiting opportunities for timely intervention and increasing the burden of complications and treatment. This study proposes a data-driven hybrid ResNet50-SVM framework for the early diagnosis and risk stratification of CKD. Clinical data were prepared through missing-value treatment, data cleaning, Min-Max normalization, reduction of redundant information, and optimized feature selection. The processed dataset was divided into training and testing subsets using an 80:20 ratio. ResNet50 was employed to extract high-level representations, while a support vector machine performed the final classification. Model performance was evaluated using accuracy, precision, recall, F1-score, confusion-matrix analysis, and training and validation behavior. The proposed framework achieved an accuracy of 98.76%, precision of 97.65%, recall of 99.21%, and F1-score of 98.34%. It also outperformed logistic regression, a convolutional neural network, and a decision tree across the reported evaluation metrics. The high recall indicates a strong capacity to identify patients with disease while reducing missed positive cases. These findings demonstrate the potential of hybrid deep-learning and machine-learning methods to support accurate, scalable, and timely CKD screening. Further validation using larger, more diverse clinical datasets and explainable AI methods is needed before deployment in real-world clinical decision-support systems.

Article Details

How to Cite
Hossain, M. J., Bakhsh, M. M., Adnan, M., Rozario, E., Mohonta, S. C., Roy, T., … Hossain, E. (2024). Data-Driven Predictive Modeling for Early Diagnosis and Risk Stratification of Chronic Diseases. The Eastasouth Journal of Information System and Computer Science, 2(01), 162–172. https://doi.org/10.58812/esiscs.v2i01.1220
Section
Articles

References

[1] S. S. Shambharkar, P. S. Moon, P. A. Bainalwar, and S. M. Boarkar, “Machine Learning-Based Approach for Early Detection and Prediction of Chronic Diseases,” in 2023 1st DMIHER International Conference on Artificial Intelligence in Education and Industry 4.0 (IDICAIEI), IEEE, Nov. 2023, pp. 1–8. doi: 10.1109/IDICAIEI58380.2023.10406914.

[2] S. Singamsetty, “Neurofusion: Advancing Alzheimer’s Diagnosis with Deep Learning and Multimodal Feature Integration,” Int. J. Educ. Appl. Sci. Res., vol. 8, no. 1, pp. 23–32, 2021, doi: 10.5281/zenodo.14889013.

[3] B. Khan, R. Naseem, F. Muhammad, G. Abbas, and S. Kim, “An empirical evaluation of machine learning techniques for chronic kidney disease prophecy,” IEEE access, vol. 8, pp. 55012–55022, 2020, doi: 10.1109/ACCESS.2020.2981689.

[4] M. F. Alam and M. F. Alam, “Predictive Artificial Intelligence Models for Early Detection and Management of Chronic Diseases To Strengthen National Healthcare Resilience in the United States,” Am. J. Interdiscip. Stud., vol. 03, no. 04, pp. 268–293, 2022, doi: 10.63125/9t5ar104.

[5] X. Yuan, S. Chen, C. Sun, and L. Yuwen, “A novel early diagnostic framework for chronic diseases with class imbalance.,” Sci. Rep., vol. 12, no. 1, p. 8614, May 2022, doi: 10.1038/s41598-022-12574-x.

[6] S. Nusinovici et al., “Logistic regression was as good as machine learning for predicting major chronic diseases,” J. Clin. Epidemiol., vol. 122, pp. 56–69, 2020, doi: 10.1016/j.jclinepi.2020.03.002.

[7] S. Agarwal, C. Prabha, and M. Gupta, “Chronic diseases prediction using machine learning – a review,” Ann. Rom. Soc. Cell Biol., vol. 25, no. 1, pp. 3495–3511, 2021.

[8] R. Alanazi, “Identification and Prediction of Chronic Diseases Using Machine Learning Approach,” J. Healthc. Eng., vol. 2022, 2022, doi: 10.1155/2022/2826127.

[9] J. Antony, P. Palsuk, S. Gupta, D. Mishra, and P. Barach, “Six Sigma in healthcare: a systematic review of the literature,” Int. J. Qual. Reliab. Manag., vol. 35, no. 5, pp. 1075–1092, 2018, doi: 10.1108/IJQRM-02-2017-0027.

[10] F. M. Delpino, Â. K. Costa, S. R. Farias, A. D. P. Chiavegatto Filho, R. A. Arcêncio, and B. P. Nunes, “Machine learning for predicting chronic diseases: a systematic review,” Public Health, vol. 205, pp. 14–25, Apr. 2022, doi: 10.1016/j.puhe.2022.01.007.

[11] V. Varma, “Data Analytics for Predictive Maintenance for Business Intelligence for Operational Efficiency,” Asian J. Comput. Sci. Eng., vol. 7, no. 4, pp. 1–7, 2022.

[12] S. P. Abhang et al., “Machine Learning-Based Monitoring and Prognosis of Chronic Kidney Disease Patients,” in 2023 International Conference on Artificial Intelligence for Innovations in Healthcare Industries (ICAIIHI), 2023, pp. 1–6. doi: 10.1109/ICAIIHI57871.2023.10489422.

[13] Z. Nenova and J. Shang, “Personalized Chronic Disease Follow‐Up Appointments: Risk‐Stratified Care Through Big Data,” Prod. Oper. Manag., vol. 31, no. 2, pp. 583–606, 2022, doi: 10.1111/poms.13568.

[14] J. J. Boutilier, T. C. Y. Chan, M. Ranjan, and S. Deo, “Risk Stratification for Early Detection of Diabetes and Hypertension in Resource-Limited Settings: Machine Learning Analysis,” J. Med. Internet Res., vol. 23, no. 1, 2021, doi: 10.2196/20123.

[15] P. Ghosh, F. M. Javed Mehedi Shamrat, S. Shultana, S. Afrin, A. A. Anjum, and A. A. Khan, “Optimization of Prediction Method of Chronic Kidney Disease Using Machine Learning Algorithm,” in 2020 15th International Joint Symposium on Artificial Intelligence and Natural Language Processing (iSAI-NLP), 2020, pp. 1–6. doi: 10.1109/iSAI-NLP51646.2020.9376787.

[16] P. Anandajayam, S. Aravindkumar, P. Arun, and A. Ajith, “Prediction of Chronic Disease by Machine Learning,” in 2019 IEEE International Conference on System, Computation, Automation and Networking (ICSCAN), 2019, pp. 1–6. doi: 10.1109/ICSCAN.2019.8878724.

[17] Z. Ullah et al., “Detecting High-Risk Factors and Early Diagnosis of Diabetes Using Machine Learning Methods,” Comput. Intell. Neurosci., vol. 2022, 2022, doi: 10.1155/2022/2557795.

[18] M. Pal, S. Parija, and G. Panda, “Improved prediction of diabetes mellitus using machine learning based approach,” in 2nd International Conference on Range Technology, ICORT 2021, 2021. doi: 10.1109/ICORT52730.2021.9581774.

[19] M. R. Narasinga Rao, T. Sajana, N. Bhavana, M. Sai Ram, and C. Nikhil Krishna, “Prediction of chronic kidney disease using machine learning technique,” J. Adv. Res. Dyn. Control Syst., vol. 10, no. 10, pp. 328–332, 2018, doi: 10.17148/IJARCCE.2018.71021.