Epidemiology-informed Graph Neural Network for Heterogeneity-aware Epidemic Forecasting

• URL: http://arxiv.org/abs/2411.17372v1
• Date: Tue, 26 Nov 2024 12:29:45 GMT
• Title: Epidemiology-informed Graph Neural Network for Heterogeneity-aware Epidemic Forecasting
• Authors: Yufan Zheng, Wei Jiang, Alexander Zhou, Nguyen Quoc Viet Hung, Choujun Zhan, Tong Chen,
• Abstract summary: Recent studies have demonstrated the strong potential of of-temporal neural networks (STGNNs) in extracting heterogeneous-temporal epidemic patterns. HeatGNN learns epidemiology-informed locations embedding different locations that reflect their own transmission mechanisms over time. HeatGNN outperforms various strong baselines of HeatHeat on different sizes of Heat.
• Score: 46.63739322178277
• License:
• Abstract: Among various spatio-temporal prediction tasks, epidemic forecasting plays a critical role in public health management. Recent studies have demonstrated the strong potential of spatio-temporal graph neural networks (STGNNs) in extracting heterogeneous spatio-temporal patterns for epidemic forecasting. However, most of these methods bear an over-simplified assumption that two locations (e.g., cities) with similar observed features in previous time steps will develop similar infection numbers in the future. In fact, for any epidemic disease, there exists strong heterogeneity of its intrinsic evolution mechanisms across geolocation and time, which can eventually lead to diverged infection numbers in two ``similar'' locations. However, such mechanistic heterogeneity is non-trivial to be captured due to the existence of numerous influencing factors like medical resource accessibility, virus mutations, mobility patterns, etc., most of which are spatio-temporal yet unreachable or even unobservable. To address this challenge, we propose a Heterogeneous Epidemic-Aware Transmission Graph Neural Network (HeatGNN), a novel epidemic forecasting framework. By binding the epidemiology mechanistic model into a GNN, HeatGNN learns epidemiology-informed location embeddings of different locations that reflect their own transmission mechanisms over time. With the time-varying mechanistic affinity graphs computed with the epidemiology-informed location embeddings, a heterogeneous transmission graph network is designed to encode the mechanistic heterogeneity among locations, providing additional predictive signals to facilitate accurate forecasting. Experiments on three benchmark datasets have revealed that HeatGNN outperforms various strong baselines. Moreover, our efficiency analysis verifies the real-world practicality of HeatGNN on datasets of different sizes.

Related papers

• Epidemiology-informed Network for Robust Rumor Detection [59.89351792706995]
  We propose a novel Epidemiology-informed Network (EIN) that integrates epidemiological knowledge to enhance performance. To adapt epidemiology theory to rumor detection, it is expected that each users stance toward the source information will be annotated. Our experimental results demonstrate that the proposed EIN not only outperforms state-of-the-art methods on real-world datasets but also exhibits enhanced robustness across varying tree depths.
  arXiv  Detail & Related papers  (2024-11-20T00:43:32Z)
• Epidemiology-Aware Neural ODE with Continuous Disease Transmission Graph [14.28921518883576]
  We introduce an innovative end-to-end framework called Epidemiology-Aware Neural ODE with Continuous Disease Transmission Graph (EARTH) We first propose EANO, which seamlessly integrates the neural ODE approach with the epidemic mechanism. We also introduce GLTG to model global infection trends and leverage these signals to guide local transmission dynamically.
  arXiv  Detail & Related papers  (2024-09-28T04:07:16Z)
• A Review of Graph Neural Networks in Epidemic Modeling [14.28921518883576]
  Since the onset of the COVID-19 pandemic, there has been a growing interest in studying epidemiological models. Graph Neural Networks have emerged as a progressively popular tool in epidemic research.
  arXiv  Detail & Related papers  (2024-03-28T21:54:48Z)
• Out-of-Distribution Generalized Dynamic Graph Neural Network with Disentangled Intervention and Invariance Promotion [61.751257172868186]
  Dynamic graph neural networks (DyGNNs) have demonstrated powerful predictive abilities by exploiting graph and temporal dynamics. Existing DyGNNs fail to handle distribution shifts, which naturally exist in dynamic graphs.
  arXiv  Detail & Related papers  (2023-11-24T02:42:42Z)
• MSGNN: Multi-scale Spatio-temporal Graph Neural Network for Epidemic Forecasting [4.635793210136456]
  Infectious disease forecasting has been a key focus and proved to be crucial in controlling epidemic. Current models broaden receptive fields by scaling the depth of graph neural networks (GNNs) We devise the Multi-scale Spatio-temporal Graph Neural Network (MSGNN) based on an innovative multi-scale view.
  arXiv  Detail & Related papers  (2023-08-30T08:21:56Z)
• MPSTAN: Metapopulation-based Spatio-Temporal Attention Network for Epidemic Forecasting [2.0297284948237366]
  We propose a 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-temporal model and adaptively defining inter-patch interactions.
  arXiv  Detail & Related papers  (2023-06-15T18:12:55Z)
• Data-Centric Epidemic Forecasting: A Survey [56.99209141838794]
  This survey delves into various data-driven methodological and practical advancements. We enumerate the large number of epidemiological datasets and novel data streams that are relevant to epidemic forecasting. We also discuss experiences and challenges that arise in real-world deployment of these forecasting systems.
  arXiv  Detail & Related papers  (2022-07-19T16:15:11Z)
• Epicasting: An Ensemble Wavelet Neural Network (EWNet) for Forecasting Epidemics [2.705025060422369]
  Infectious diseases remain among the top contributors to human illness and death worldwide. Forecasts of epidemics can assist stakeholders in tailoring countermeasures to the situation at hand.
  arXiv  Detail & Related papers  (2022-06-21T19:31:25Z)
• Handling Distribution Shifts on Graphs: An Invariance Perspective [78.31180235269035]
  We formulate the OOD problem on graphs and develop a new invariant learning approach, Explore-to-Extrapolate Risk Minimization (EERM) EERM resorts to multiple context explorers that are adversarially trained to maximize the variance of risks from multiple virtual environments. We prove the validity of our method by theoretically showing its guarantee of a valid OOD solution.
  arXiv  Detail & Related papers  (2022-02-05T02:31:01Z)
• STELAR: Spatio-temporal Tensor Factorization with Latent Epidemiological Regularization [76.57716281104938]
  We develop a tensor method to predict the evolution of epidemic trends for many regions simultaneously. STELAR enables long-term prediction by incorporating latent temporal regularization through a system of discrete-time difference equations. We conduct experiments using both county- and state-level COVID-19 data and show that our model can identify interesting latent patterns of the epidemic.
  arXiv  Detail & Related papers  (2020-12-08T21:21:47Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.