Examining COVID-19 Forecasting using Spatio-Temporal Graph Neural Networks
Not on the map. Why?
Abstract (arXiv)
In this work, we examine a novel forecasting approach for COVID-19 case prediction that uses Graph Neural Networks and mobility data. In contrast to existing time series forecasting models, the proposed approach learns from a single large-scale spatio-temporal graph, where nodes represent the region-level human mobility, spatial edges represent the human mobility based inter-region connectivity, and temporal edges represent node features through time. We evaluate this approach on the US county level COVID-19 dataset, and demonstrate that the rich spatial and temporal information leveraged by the graph neural network allows the model to learn complex dynamics. We show a 6% reduction of RMSLE and an absolute Pearson Correlation improvement from 0.9978 to 0.998 compared to the best performing baseline models. This novel source of information combined with graph based deep learning approaches can be a powerful tool to understand the spread and evolution of COVID-19. We encourage others to further develop a novel modeling paradigm for infectious disease based on GNNs and high resolution mobility data.
Key facts. Cited by 64 papers in the map (data through 14 Oct 2026). First posted on arXiv 6 Jul 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 13; 2022: 15; 2023: 17; 2024: 9; 2025: 6; 2026: 4.
64 papers.
Sort:
By area:
- 2023A Survey on Graph Neural Networks for Time Series: Forecasting, Classification, Imputation, and Anomaly Detectioncited by 145
- 2023Spatio-Temporal Graph Neural Networks for Predictive Learning in Urban Computing: A Surveycited by 140
- 2021Automated Machine Learning on Graphs: A Surveycited by 39
- 2022Learning the Dynamics of Physical Systems from Sparse Observations with Finite Element Networkscited by 34
- 2024A Survey of Deep Learning and Foundation Models for Time Series Forecastingcited by 33
- 2023Networked Time Series Imputation via Position-aware Graph Enhanced Variational Autoencoderscited by 32
- 2024A Review of Graph Neural Networks in Epidemic Modelingcited by 31
- 2022Modeling Multivariate Biosignals With Graph Neural Networks and Structured State Space Modelscited by 29
- 2021A GNN-RNN Approach for Harnessing Geospatial and Temporal Information: Application to Crop Yield Predictioncited by 28
- 2023Graph Deep Learning for Time Series Forecastingcited by 26
- 2023Environment-Aware Dynamic Graph Learning for Out-of-Distribution Generalizationcited by 26
- 2021Heterogeneous Temporal Graph Neural Networkcited by 23
38 more
- 2023Spatio-Temporal Graph Neural Networks: A Surveycited by 23
- 2022Graph Neural Networks Designed for Different Graph Types: A Surveycited by 21
- 2022TF-GNN: Graph Neural Networks in TensorFlowcited by 21
- 2023Neurosymbolic AI for Reasoning over Knowledge Graphs: A Surveycited by 21
- 2023Dynamic Graph Representation Learning with Neural Networks: A Surveycited by 15
- 2023Spatiotemporal-Augmented Graph Neural Networks for Human Mobility Simulationcited by 15
- 2022Temporal Multiresolution Graph Neural Networks For Epidemic Predictioncited by 11
- 2021Robust Graph Neural Networks via Probabilistic Lipschitz Constraintscited by 10
- 2021Estimating the State of Epidemics Spreading with Graph Neural Networkscited by 9
- 2022Automated Graph Machine Learning: Approaches, Libraries, Benchmarks and Directionscited by 9
- 2022Data-Centric Epidemic Forecasting: A Surveycited by 8
- 2021Combining Graph Neural Networks and Spatio-temporal Disease Models to Predict COVID-19 Cases in Germanycited by 7
- 2024Expressive Power of Temporal Message Passingcited by 7
- 2023Graph Neural Network for spatiotemporal data: methods and applicationscited by 6
- 2023MSGNN: Multi-scale Spatio-temporal Graph Neural Network for Epidemic Forecastingcited by 6
- 2022Sparsification and Filtering for Spatial-temporal GNN in Multivariate Time-seriescited by 5
- 2023MPSTAN: Metapopulation-based Spatio-Temporal Attention Network for Epidemic Forecastingcited by 5
- 2023Revealing the Power of Masked Autoencoders in Traffic Forecastingcited by 5
- 2021Early Detection of COVID-19 Hotspots Using Spatio-Temporal Datacited by 3
- 2021Multiwave COVID-19 Prediction from Social Awareness using Web Search and Mobility Datacited by 3
- 2022Analysis of the Spatio-temporal Dynamics of COVID-19 in Massachusetts via Spectral Graph Wavelet Theorycited by 3
- 2023Traffic4cast at NeurIPS 2022 -- Predict Dynamics along Graph Edges from Sparse Node Data: Whole City Traffic and ETA from Stationary Vehicle Detectorscited by 3
- 2024Edge Graph Intelligence: Reciprocally Empowering Edge Networks with Graph Intelligencecited by 3
- 2021Machine learning applications for COVID-19: A state-of-the-art reviewcited by 2
- 2021Deep diffusion-based forecasting of COVID-19 by incorporating network-level mobility informationcited by 2
- 2024Convection-Diffusion Equation: A Theoretically Certified Framework for Neural Networkscited by 2
- 2025The Logical Expressiveness of Temporal GNNs via Two-Dimensional Product Logicscited by 2
- 2021Real-time Prediction for Mechanical Ventilation in COVID-19 Patients using A Multi-task Gaussian Process Multi-objective Self-attention Networkcited by 1
- 2021Improving COVID-19 Forecasting using eXogenous Variablescited by 1
- 2022Fine-Grained Population Mobility Data-Based Community-Level COVID-19 Prediction Modelcited by 1
- 2022Spatio-Temporal-Frequency Graph Attention Convolutional Network for Aircraft Recognition Based on Heterogeneous Radar Networkcited by 1
- 2024Graph Neural Network Surrogates to leverage Mechanistic Expert Knowledge towards Reliable and Immediate Pandemic Responsecited by 1
- 2025Urban Incident Prediction with Graph Neural Networks: Integrating Government Ratings and Crowdsourced Reportscited by 1
- 2026Prior Knowledge-enhanced Spatio-temporal Epidemic Forecastingcited by 1
- 2021STOPPAGE: Spatio-temporal Data Driven Cloud-Fog-Edge Computing Framework for Pandemic Monitoring and Managementcited by 0
- 2022Temporal-Spatial dependencies ENhanced deep learning model (TSEN) for household leverage series forecastingcited by 0
- 2022Deep Learning-Derived Optimal Aviation Strategies to Control Pandemicscited by 0
- 2022Spatial-Temporal Synchronous Graph Transformer network (STSGT) for COVID-19 forecastingcited by 0
50 most cited of 64 All 64 (JSON)
- 2026Graph World Models for Constrained Epidemic Policy Planning28 Sep 2026
- 2026Spatio-temporal stochastic graph-based learning for infectious disease forecasting28 May 2026
- 2026Physics-Informed Neural Network with Adaptive Clustering Learning Mechanism for Information Popularity Prediction20 Mar 2026
- 2026Prior Knowledge-enhanced Spatio-temporal Epidemic Forecasting25 Feb 2026
- 2025Epi2-Net: Advancing Epidemic Dynamics Forecasting with Physics-Inspired Neural Networks4 Aug 2025
- 2025Forecasting Coccidioidomycosis (Valley Fever) in Arizona: A Graph Neural Network Approach14 Jul 2025
- 2025Urban Incident Prediction with Graph Neural Networks: Integrating Government Ratings and Crowdsourced Reports10 Jun 2025
- 2025Integrating Spatiotemporal Features in LSTM for Spatially Informed COVID-19 Hospitalization Forecasting6 Jun 2025
- 2025The Logical Expressiveness of Temporal GNNs via Two-Dimensional Product Logics17 May 2025
- 2025Leveraging graph neural networks and mobility data for COVID-19 forecasting20 Jan 2025
- 2024MP-PINN: A Multi-Phase Physics-Informed Neural Network for Epidemic Forecasting11 Nov 2024
- 2024Graph Neural Network Surrogates to leverage Mechanistic Expert Knowledge towards Reliable and Immediate Pandemic Response10 Nov 2024
38 more
- 2024Expressive Power of Temporal Message Passing19 Aug 2024
- 2024Edge Graph Intelligence: Reciprocally Empowering Edge Networks with Graph Intelligence7 Jul 2024
- 2024Generative Probabilistic Planning for Optimizing Supply Chain Networks11 Apr 2024
- 2024A Review of Graph Neural Networks in Epidemic Modeling28 Mar 2024
- 2024Convection-Diffusion Equation: A Theoretically Certified Framework for Neural Networks23 Mar 2024
- 2024A Survey of Deep Learning and Foundation Models for Time Series Forecasting25 Jan 2024
- 2024Global Prediction of COVID-19 Variant Emergence Using Dynamics-Informed Graph Neural Networks7 Jan 2024
- 2023Backbone-based Dynamic Graph Spatio-Temporal Network for Epidemic Forecasting1 Dec 2023
- 2023Environment-Aware Dynamic Graph Learning for Out-of-Distribution Generalization18 Nov 2023
- 2023Graph Deep Learning for Time Series Forecasting24 Oct 2023
- 2023PASTA: PArallel Spatio-Temporal Attention with spatial auto-correlation gating for fine-grained crowd flow prediction2 Oct 2023
- 2023Revealing the Power of Masked Autoencoders in Traffic Forecasting26 Sep 2023
- 2023MSGNN: Multi-scale Spatio-temporal Graph Neural Network for Epidemic Forecasting30 Aug 2023
- 2023Machine Learning for Infectious Disease Risk Prediction: A Survey6 Aug 2023
- 2023A Survey on Graph Neural Networks for Time Series: Forecasting, Classification, Imputation, and Anomaly Detection7 Jul 2023
- 2023MPSTAN: Metapopulation-based Spatio-Temporal Attention Network for Epidemic Forecasting15 Jun 2023
- 2023Spatiotemporal-Augmented Graph Neural Networks for Human Mobility Simulation15 Jun 2023
- 2023Graph Neural Network for spatiotemporal data: methods and applications30 May 2023
- 2023Networked Time Series Imputation via Position-aware Graph Enhanced Variational Autoencoders29 May 2023
- 2023Dynamic Graph Representation Learning with Neural Networks: A Survey12 Apr 2023
- 2023Spatio-Temporal Graph Neural Networks for Predictive Learning in Urban Computing: A Survey25 Mar 2023
- 2023Traffic4cast at NeurIPS 2022 -- Predict Dynamics along Graph Edges from Sparse Node Data: Whole City Traffic and ETA from Stationary Vehicle Detectors14 Mar 2023
- 2023Neurosymbolic AI for Reasoning over Knowledge Graphs: A Survey14 Feb 2023
- 2023Spatio-Temporal Graph Neural Networks: A Survey25 Jan 2023
- 2022A Mobility-Aware Deep Learning Model for Long-Term COVID-19 Pandemic Prediction and Policy Impact Analysis5 Dec 2022
- 2022Modeling Multivariate Biosignals With Graph Neural Networks and Structured State Space Models21 Nov 2022
- 2022Spatial-Temporal Synchronous Graph Transformer network (STSGT) for COVID-19 forecasting31 Oct 2022
- 2022Temporal-Spatial dependencies ENhanced deep learning model (TSEN) for household leverage series forecasting17 Oct 2022
- 2022Deep Learning-Derived Optimal Aviation Strategies to Control Pandemics12 Oct 2022
- 2022Analysis of the Spatio-temporal Dynamics of COVID-19 in Massachusetts via Spectral Graph Wavelet Theory28 Jul 2022
- 2022Data-Centric Epidemic Forecasting: A Survey19 Jul 2022
- 2022TF-GNN: Graph Neural Networks in TensorFlow7 Jul 2022
- 2022Temporal Multiresolution Graph Neural Networks For Epidemic Prediction30 May 2022
- 2022Spatio-Temporal-Frequency Graph Attention Convolutional Network for Aircraft Recognition Based on Heterogeneous Radar Network15 Apr 2022
- 2022Graph Neural Networks Designed for Different Graph Types: A Survey6 Apr 2022
- 2022Learning the Dynamics of Physical Systems from Sparse Observations with Finite Element Networks16 Mar 2022
- 2022Sparsification and Filtering for Spatial-temporal GNN in Multivariate Time-series8 Mar 2022
- 2022Fine-Grained Population Mobility Data-Based Community-Level COVID-19 Prediction Model13 Feb 2022
50 newest of 64 All 64 (JSON)
- 2026Graph World Models for Constrained Epidemic Policy Planning28 Sep 2026
- 2023A Survey on Graph Neural Networks for Time Series: Forecasting, Classification, Imputation, and Anomaly Detectioncited by 145
- 2023Spatio-Temporal Graph Neural Networks for Predictive Learning in Urban Computing: A Surveycited by 140
- 2021Automated Machine Learning on Graphs: A Surveycited by 39
- 2022Learning the Dynamics of Physical Systems from Sparse Observations with Finite Element Networkscited by 34
- 2024A Survey of Deep Learning and Foundation Models for Time Series Forecastingcited by 33
- 2023Networked Time Series Imputation via Position-aware Graph Enhanced Variational Autoencoderscited by 32
- 2024A Review of Graph Neural Networks in Epidemic Modelingcited by 31
- 2021A GNN-RNN Approach for Harnessing Geospatial and Temporal Information: Application to Crop Yield Predictioncited by 28
- 2023Graph Deep Learning for Time Series Forecastingcited by 26
- 2023Environment-Aware Dynamic Graph Learning for Out-of-Distribution Generalizationcited by 26
- 2021Heterogeneous Temporal Graph Neural Networkcited by 23
- 2023Spatio-Temporal Graph Neural Networks: A Surveycited by 23
12 most cited of 61 All 61 (JSON)
- 2026Spatio-temporal stochastic graph-based learning for infectious disease forecasting28 May 2026
- 2026Physics-Informed Neural Network with Adaptive Clustering Learning Mechanism for Information Popularity Prediction20 Mar 2026
- 2026Prior Knowledge-enhanced Spatio-temporal Epidemic Forecasting25 Feb 2026
- 2025Epi2-Net: Advancing Epidemic Dynamics Forecasting with Physics-Inspired Neural Networks4 Aug 2025
- 2025Forecasting Coccidioidomycosis (Valley Fever) in Arizona: A Graph Neural Network Approach14 Jul 2025
- 2025Urban Incident Prediction with Graph Neural Networks: Integrating Government Ratings and Crowdsourced Reports10 Jun 2025
- 2025Integrating Spatiotemporal Features in LSTM for Spatially Informed COVID-19 Hospitalization Forecasting6 Jun 2025
- 2025The Logical Expressiveness of Temporal GNNs via Two-Dimensional Product Logics17 May 2025
- 2025Leveraging graph neural networks and mobility data for COVID-19 forecasting20 Jan 2025
- 2024MP-PINN: A Multi-Phase Physics-Informed Neural Network for Epidemic Forecasting11 Nov 2024
- 2024Graph Neural Network Surrogates to leverage Mechanistic Expert Knowledge towards Reliable and Immediate Pandemic Response10 Nov 2024
- 2024Expressive Power of Temporal Message Passing19 Aug 2024
12 newest of 61 All 61 (JSON)