DISEASE PREDICTION USING ARTIFICIAL NEURAL NETWORKS
Abstract
In recent times, the medical world has been dealing with a very serious challenge detecting diseases early enough before they become life-threatening. Many patients, especially in developing countries like Nigeria, only come to know they have a chronic disease when it has already progressed to an advanced and harder-to-treat stage. This is mostly because traditional methods of diagnosis rely heavily on doctors and laboratory technicians manually reviewing test results, a process that can be slow, expensive, and sometimes prone to human error, especially in environments with limited medical personnel or equipment.
This work investigates how Artificial Neural Networks (ANN), which is a core technique in the field of machine learning and artificial intelligence, can be trained to automatically predict whether a patient has diabetes based on certain routine medical examination inputs. The work makes use of the widely accepted Pima Indians Diabetes Dataset, which contains the medical records of female patients of Pima Indian heritage, including features such as glucose concentration, blood pressure, body mass index (BMI), age, insulin level, skin thickness, number of pregnancies, and a diabetes pedigree function score. The ANN model was designed with three hidden layers, batch normalization, and dropout regularization to ensure it generalized well on unseen data. After complete training and evaluation on the test portion of the dataset, the model achieved a Test Accuracy of 73.38%, Precision of 59.70%, Recall (Sensitivity) of 74.07%, an F1-Score of 66.12%, and an AUC-ROC score of 0.798. The relatively high Recall score is especially important in a medical context because it means the model successfully identified the majority of actual diabetic patients. These results confirm that Artificial Neural Networks are a highly promising tool for disease prediction and early diagnosis in clinical settings. The simulation was implemented entirely in Python using TensorFlow and Keras, and all results are verifiable from the charts included in this report.
References
Goodfellow, I., Bengio, Y., &Courville, A. (2016). Deep learning. MIT Press. https://www.deeplearningbook.org
International Diabetes Federation. (2021). IDF diabetes atlas (10th ed.). International Diabetes Federation. https://diabetesatlas.org
Kavakiotis, I., Tsave, O., Salifoglou, A., Maglaveras, N., Vlahavas, I., &Chouvarda, I. (2017).Machine learning and data mining methods in diabetes research. Computational and Structural Biotechnology Journal, 15, 104–116. https://doi.org/10.1016/j.csbj.2016.12.005
Noor, Sahar & Khan, Mohammed & Hussain, Iqbal & Khan, Dr. Afzal & Syed, Riz & Shah, S.W & Babar, M.I. (2009). Application of a Hybrid Artificial Neural Networks Model to a Scheduling Policy System. Journal of Engineering and Applied Sciences. 28. 89-102.
Luckyn, Boma,Josiah , Alabere ,Idayana & Ogra , O. Emmanuella (2024) Predictive modeling for diaper sales in retail: an artificial neural network approach - Iconic Research and Engineering Journals, 7(8) 283 -288
Obermeyer, Z., & Emanuel, E. J. (2016). Predicting the future — big data, machine learning, and clinical medicine. The New England Journal of Medicine, 375(13), 1216–1219. https://doi.org/10.1056/NEJMp1606181
Pima Indians Diabetes Database. (1988). UCI Machine Learning Repository. Donated by Smith, J. W., Everhart, J. E., Dickson, W. C., Knowler, W. C., & Johannes, R. S. Retrieved from https://www.kaggle.com/datasets/uciml/pima-indians-diabetes-database
Schmidhuber, J. (2015). Deep learning in neural networks: An overview. Neural Networks, 61, 85–117. https://doi.org/10.1016/j.neunet.2014.09.003
Sisodia, D., &Sisodia, D. S. (2018). Prediction of diabetes using classification algorithms. Procedia Computer Science, 132, 1578–1585. https://doi.org/10.1016/j.procs.2018.05.122
Smith, J. W., Everhart, J. E., Dickson, W. C., Knowler, W. C., & Johannes, R. S. (1988). Using the ADAP learning algorithm to forecast the onset of diabetes mellitus. Proceedings of the Annual Symposium on Computer Application in Medical Care, 261–265.
Refbacks
- There are currently no refbacks.