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

被引:0
|
作者
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
相关论文
共 50 条
  • [1] SIRVD-DL: A COVID-19 deep learning prediction model based on time-dependent SIRVD
    Liao, Zhifang
    Lan, Peng
    Fan, Xiaoping
    Kelly, Benjamin
    Innes, Aidan
    Liao, Zhining
    [J]. COMPUTERS IN BIOLOGY AND MEDICINE, 2021, 138
  • [2] Spreading Infection Simulation in Japan Based on SIRVD Model Considering COVID-19 Vaccine
    Omae, Yuto
    Kakimoto, Yohei
    Toyotani, Jun
    Hara, Kazuyuki
    Gon, Yasuhiro
    Takahashi, Hirotaka
    [J]. Journal of Japan Industrial Management Association, 2022, 73 (01): : 27 - 30
  • [3] Improved LSTM-based deep learning model for COVID-19 prediction using optimized approach
    Zhou, Luyu
    Zhao, Chun
    Liu, Ning
    Yao, Xingduo
    Cheng, Zewei
    [J]. ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 122
  • [4] A novel bidirectional LSTM deep learning approach for COVID-19 forecasting
    Aung, Nway Nway
    Pang, Junxiong
    Chua, Matthew Chin Heng
    Tan, Hui Xing
    [J]. SCIENTIFIC REPORTS, 2023, 13 (01)
  • [5] A novel bidirectional LSTM deep learning approach for COVID-19 forecasting
    Nway Nway Aung
    Junxiong Pang
    Matthew Chin Heng Chua
    Hui Xing Tan
    [J]. Scientific Reports, 13
  • [6] A Study on the Prediction of COVID-19 Confirmed Cases Using Deep Learning and AdaBoost-Bi-LSTM Model
    Shin, Dong-Ryeol
    Chae, Gayoung
    Park, Minjae
    [J]. INTERNATIONAL JOURNAL OF RELIABILITY QUALITY AND SAFETY ENGINEERING, 2024, 31 (01)
  • [7] Time series prediction of COVID-19 transmission in America using LSTM and XGBoost algorithms
    Luo, Junling
    Zhang, Zhongliang
    Fu, Yao
    Rao, Feng
    [J]. RESULTS IN PHYSICS, 2021, 27
  • [8] LSTM algorithm optimization for COVID-19 prediction model
    Sembiring, Irwan
    Wahyuni, Sri Ngudi
    Sediyono, Eko
    [J]. HELIYON, 2024, 10 (04)
  • [9] Deep learning-based approach for COVID-19 spread prediction
    Cumbane, Silvino Pedro
    Gidofalvi, Gyozo
    [J]. INTERNATIONAL JOURNAL OF DATA SCIENCE AND ANALYTICS, 2024,
  • [10] Deep learning-based approach for COVID-19 spread prediction
    Division of Geoinformatics, Department of Urban Planning and Environment, KTH Royal Institute of Technology, Teknikringen 10A, Stockholm
    114 28, Sweden
    不详
    3453, Mozambique
    [J]. Int. J. Data Sci. Anal.,