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

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.

Automated summary

This study proposes a neural network model embedded in a Bayesian framework for modeling and predicting the number of infectious disease cases in areal units. The model takes into account the impact of human movement, spatial neighborhood, and temporal correlation on disease spread. The results show that the model is able to predict the number of COVID-19 cases in space and time, with human mobility playing a significant role.