Multi-region infectious disease prediction modeling based on spatio-temporal graph neural network and the dynamic model

by Xiaoyi Wang, Zhen Jin

Human mobility between different regions is a major factor in large-scale outbreaks of infectious diseases. Deep learning models incorporating infectious disease transmission dynamics for predicting the spread of multi-regional outbreaks due to human mobility have become a hot research topic. In this study, we incorporate the Graph Transformer Neural Network and graph learning mechanisms into a metapopulation SIR model to build a hybrid framework, Metapopulation Graph Transformer Neural Network (M-Graphormer), for high-dimensional parameter estimation and multi-regional epidemic prediction. The framework effectively solves the problem that existing models may lose some hidden spatial dependencies in the data when dealing with the dynamic graph structure of the network due to human mobility. We performed multi-wave infectious disease prediction in multiple regions based on real epidemic data. The results show that the framework is capable of performing high-dimensional parameter estimation and accurately predicting epidemic transmission dynamics in multiple regions even with low data quality. In addition, we retrospectively extrapolate the temporal evolution patterns of contact rate under different interventions implemented in different regions, reflecting the dynamics of intervention intensity and the need for flexibility in adjusting interventions in different regions. To provide early warning of infectious disease transmission, we retrospectively predicted the arrival time of infectious diseases using data from the early stages of outbreaks.