Deep learning infused SIRVD model for COVID-19 prediction: XGBoost-SIRVD-LSTM approach

被引:2
作者
Alkhalefah, Hisham [1 ]
Preethi, D. [2 ]
Khare, Neelu [3 ]
Abidi, Mustufa Haider [1 ]
Umer, Usama [1 ]
机构
[1] King Saud Univ, Adv Mfg Inst, Riyadh, Saudi Arabia
[2] SRM Inst Sci & Technol, Fac Engn & Technol, Dept Comp Sci & Engn, Chennai, Tamil Nadu, India
[3] Vellore Inst Technol, Sch Comp Sci Engn & Informat Syst SCORE, Vellore, Tamil Nadu, India
关键词
deep learning; extreme gradient boosting (XGBoost); susceptible-infected-recovered-vaccination-deceased (SIRVD); long short-term memory (LSTM); feature selection; COVID-19; prediction;
D O I
10.3389/fmed.2024.1427239
中图分类号
R5 [内科学];
学科分类号
1002 ; 100201 ;
摘要
The global impact of the ongoing COVID-19 pandemic, while somewhat contained, remains a critical challenge that has tested the resilience of humanity. Accurate and timely prediction of COVID-19 transmission dynamics and future trends is essential for informed decision-making in public health. Deep learning and mathematical models have emerged as promising tools, yet concerns regarding accuracy persist. This research suggests a novel model for forecasting the COVID-19's future trajectory. The model combines the benefits of machine learning models and mathematical models. The SIRVD model, a mathematical based model that depicts the reach of the infection via population, serves as basis for the proposed model. A deep prediction model for COVID-19 using XGBoost-SIRVD-LSTM is presented. The suggested approach combines Susceptible-Infected-Recovered-Vaccinated-Deceased (SIRVD), and a deep learning model, which includes Long Short-Term Memory (LSTM) and other prediction models, including feature selection using XGBoost method. The model keeps track of changes in each group's membership over time. To increase the SIRVD model's accuracy, machine learning is applied. The key properties for forecasting the spread of the infection are found using a method called feature selection. Then, in order to learn from these features and create predictions, a model involving deep learning is applied. The performance of the model proposed was assessed with prediction metrics such as R2, root mean square error (RMSE), mean absolute percentage error (MAPE), and normalized root mean square error (NRMSE). The results are also validated to those of other prediction models. The empirical results show that the suggested model outperforms similar models. Findings suggest its potential as a valuable tool for pandemic management and public health decision-making.
引用
收藏
页数:12
相关论文
共 31 条
  • [1] Measuring and Preventing COVID-19 Using the SIR Model and Machine Learning in Smart Health Care
    Alanazi, Saad Awadh
    Kamruzzaman, M. M.
    Alruwaili, Madallah
    Alshammari, Nasser
    Alqahtani, Salman Ali
    Karime, Ali
    [J]. JOURNAL OF HEALTHCARE ENGINEERING, 2020, 2020
  • [2] Time series predicting of COVID-19 based on deep learning
    Alassafi, Madini O.
    Jarrah, Mutasem
    Alotaibi, Reem
    [J]. NEUROCOMPUTING, 2022, 468 : 335 - 344
  • [3] Ali Rasha H., 2021, Journal of Physics: Conference Series, V1879, DOI 10.1088/1742-6596/1879/2/022083
  • [4] Forecasting of COVID-19 using deep layer Recurrent Neural Networks (RNNs) with Gated Recurrent Units (GRUs) and Long Short-Term Memory (LSTM) cells
    ArunKumar, K. E.
    Kalaga, Dinesh, V
    Kumar, Ch Mohan Sai
    Kawaji, Masahiro
    Brenza, Timothy M.
    [J]. CHAOS SOLITONS & FRACTALS, 2021, 146
  • [5] On exact integrability of a Covid-19 model: SIRV
    Babaei, Navid Amiri
    Ozer, Teoman
    [J]. MATHEMATICAL METHODS IN THE APPLIED SCIENCES, 2024, 47 (05) : 3529 - 3546
  • [6] Deep learning via LSTM models for COVID-19 infection forecasting in India
    Chandra, Rohitash
    Jain, Ayush
    Chauhan, Divyanshu Singh
    [J]. PLOS ONE, 2022, 17 (01):
  • [7] A SIR model assumption for the spread of COVID-19 in different communities
    Cooper, Ian
    Mondal, Argha
    Antonopoulos, Chris G.
    [J]. CHAOS SOLITONS & FRACTALS, 2020, 139
  • [8] An efficient XGBoost-DNN-based classification model for network intrusion detection system
    Devan, Preethi
    Khare, Neelu
    [J]. NEURAL COMPUTING & APPLICATIONS, 2020, 32 (16) : 12499 - 12514
  • [9] Forecasting of COVID-19 cases using deep learning models: Is it reliable and practically significant?
    Devaraj, Jayanthi
    Elavarasan, Rajvikram Madurai
    Pugazhendhi, Rishi
    Shafiullah, G. M.
    Ganesan, Sumathi
    Jeysree, Ajay Kaarthic
    Khan, Irfan Ahmad
    Hossain, Eklas
    [J]. RESULTS IN PHYSICS, 2021, 21
  • [10] An interactive web-based dashboard to track COVID-19 in real time
    Dong, Ensheng
    Du, Hongru
    Gardner, Lauren
    [J]. LANCET INFECTIOUS DISEASES, 2020, 20 (05) : 533 - 534