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

10/18/2020
by   Yunling Zheng, et al.
0

As the COVID-19 pandemic evolves, reliable prediction plays an important role for policy making. The classical infectious disease model SEIR (susceptible-exposed-infectious-recovered) is a compact yet simplistic temporal model. The data-driven machine learning models such as RNN (recurrent neural networks) can suffer in case of limited time series data such as COVID-19. In this paper, we combine SEIR and RNN on a graph structure to develop a hybrid spatio-temporal model to achieve both accuracy and efficiency in training and forecasting. We introduce two features on the graph structure: node feature (local temporal infection trend) and edge feature (geographic neighbor effect). For node feature, we derive a discrete recursion (called I-equation) from SEIR so that gradient descend method applies readily to its optimization. For edge feature, we design an RNN model to capture the neighboring effect and regularize the landscape of loss function so that local minima are effective and robust for prediction. The resulting hybrid model (called IeRNN) improves the prediction accuracy on state-level COVID-19 new case data from the US, out-performing standard temporal models (RNN, SEIR, and ARIMA) in 1-day and 7-day ahead forecasting. Our model accommodates various degrees of reopening and provides potential outcomes for policymakers.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
07/14/2020

A Recurrent Neural Network and Differential Equation Based Spatiotemporal Infectious Disease Model with Application to COVID-19

The outbreaks of Coronavirus Disease 2019 (COVID-19) have impacted the w...
research
03/22/2021

Spatio-Temporal Neural Network for Fitting and Forecasting COVID-19

We established a Spatio-Temporal Neural Network, namely STNN, to forecas...
research
10/31/2022

Spatial-Temporal Synchronous Graph Transformer network (STSGT) for COVID-19 forecasting

COVID-19 has become a matter of serious concern over the last few years....
research
05/12/2023

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

Modeling and simulations of pandemic dynamics play an essential role in ...
research
10/06/2022

Interpreting County Level COVID-19 Infection and Feature Sensitivity using Deep Learning Time Series Models

Interpretable machine learning plays a key role in healthcare because it...
research
03/20/2023

Graph Neural Rough Differential Equations for Traffic Forecasting

Traffic forecasting is one of the most popular spatio-temporal tasks in ...
research
02/17/2023

Fast Temporal Wavelet Graph Neural Networks

Spatio-temporal signals forecasting plays an important role in numerous ...

Please sign up or login with your details

Forgot password? Click here to reset