A self-driven ESN-DSS approach for effective COVID-19 time series prediction and modelling

Epidemiol Infect. 2024 Nov 22:152:e146. doi: 10.1017/S0950268824000992.

Abstract

Since the outbreak of the COVID-19 epidemic, it has posed a great crisis to the health and economy of the world. The objective is to provide a simple deep-learning approach for predicting, modelling, and evaluating the time evolutions of the COVID-19 epidemic. The Dove Swarm Search (DSS) algorithm is integrated with the echo state network (ESN) to optimize the weight. The ESN-DSS model is constructed to predict the evolution of the COVID-19 time series. Specifically, the self-driven ESN-DSS is created to form a closed feedback loop by replacing the input with the output. The prediction results, which involve COVID-19 temporal evolutions of multiple countries worldwide, indicate the excellent prediction performances of our model compared with several artificial intelligence prediction methods from the literature (e.g., recurrent neural network, long short-term memory, gated recurrent units, variational auto encoder) at the same time scale. Moreover, the model parameters of the self-driven ESN-DSS are determined which acts as a significant impact on the prediction performance. As a result, the network parameters are adjusted to improve the prediction accuracy. The prediction results can be used as proposals to help governments and medical institutions formulate pertinent precautionary measures to prevent further spread. In addition, this study is not only limited to COVID-19 time series forecasting but also applicable to other nonlinear time series prediction problems.

Keywords: COVID-19; Dove Swarm Search; deep learning; echo state networks; self-driven; time series forecasting.

MeSH terms

  • Algorithms
  • COVID-19* / epidemiology
  • COVID-19* / prevention & control
  • Deep Learning
  • Humans
  • Neural Networks, Computer
  • SARS-CoV-2