An Augmented Artificial Intelligence Approach for Chronic Diseases Prediction.

Journal: Frontiers in public health
Published Date:

Abstract

Chronic diseases are increasing in prevalence and mortality worldwide. Early diagnosis has therefore become an important research area to enhance patient survival rates. Several research studies have reported classification approaches for specific disease prediction. In this paper, we propose a novel augmented artificial intelligence approach using an artificial neural network (ANN) with particle swarm optimization (PSO) to predict five prevalent chronic diseases including breast cancer, diabetes, heart attack, hepatitis, and kidney disease. Seven classification algorithms are compared to evaluate the proposed model's prediction performance. The ANN prediction model constructed with a PSO based feature extraction approach outperforms other state-of-the-art classification approaches when evaluated with accuracy. Our proposed approach gave the highest accuracy of 99.67%, with the PSO. However, the classification model's performance is found to depend on the attributes of data used for classification. Our results are compared with various chronic disease datasets and shown to outperform other benchmark approaches. In addition, our optimized ANN processing is shown to require less time compared to random forest (RF), deep learning and support vector machine (SVM) based methods. Our study could play a role for early diagnosis of chronic diseases in hospitals, including through development of online diagnosis systems.

Authors

  • Junaid Rashid
    Department of Computer Science and Engineering, Kongju National University, Gongju 31080, Chungcheongnam-do, Korea.
  • Saba Batool
    Department of Computer Science, COMSATS University Islamabad, Islamabad, Pakistan.
  • Jungeun Kim
    Department of Computer Science and Engineering, Kongju National University, Gongju 31080, Chungcheongnam-do, Korea.
  • Muhammad Wasif Nisar
    Department of Computer Science, COMSATS University Islamabad, Islamabad, Pakistan.
  • Amir Hussain
    Cognitive Signal-Image and Control Processing Research Laboratory, School of Natural Sciences, University of Stirling, Stirling, FK9 4LA, United Kingdom.
  • Sapna Juneja
    IMS Engineering College, Ghaziabad, India.
  • Riti Kushwaha
    Department of Computer Science, Bennett University, Greater Noida, India.