Variational-LSTM autoencoder to forecast the spread of coronavirus across the globe

PLoS One. 2021 Jan 28;16(1):e0246120. doi: 10.1371/journal.pone.0246120. eCollection 2021.

Abstract

Modelling the spread of coronavirus globally while learning trends at global and country levels remains crucial for tackling the pandemic. We introduce a novel variational-LSTM Autoencoder model to predict the spread of coronavirus for each country across the globe. This deep Spatio-temporal model does not only rely on historical data of the virus spread but also includes factors related to urban characteristics represented in locational and demographic data (such as population density, urban population, and fertility rate), an index that represents the governmental measures and response amid toward mitigating the outbreak (includes 13 measures such as: 1) school closing, 2) workplace closing, 3) cancelling public events, 4) close public transport, 5) public information campaigns, 6) restrictions on internal movements, 7) international travel controls, 8) fiscal measures, 9) monetary measures, 10) emergency investment in health care, 11) investment in vaccines, 12) virus testing framework, and 13) contact tracing). In addition, the introduced method learns to generate a graph to adjust the spatial dependences among different countries while forecasting the spread. We trained two models for short and long-term forecasts. The first one is trained to output one step in future with three previous timestamps of all features across the globe, whereas the second model is trained to output 10 steps in future. Overall, the trained models show high validation for forecasting the spread for each country for short and long-term forecasts, which makes the introduce method a useful tool to assist decision and policymaking for the different corners of the globe.

MeSH terms

  • COVID-19 / epidemiology
  • Coronavirus / isolation & purification
  • Coronavirus Infections / epidemiology*
  • Disease Outbreaks
  • Epidemiologic Methods
  • Epidemiological Monitoring
  • Forecasting
  • Global Health
  • Models, Statistical
  • Pandemics
  • SARS-CoV-2 / isolation & purification

Grants and funding

The authors received no specific funding for this work.