Diabetes Mellitus Disease Prediction using Machine Learning Algorithms
DOI:
https://doi.org/10.26418/justin.v12i4.84620Keywords:
Diabetes, Random Forest, Logistic Regression, Decision TreeAbstract
Diabetes mellitus is a chronic disease with a rapidly increasing global prevalence, affecting around 422 million people, predominantly in low- and middle-income countries. Effective management of diabetes requires early detection and timely intervention. This study aims to develop an accurate predictive model for diabetes mellitus using three machine learning algorithms: Random Forest, Logistic Regression, and Decision Tree. The Pima Indians Diabetes dataset, comprising 768 patient records with various health indicators, was utilized for model training and evaluation. Exploratory data analysis revealed significant correlations between glucose levels, BMI, age, and diabetes risk. The dataset was split into 80% training and 20% testing sets. Models were validated using cross-validation and evaluated based on accuracy, precision, recall, and F1-score. Results indicated that Logistic Regression achieved the highest accuracy (75%) and balanced performance in identifying both positive and negative cases. Decision Tree excelled in recall, while Random Forest showed a slightly lower balance between precision and recall. The ROC curve analysis demonstrated that Random Forest had the highest AUC (0.82), followed by Logistic Regression (0.81) and Decision Tree (0.73). This study confirms that machine learning algorithms can effectively predict diabetes, providing valuable tools for early detection and intervention, ultimately reducing the global burden of diabetes mellitus.References
WHO, “World Health Organization Diabetes,†2023. Accessed: Jul. 12, 2024. [Online]. Available: https://www.who.int/news-room/fact-sheets/detail/diabetes
F. Sugandh et al., “Advances in the Management of Diabetes Mellitus: A Focus on Personalized Medicine,†Cureus, 2023, doi: 10.7759/cureus.43697.
International Diabetes Federation, IDF Diabetes Atlas 2021 _ IDF Diabetes Atlas. 2021.
N. Ali, W. Khan, A. Ahmad, M. M. Masud, H. Adam, and L. A. Ahmed, “Predictive Modeling for the Diagnosis of Gestational Diabetes Mellitus Using Epidemiological Data in the United Arab Emirates,†Information (Switzerland), vol. 13, no. 10, 2022, doi: 10.3390/info13100485.
Y. Mao et al., “Value of machine learning algorithms for predicting diabetes risk: A subset analysis from a real-world retrospective cohort study,†J Diabetes Investig, vol. 14, no. 2, 2023, doi: 10.1111/jdi.13937.
E. K. Oikonomou and R. Khera, “Machine learning in precision diabetes care and cardiovascular risk prediction,†2023. doi: 10.1186/s12933-023-01985-3.
Y. Qin et al., “Machine Learning Models for Data-Driven Prediction of Diabetes by Lifestyle Type,†Int J Environ Res Public Health, vol. 19, no. 22, 2022, doi: 10.3390/ijerph192215027.
J. J. Khanam and S. Y. Foo, “A comparison of machine learning algorithms for diabetes prediction,†ICT Express, vol. 7, no. 4, 2021, doi: 10.1016/j.icte.2021.02.004.
S. Mehrolia, S. Alagarsamy, and V. M. Solaikutty, “Customers response to online food delivery services during COVID-19 outbreak using binary logistic regression,†Int J Consum Stud, vol. 45, no. 3, 2021, doi: 10.1111/ijcs.12630.
M. Schonlau and R. Y. Zou, “The random forest algorithm for statistical learning,†Stata Journal, vol. 20, no. 1, 2020, doi: 10.1177/1536867X20909688.
L. Breiman and A. Cutler, “Breiman and Cutler’s random forests for classification and regression,†Package “randomForest,†2012.
A. Seraj et al., “Cross-validation,†in Handbook of HydroInformatics: Volume I: Classic Soft-Computing Techniques, 2022. doi: 10.1016/B978-0-12-821285-1.00021-X.
M. Phongying and S. Hiriote, “Diabetes Classification Using Machine Learning Techniques,†Computation, vol. 11, no. 5, 2023, doi: 10.3390/computation11050096.
U. e. Laila, K. Mahboob, A. W. Khan, F. Khan, and W. Taekeun, “An Ensemble Approach to Predict Early-Stage Diabetes Risk Using Machine Learning: An Empirical Study,†Sensors, vol. 22, no. 14, 2022, doi: 10.3390/s22145247.
M. Rashedul Islam, M. Begum, and M. Nasim Akhtar, “Recursive approach for multiple step-ahead software fault prediction through long short-term memory (LSTM),†Journal of Discrete Mathematical Sciences and Cryptography, vol. 25, no. 7, 2022, doi: 10.1080/09720529.2022.2133251.
J. W. Smith, J. E. Everhart, W. C. Dickson, W. C. Knowler, and R. S. Johannes, “Using the ADAP learning algorithm to forecast the onset of diabetes mellitus,†in Proceedings - Annual Symposium on Computer Applications in Medical Care, 1988.
Jacob Shreffler and Martin R. Huecker., “Exploratory Data Analysis: Frequencies, Descriptive Statistics, Histograms, and Boxplots.†Accessed: Jul. 11, 2024. [Online]. Available: https://www.ncbi.nlm.nih.gov/books/NBK557570/
Downloads
Published
Issue
Section
License
Copyright (c) 2024 JUSTIN (Jurnal Sistem dan Teknologi Informasi)
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
The author owns the copyright in his paper and agrees to publish his paper to JUSTIN by giving the rights to the first publication of his paper which is simultaneously licensed under the Creative Commons Attribution License, namely the Similar International 4.0 license (CC BY-NC-SA 4.0).
This is a human-readable summary of (and not a substitute for) the license. Disclaimer.
You are free to:Share "” copy and redistribute the material in any medium or format
Adapt "” remix, transform, and build upon the material
The licensor cannot revoke these freedoms as long as you follow the license terms.
Under the following terms:
Attribution "” You must give appropriate credit, provide a link to the license, and indicate if changes were made. You may do so in any reasonable manner, but not in any way that suggests the licensor endorses you or your use.
NonCommercial "” You may not use the material for commercial purposes.
ShareAlike "” If you remix, transform, or build upon the material, you must distribute your contributions under the same license as the original.
No additional restrictions "” You may not apply legal terms or technological measures that legally restrict others from doing anything the license permits.
Notices:
You do not have to comply with the license for elements of the material in the public domain or where your use is permitted by an applicable exception or limitation.
No warranties are given. The license may not give you all of the permissions necessary for your intended use. For example, other rights such as publicity, privacy, or moral rights may limit how you use the material.
