Predicting COVID-19 pandemic by spatio-temporal graph neural networks: A New Zealand's study

by   Viet Bach Nguyen, et al.

Modeling and simulations of pandemic dynamics play an essential role in understanding and addressing the spreading of highly infectious diseases such as COVID-19. In this work, we propose a novel deep learning architecture named Attention-based Multiresolution Graph Neural Networks (ATMGNN) that learns to combine the spatial graph information, i.e. geographical data, with the temporal information, i.e. timeseries data of number of COVID-19 cases, to predict the future dynamics of the pandemic. The key innovation is that our method can capture the multiscale structures of the spatial graph via a learning to cluster algorithm in a data-driven manner. This allows our architecture to learn to pick up either local or global signals of a pandemic, and model both the long-range spatial and temporal dependencies. Importantly, we collected and assembled a new dataset for New Zealand. We established a comprehensive benchmark of statistical methods, temporal architectures, graph neural networks along with our spatio-temporal model. We also incorporated socioeconomic cross-sectional data to further enhance our prediction. Our proposed model have shown highly robust predictions and outperformed all other baselines in various metrics for our new dataset of New Zealand along with existing datasets of England, France, Italy and Spain. For a future work, we plan to extend our work for real-time prediction and global scale. Our data and source code are publicly available at


page 1

page 12


Temporal Multiresolution Graph Neural Networks For Epidemic Prediction

In this paper, we introduce Temporal Multiresolution Graph Neural Networ...

ST-RAP: A Spatio-Temporal Framework for Real Estate Appraisal

In this paper, we introduce ST-RAP, a novel Spatio-Temporal framework fo...

Modeling the geospatial evolution of COVID-19 using spatio-temporal convolutional sequence-to-sequence neural networks

Europe was hit hard by the COVID-19 pandemic and Portugal was one of the...

STG4Traffic: A Survey and Benchmark of Spatial-Temporal Graph Neural Networks for Traffic Prediction

Traffic prediction has been an active research topic in the domain of sp...

A Spatial-Temporal Graph Based Hybrid Infectious Disease Model with Application to COVID-19

As the COVID-19 pandemic evolves, reliable prediction plays an important...

Understanding Human Innate Immune System Dependencies using Graph Neural Networks

Since the rapid outbreak of Covid-19 and with no approved vaccines to da...

S2SNet: A Pretrained Neural Network for Superconductivity Discovery

Superconductivity allows electrical current to flow without any energy l...

Please sign up or login with your details

Forgot password? Click here to reset