Abstract:Accurate epidemic forecasting plays a vital role for governments in developing effective prevention measures for suppressing epidemics. Most of the present spatio-temporal models cannot provide a general framework for stable, and accurate forecasting of epidemics with diverse evolution trends. Incorporating epidemiological domain knowledge ranging from single-patch to multi-patch into neural networks is expected to improve forecasting accuracy. However, relying solely on single-patch knowledge neglects inter-patch interactions, while constructing multi-patch knowledge is challenging without population mobility data. To address the aforementioned problems, we propose a novel hybrid model called Metapopulation-based Spatio-Temporal Attention Network (MPSTAN). This model aims to improve the accuracy of epidemic forecasting by incorporating multi-patch epidemiological knowledge into a spatio-temporal model and adaptively defining inter-patch interactions. Moreover, we incorporate inter-patch epidemiological knowledge into both the model construction and loss function to help the model learn epidemic transmission dynamics. Extensive experiments conducted on two representative datasets with different epidemiological evolution trends demonstrate that our proposed model outperforms the baselines and provides more accurate and stable short- and long-term forecasting. We confirm the effectiveness of domain knowledge in the learning model and investigate the impact of different ways of integrating domain knowledge on forecasting. We observe that using domain knowledge in both model construction and loss functions leads to more efficient forecasting, and selecting appropriate domain knowledge can improve accuracy further.

EpiRiskNet: incorporating graph structure and static data as prior knowledge for improved time-series forecasting

A Knowledge-based and Data-driven Approach for Predicting Acute Kidney Injury in Patients with Heart Failure.

Foresee Urban Sparse Traffic Accidents: A Spatiotemporal Multi-Granularity Perspective

EpiGNN: Exploring Spatial Transmission with Graph Neural Network for Regional Epidemic Forecasting

A Study on Graph-Structured Recurrent Neural Networks and Sparsification with Application to Epidemic Forecasting.

A Novel Multi-scale Spatiotemporal Graph Neural Network for Epidemic Prediction.

Dynamic road crime risk prediction with urban open data

Inter- and Intra-Series Embeddings Fusion Network for Epidemiological Forecasting

Backbone-based Dynamic Graph Spatio-Temporal Network for Epidemic Forecasting

Modeling epidemic dynamics using Graph Attention based Spatial Temporal networks

Time-aware Context-Gated Graph Attention Network for Clinical Risk Prediction

Predicting Citywide Crowd Flows in Critical Areas Based on Dynamic Spatio-Temporal Network

StageNet: Stage-Aware Neural Networks for Health Risk Prediction

Urban Traffic Accident Risk Prediction Revisited: Regionality, Proximity, Similarity and Sparsity

Temporal-Aware Graph Neural Network for Credit Risk Prediction.

Temporal-Based Risk Forecasting Approach for Key Areas on Surveillance Sensor Networks of High-Speed Railway Transport Hub.

Graph Neural Ordinary Differential Equations for Epidemic Forecasting

A Multi-Hazard Risk Assessment Model for a Road Network Based on Neural Networks and Fuzzy Comprehensive Evaluation

MPSTAN: Metapopulation-based Spatio-Temporal Attention Network for Epidemic Forecasting

Improving influenza surveillance based on multi-granularity deep spatiotemporal neural network

Spatial-temporal correlated graph neural networks based on neighborhood feature selection for traffic data prediction