Abstract
The COVID-19 pandemic has posed great challenges to public health services, government agencies, and policymakers, raising huge social conflicts between public health and economic resilience. Policies such as reopening or closure of business activities are formulated based on scientific projections of infection risks obtained from infection dynamics models. Though most parameters in epidemic prediction service models can be set with domain knowledge of COVID-19, a key parameter, namely, human mobility, is often challenging to estimate due to complex spatio-temporal correlations and social contexts under escalating COVID-19 facilities. Moreover, how to integrate the various implicit features to accurately predict infectious cases is still an open issue. To address this challenge, we formulate the problem as a spatio-temporal network representation problem and propose STEP, a Spatio-Temporal Epidemic Prediction framework, to estimate pandemic infection risk of a city by integrating various real-world conditions (e.g., City Risk Index, climate, and medical conditions) into graph-structured data. We also employ a multi-head attention mechanism in representation learning to extract implicit features for a given city. Extensive experiments have been conducted upon the real-world dataset for 51 states (50 states and Washington, D.C.) of the USA. Experimental results show that STEP can yield more accurate pandemic infection risk estimation than baseline methods. Moreover, STEP outperforms other methods in both short-term and long-term prediction.
- [1] . 2020. The scales of human mobility. Nature 587, 7834 (2020), 402–407.Google ScholarCross Ref
- [2] . 2022. A generic deep learning based cough analysis system from clinically validated samples for point-of-need Covid-19 test and severity levels. IEEE Transactions on Service Computing 15, 3 (2022), 1220–1232.Google Scholar
- [3] . 2020. Big networks: A survey. Comput. Sci. Rev. 37 (2020), 1–1.Google ScholarCross Ref
- [4] . 2013. The hidden geometry of complex, network-driven contagion phenomena. Science 342, 6164 (2013), 1337–1342.Google ScholarCross Ref
- [5] . 2020. Evolution and epidemic spread of SARS-CoV-2 in Brazil. Science 369, 6508 (2020), 1255–1260.Google ScholarCross Ref
- [6] . 2016. Path-based epidemic spreading in networks. IEEE/ACM Trans. Netw. 25, 1 (2016), 565–578.Google ScholarDigital Library
- [7] . 2019. Cluster-GCN: An efficient algorithm for training deep and large graph convolutional networks. In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 257–266.Google ScholarDigital Library
- [8] . 2014. Empirical evaluation of gated recurrent neural networks on sequence modeling. arXiv preprint arXiv:1412.3555 (2014).Google Scholar
- [9] . 2021. Predicting the dynamics of the COVID-19 pandemic in the United States using graph theory-based neural networks. Int. J. Environ. Res. Pub. Health 18, 7 (2021), 3834.Google ScholarCross Ref
- [10] . 2020. Inferring change points in the spread of COVID-19 reveals the effectiveness of interventions. Science 369, 6500 (2020).Google ScholarCross Ref
- [11] . 2021. Time series change point detection with self-supervised contrastive predictive coding. In Proceedings of the Web Conference. 3124–3135.Google ScholarDigital Library
- [12] . 2019. Learning dynamic context graphs for predicting social events. In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 1007–1016.Google ScholarDigital Library
- [13] . 2020. Cola-GNN: Cross-location attention based graph neural networks for long-term ILI prediction. In Proceedings of the 29th ACM International Conference on Information & Knowledge Management. 245–254.Google ScholarDigital Library
- [14] . 2021. Combining graph neural networks and spatio-temporal disease models to predict Covid-19 cases in Germany. arXiv preprint arXiv:2101.00661 (2021).Google Scholar
- [15] . 2020. Be aware of the hot zone: A warning system of hazard area prediction to intervene novel coronavirus COVID-19 outbreak. In Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval. 2241–2250.Google ScholarDigital Library
- [16] . 2016. LSTM: A search space odyssey. IEEE Trans. Neural Netw. Learn. Syst. 28, 10 (2016), 2222–2232.Google ScholarCross Ref
- [17] . 2021. Research on the forecast of the spread of COVID-19. In Proceedings of the 11th International Conference on Biomedical Engineering and Technology. 47–51.Google ScholarDigital Library
- [18] . 2014. Exact analytical solutions of the susceptible-infected-recovered (SIR) epidemic model and of the SIR model with equal death and birth rates. Appl. Math. Comput. 236 (2014), 184–194.Google ScholarCross Ref
- [19] . 2019. Curing epidemics on networks using a Polya contagion model. IEEE/ACM Trans. Netw. 27, 5 (2019), 2085–2097.Google ScholarDigital Library
- [20] . 2019. Spatio-temporal capsule-based reinforcement learning for mobility-on-demand network coordination. In Proceedings of the World Wide Web Conference. 2806–2813.Google ScholarDigital Library
- [21] . 2020. The effect of large-scale anti-contagion policies on the COVID-19 pandemic. Nature 584, 7820 (2020), 262–267.Google ScholarCross Ref
- [22] . 2021. Temporal hierarchical graph attention network for traffic prediction. ACM Trans. Intell. Syst. Technol. 12, 6 (2021).Google ScholarDigital Library
- [23] . 2016. Location prediction: A temporal-spatial Bayesian model. ACM Trans. Intell. Syst. Technol. 7, 3 (2016).Google ScholarDigital Library
- [24] . 2020. Examining COVID-19 forecasting using spatio-temporal graph neural networks. arXiv preprint arXiv:2007.03113 (2020).Google Scholar
- [25] . 2017. Semi-supervised classification with graph convolutional networks. In Proceedings of the International Conference on Learning Representations (ICLR).Google Scholar
- [26] . 2018. Big trajectory data: A survey of applications and services. IEEE Access 6 (2018), 58295–58306.Google ScholarCross Ref
- [27] . 2015. Advances in nowcasting influenza-like illness rates using search query logs. Sci. Rep. 5, 1 (2015), 1–10.Google ScholarCross Ref
- [28] . 2020. AutoST: Efficient neural architecture search for spatio-temporal prediction. In Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 794–802.Google ScholarDigital Library
- [29] . 2018. Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. In Proceedings of the 6th International Conference on Learning Representations.Google Scholar
- [30] . 2021. Tracing the pace of COVID-19 research: Topic modeling and evolution. Big Data Res. 25 (2021), 100236.Google ScholarDigital Library
- [31] . 2020. CRI: Measuring city infection risk amid COVID-19. In Proceedings of the IEEE Asia-Pacific Conference on Computer Science and Data Engineering (CSDE). IEEE, 1–6.Google ScholarCross Ref
- [32] . 2019. Break the Ceiling: Stronger Multi-scale Deep Graph Convolutional Networks. Curran Associates Inc.Google Scholar
- [33] . 2018. LC-RNN: A deep learning model for traffic speed prediction. In Proceedings of the 27th International Joint Conference on Artificial Intelligence. 3470–3476.Google ScholarCross Ref
- [34] . 2021. A human mobility data driven hybrid GNN+ RNN based model for epidemic prediction. In Proceedings of the IEEE International Conference on Big Data (Big Data). IEEE, 857–866.Google ScholarCross Ref
- [35] . 2017. Opportunities and challenges in modeling emerging infectious diseases. Science 357, 6347 (2017), 149–152.Google ScholarCross Ref
- [36] . 2019. Spatio-temporal event forecasting and precursor identification. In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 3237–3238.Google ScholarDigital Library
- [37] . 1974. Geographic and temporal development of plagues. Nature 250, 5469 (1974), 726–729.Google ScholarCross Ref
- [38] . 2020. STP-TrellisNets: Spatial-temporal parallel TrellisNets for metro station passenger flow prediction. In Proceedings of the 29th ACM International Conference on Information & Knowledge Management. 1185–1194.Google ScholarDigital Library
- [39] . 2021. Transfer graph neural networks for pandemic forecasting. In Proceedings of the 35th AAAI Conference on Artificial Intelligence, 33rd Conference on Innovative Applications of Artificial Intelligence, 11th Symposium on Educational Advances in Artificial Intelligence. AAAI Press, 4838–4845.Google ScholarCross Ref
- [40] . 2017. Using participatory web-based surveillance data to improve seasonal influenza forecasting in Italy. In Proceedings of the 26th International Conference on World Wide Web. 303–310.Google ScholarDigital Library
- [41] . 2018. DeepInf: Social influence prediction with deep learning. In Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 2110–2119.Google ScholarDigital Library
- [42] . 2018. SIR-Hawkes: Linking epidemic models and Hawkes processes to model diffusions in finite populations. In Proceedings of the World Wide Web Conference. 419–428.Google ScholarDigital Library
- [43] . 2021. The universal visitation law of human mobility. Nature 593, 7860 (2021), 522–527.Google ScholarCross Ref
- [44] . 2020. Simulating the impact of hospital capacity and social isolation to minimize the propagation of infectious diseases. In Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 3451–3457.Google ScholarDigital Library
- [45] . 2020. Network representation learning: From traditional feature learning to deep learning. IEEE Access 8 (2020), 205600–205617.Google ScholarCross Ref
- [46] . 2022. Estimating the state of epidemics spreading with graph neural networks. Nonlin. Dynam. 109, 1 (2022), 249–263.Google ScholarCross Ref
- [47] . 2019. Urban human mobility: Data-driven modeling and prediction. ACM SIGKDD Explor. Newslett. Arch. 21, 1 (2019), 1–19.Google ScholarDigital Library
- [48] . 2018. Inferring metapopulation propagation network for intra-city epidemic control and prevention. In Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 830–838.Google ScholarDigital Library
- [49] . 2016. Semi-supervised classification with graph convolutional networks. In Proceedings of the International Conference on Learning Representations (ICLR’17).Google Scholar
- [50] . 2020. Evidence that coronavirus superspreading is fat-tailed. Proc. Nat. Acad. Sci. 117, 47 (2020), 29416–29418.Google ScholarCross Ref
- [51] . 2018. Modeling and analysis of large-scale urban mobility for green transportation. IEEE Trans. Industr. Inform. 14, 4 (2018), 1469–1481.Google ScholarCross Ref
- [52] . 2021. Graph learning: A survey. IEEE Trans. Artif. Intell. 2, 2 (2021), 109–127.Google ScholarCross Ref
- [53] . 2020. Deep graph convolutional networks for incident-driven traffic speed prediction. In Proceedings of the 29th ACM International Conference on Information & Knowledge Management. 1665–1674.Google ScholarDigital Library
- [54] . 2020. Multivariate relations aggregation learning in social networks. In Proceedings of the ACM/IEEE Joint Conference on Digital Libraries. 77–86.Google ScholarDigital Library
- [55] . 2014. Accurate and interpretable Bayesian MARS for traffic flow prediction. IEEE Trans. Intell. Transport. Syst. 15, 6 (2014), 2457–2469.Google ScholarCross Ref
- [56] . 2019. Learning from multiple cities: A meta-learning approach for spatial-temporal prediction. In Proceedings of the World Wide Web Conference. 2181–2191.Google ScholarDigital Library
- [57] . 2019. Co-prediction of multiple transportation demands based on deep spatio-temporal neural network. In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 305–313.Google ScholarDigital Library
- [58] . 2018. Deep distributed fusion network for air quality prediction. In Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 965–973.Google ScholarDigital Library
- [59] . 2019. Science of scientific team science: A survey. Comput. Sci. Rev. 31 (2019), 72–83.Google ScholarDigital Library
- [60] . 2021. Data-driven decision-making in COVID-19 response: A survey. IEEE Transactions on Computational Social Systems 8, 4 (2021), 1016–1029.Google Scholar
- [61] . 2020. Detecting outlier patterns with query-based artificially generated searching conditions. IEEE Trans. Computat. Soc. Syst. 8, 1 (2020), 134–147.Google ScholarCross Ref
- [62] . 2020. Data-driven computational social science: A survey. Big Data Res. 21 (2020), 2214–5796.Google ScholarCross Ref
- [63] . 2020. Curb-GAN: Conditional urban traffic estimation through spatio-temporal generative adversarial networks. In Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 842–852.Google ScholarDigital Library
- [64] . 2019. T-GCN: A temporal graph convolutional network for traffic prediction. IEEE Trans. Intell. Transport. Syst. 21, 9 (2019), 3848–3858.Google ScholarCross Ref
- [65] . 2020. A pneumonia outbreak associated with a new coronavirus of probable bat origin. Nature 579, 7798 (2020), 270–273.Google ScholarCross Ref
Index Terms
-
Spatio-temporal Graph Learning for Epidemic Prediction
-
Recommendations
-
STANN: Spatio-Temporal Attention-based Neural Network for Epidemic Prediction
WI-IAT '21: IEEE/WIC/ACM International Conference on Web Intelligence and Intelligent Agent TechnologyAccurately forecasting the future trend of an epidemic plays an essential role in making effective and efficient public health policies for disease prevention and control. In reality, the local transmission of an epidemic depends not only on the ...
-
Intervention-Aware Epidemic Prediction by Enhanced Whale Optimization
Knowledge Science, Engineering and ManagementAbstractIn recent decades, new epidemics have seriously endangered people’s lives and are now the leading cause of death in the world. The prevention of pandemic diseases has therefore become a top priority today. However, effective prevention remains a ...
-
Epidemic Prediction Algorithm Based on Deep Learning and Epidemic Dynamics
ICMLCA '23: Proceedings of the 2023 4th International Conference on Machine Learning and Computer ApplicationDisease prediction has always been an important research topic. The purpose of this algorithm is to combine climate factors and use deep learning methods to establish a model that can predict the development of infectious disease epidemic. This model ...
Comments