Combining Graph Neural Networks and Spatio-temporal Disease Models to Predict COVID-19 Cases in Germany

Abstract

During 2020, the infection rate of COVID-19 has been investigated by many scholars from different research fields. In this context, reliable and interpretable forecasts of disease incidents are a vital tool for policymakers to manage healthcare resources. Several experts have called for the necessity to account for human mobility to explain the spread of COVID-19. Existing approaches are often applying standard models of the respective research field. This habit, however, often comes along with certain restrictions. For instance, most statistical or epidemiological models cannot directly incorporate unstructured data sources, including relational data that may encode human mobility. In contrast, machine learning approaches may yield better predictions by exploiting these data structures, yet lack intuitive interpretability as they are often categorized as black-box models. We propose a trade-off between both research directions and present a multimodal learning approach that combines the advantages of statistical regression and machine learning models for predicting local COVID-19 cases in Germany. This novel approach enables the use of a richer collection of data types, including mobility flows and colocation probabilities, and yields the lowest MSE scores throughout our observational period in our benchmark study. The results corroborate the necessity of including mobility data and showcase the flexibility and interpretability of our approach.

References in corpus (15)

• Practical Bayesian Optimization of Machine Learning Algorithms
• Inductive Representation Learning on Large Graphs
• Neural Message Passing for Quantum Chemistry
• Geometric deep learning: going beyond Euclidean data
• Spectral Networks and Locally Connected Networks on Graphs
• Evidence of economic segregation from mobility lockdown during COVID-19 epidemic
• graph2vec: Learning Distributed Representations of Graphs
• Examining COVID-19 Forecasting using Spatio-Temporal Graph Neural Networks
• Bayesian structured additive distributional regression with an application to regional income inequality in Germany
• Neural Additive Models: Interpretable Machine Learning with Neural Nets
• On the Interplay of Regional Mobility, Social Connectedness, and the Spread of COVID-19 in Germany
• Nowcasting fatal COVID-19 infections on a regional level in Germany
• STAN: Spatio-Temporal Attention Network for Pandemic Prediction Using Real World Evidence
• Neural Mixture Distributional Regression
• Regional now- and forecasting for data reported with delay: Towards surveillance of COVID-19 infections

Cited by in corpus (3)

• Deep learning of contagion dynamics on complex networks
• Integrating LSTMs and GNNs for COVID-19 Forecasting
• Regional now- and forecasting for data reported with delay: Towards surveillance of COVID-19 infections