Diabetes mellitus prediction using machine learning within the scope of a generic framework

Nidhi Arora, Shilpa Srivastava, Ritu Agarwal, Vandana Mehndiratta, Aprna Tripathi

Abstract


Artificial intelligence (AI) based automated disease prediction has recently taken a significant place in the field of health informatics. However, due to unavailability of real time large scale medical data, the dynamic learning of prediction models remains principally subsided. This paper, therefore proposes a dynamic predictive modelling framework for chronic diseases prediction in real-time. The framework premise suggests creation of a centralized patient-indexed medical database to dynamically train machine learning (ML) models and predict risk levels of chronic diseases in real time. In this study, comprehensive empirical evaluations to train seven state-of-the-art ML models for diabetes risk prediction are performed in context of phase 2 of the suggested framework. The selected optimal model can then be dynamically applied to predict diabetes in phase 3 of the framework. Various metrics such as accuracy, precision, Recall, F1-score and receiver operating characteristic (ROC) curve are employed for evaluating performances of the trained models. Parameter tunings using different type of kernels, different number of neighbors and estimators are rigorously performed in order to create a suggestive literature for healthcare prediction ecosystem. Comparative analysis indicates high prediction accuracies on diabetes test data records for neural network and support vector machine (SVM) models as compared to other applied models.

Keywords


Chronic diseases; Classification; Diabetes; Machine learning; Predictive modelling

Full Text:

PDF


DOI: http://doi.org/10.11591/ijeecs.v32.i3.pp1724-1735

Refbacks

  • There are currently no refbacks.


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

Indonesian Journal of Electrical Engineering and Computer Science (IJEECS)
p-ISSN: 2502-4752, e-ISSN: 2502-4760
This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).

shopify stats IJEECS visitor statistics