A Bayesian machine learning approach for spatio-temporal prediction of COVID-19 cases

Abstract

Modeling the spread of infectious diseases in space and time needs to take care of complex dependencies and uncertainties. Machine learning methods, and neural networks, in particular, are useful in modeling this sort of complex problems, although they generally lack of probabilistic interpretations. We propose a neural network method embedded in a Bayesian framework for modeling and predicting the number of cases of infectious diseases in areal units. A key feature is that our combined model considers the impact of human movement on the spread of the infectious disease, as an additional random factor to the also considered spatial neighborhood and temporal correlation components. Our model is evaluated over a COVID-19 dataset for 245 health zones of Castilla-Leon (Spain). The results show that a Bayesian model informed by a neural network method is generally able to predict the number of cases of COVID-19 in both space and time, with the human mobility factor having a strong influence on the model, together with the number of infections and deaths in nearby areas.