Awesome Epidemic Modeling Papers

📢📢📢 [KDD 2024] This repository keeps track of the latest papers on epidemic modeling with deep learning methods. We categorize them based on individual tasks.

If you want to add new entries, please make PRs with the same format.

You can follow our latest survey to stay updated with the most recent advancements:

🔥🔥🔥[A Review of Graph Neural Networks in Epidemic Modeling]

We also recommend 🔥EpiLearn, a python library dedicated to boosting epidemic modeling and analysis with machine learning.

If you find this repo helpful, we would appreciate it if you could kindly cite our survey.

@article{liu2024review,
  title={A Review of Graph Neural Networks in Epidemic Modeling},
  author={Zewen Liu, Guancheng Wan, B. Aditya Prakash, Max S. Y. Lau, Wei Jin},
  journal={Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD)},
  year={2024}
}

Notes:

The current version only includes papers related to GNNs and we are working on adding more nonGNN papers.

This repo will be consistently updated to keep track of the latest papers related to deep learning in epidemic modeling.

Catalog

• Awesome Epidemic Modeling Papers
  • Notes:
  • Catalog
  • Toolkit/Library/Package
    • Python
    • R
  • Detection
  • Surveillance
  • Projection
  • Prediction
  • Classic Methods
  • Other Related Surveys

Toolkit/Library/Package

Python

1. [2024] 🔥EpiLearn: A Python Library for Machine Learning in Epidemic Modeling [paper] [code]

   • EpiLearn is a Pytorch-based machine learning tool-kit for epidemic data modeling and analysis. We provide numerour features including:
     • Implementation of Epidemic Models
     • Simulation of Epidemic Spreading
     • Visualization of Epidemic Data
     • Unified Pipeline for Epidemic Tasks

2. [2021] Eir: a python package for epidemic simulation [paper] [code]

3. [2020] Eon (epidemics on networks): a fast, flexible python package for simulation, analytic approximation, and analysis of epidemics on networks [paper] [code]

R

1. [2022] EpiEstim [paper] [code]
2. [2018] EpiModel: An R Package for Mathematical Modeling of Infectious Disease over Networks [paper] [code]
3. [2018] Epinet [paper] [code]
4. [2016] Epifit [code]
5. [2015] EpiDynamics [code]

Detection

1. [2023] Inferring Patient Zero on Temporal Networks via Graph Neural Networks [paper]
2. [2021] Source Detection on Networks Using Spatial Temporal Graph Convolutional Networks [paper] [code]
3. [2020] Finding Patient Zero: Learning Contagion Source with Graph Neural Networks [paper]

Surveillance

1. [2023] WDCIP: spatio-temporal AI-driven disease control intelligent platform for combating COVID-19 pandemic [paper]
2. [2023] COVID-19 infection inference with graph neural networks [paper]
3. [2023] Detection of Patients at Risk of Enterobacteriaceae Infection Using Graph Neural Networks: a Retrospective Study [paper]
4. [2023] Devil in the Landscapes: Inferring Epidemic Exposure Risks from Street View Imagery [paper]
5. [2023] Novel Graph Topology Learning for Spatio-Temporal Analysis of COVID-19 Spread [paper]

Projection

1. [2023] Contact Tracing and Epidemic Intervention via Deep Reinforcement Learning [paper]
2. [2021] Effective vaccination strategy using graph neural network ansatz [paper]
3. [2021] Controlling Graph Dynamics with Reinforcement Learning and Graph Neural Networks [paper]
4. [2020] Reinforced Epidemic Control: Saving Both Lives and Economy [paper] [code]

Prediction

1. [2024] Privacy-preserving individual-level covid-19 infection prediction via federated graph learning [paper] [code]

2. [2023] Research on the Forecast of the Spread of COVID-19 [paper]

3. [2023] RESEAT: Recurrent Self-Attention Network for Multi-Regional Influenza Forecasting [paper]

4. [2023] EINNs: Epidemiologically-informed Neural Networks [paper]

5. [2023] A Graph Based Deep Learning Framework for Predicting Spatio-Temporal Vaccine Hesitancy [paper]

6. [2023] HOPE: High-order Graph ODE For Modeling Interacting Dynamics [paper]

7. [2023] Epidemiology-Aware Deep Learning for Infectious Disease Dynamics Prediction [paper]

8. [2023] MepoGNN: Metapopulation Epidemic Forecasting with Graph Neural Network [paper][code]

9. [2023] Metapopulation Graph Neural Networks: Deep Metapopulation Epidemic Modeling with Human Mobility [paper]

10. [2023] Forecasting Infections with Spatio-Temporal Graph Neural Networks: A Case Study of the Dutch SARS-CoV-2 Spread [paper]

11. [2023] MSGNN: Multi-scale Spatio-temporal Graph Neural Network for Epidemic Forecasting [paper]

12. [2023] Dynamic Adaptive Spatio--Temporal Graph Network for COVID-19 Forecasting [paper]

13. [2023] Predicting COVID-19 pandemic by spatio-temporal graph neural networks: A New Zealand's study [paper] [code]

14. [2023] Graph Neural Network Modeling of Web Search Activity for Real-time Pandemic Forecasting [paper]

15. [2023] Enhancing Spatial Spread Prediction of Infectious Diseases through Integrating Multi-scale Human Mobility Dynamics [paper]

16. [2023] Predicting Influenza with Pandemic-Awareness via Dynamic Virtual Graph Significance Networks [paper]

17. [2023] DeepDynaForecast: Phylogenetic-informed Graph Deep Learning for Epidemic Transmission Dynamic Prediction [paper] [code]

18. [2023] Human Mobility Modeling during the COVID-19 Pandemic via Deep Graph Diffusion Infomax [paper]

19. [2022] EpiGNN: Exploring Spatial Transmission with Graph Neural Network for Regional Epidemic Forecasting [paper] [code]

20. [2022] Self-Attention-Based Deep Learning Network for Regional Influenza Forecasting [paper]

21. [2022] Estimating the State of Epidemics Spreading with Graph Neural Networks [paper] [code]

22. [2022] Adaptively Temporal Graph Convolution Model for Epidemic Prediction of Multiple Age Groups [paper]

23. [2022] CausalGNN: Causal-Based Graph Neural Networks for Spatio-Temporal Epidemic Forecasting [paper]

24. [2022] Visualization Method for the Spreading Curve of COVID-19 in Universities Using GNN [paper]

25. [2022] Combining Graph Neural Networks and Spatio-Temporal Disease Models to Improve the Prediction of Weekly COVID-19 Cases in Germany [paper]

26. [2022] Hierarchical Spatio-Temporal Graph Neural Networks for Pandemic Forecasting [paper]

27. [2021] HierST: A Unified Hierarchical Spatial-temporal Framework for COVID-19 Trend Forecasting [paper] [code]

28. [2021] A Human Mobility Data Driven Hybrid GNN+RNN Based Model For Epidemic Prediction [paper]

29. [2021] Integrating LSTMs and GNNs for COVID-19 Forecasting [paper] [code]

30. [2021] Prediction of the Effects of Epidemic Spreading with Graph Neural Networks [paper] [code]

31. [2021] Deep learning of contagion dynamics on complex networks [paper] [code] [data]

32. [2021] Transfer Graph Neural Networks for Pandemic Forecasting [paper] [code]

33. [2021] Coupled Graph ODE for Learning Interacting System Dynamics [paper] [code]

34. [2021] Predicting the Dynamics of the COVID-19 Pandemic in the United States Using Graph Theory-Based Neural Networks [paper] [code]

35. [2020] An Epidemiological Neural Network Exploiting Dynamic Graph Structured Data Applied to the COVID-19 Outbreak [paper] [code]

36. [2020] STAN: Spatio-Temporal Attention Network for Pandemic Prediction Using Real-World Evidence [paper]

37. [2020] Examining COVID-19 Forecasting Using Spatio-Temporal Graph Neural Networks [paper]

38. [2019] A Study on Graph-Structured Recurrent Neural Networks and Sparsification with Application to Epidemic Forecasting [paper]

Classic Methods

1. [2023] EpiMob: Interactive Visual Analytics of Citywide Human Mobility Restrictions for Epidemic Control [paper]
2. [2021] Countrywide Origin-Destination Matrix Prediction and Its Application for COVID-19 [paper]
3. [2021] Networks and the epidemiology of infectious disease [paper]
4. [2020] Modelling the impact of testing, contact tracing and household quarantine on second waves of COVID-19 [paper]
5. [2020] Modelling transmission and control of the COVID-19 pandemic in Australia [paper]
6. [2020] Inferring change points in the spread of COVID-19 reveals the effectiveness of interventions [paper]
7. [2020] Monitoring Italian COVID-19 spread by a forced SEIRD model [paper]
8. [2018] Real-time forecasting of infectious disease dynamics with a stochastic semi-mechanistic model [paper]
9. [2017] Ethics of Epidemics, Research and Surveillance: a WHO Workshop Report [paper]
10. [2017] Reproduction numbers of infectious disease models [paper]
11. [2013] Forecasting methods and models of disease spread [paper]
12. [2009] Multiscale mobility networks and the spatial spreading of infectious diseases [paper]
13. [2008] Mathematical models of infectious disease transmission [paper]
14. [1927] A contribution to the mathematical theory of epidemics [paper]

Other Related Surveys

1. [2022] Data-Centric Epidemic Forecasting: A Survey [paper]
2. [2022] Machine Learning, Deep Learning, and Mathematical Models to Analyze Forecasting and Epidemiology of COVID-19: A Systematic Literature Review [paper]
3. [2022] Significance of deep learning for Covid-19: state-of-the-art review [paper]
4. [2022] Deep learning for Covid-19 forecasting: State-of-the-art review [paper]
5. [2022] A Survey on Mathematical, Machine Learning and Deep Learning Models for COVID-19 Transmission and Diagnosis [paper]
6. [2021] Deep Learning applications for COVID-19 [paper]
7. [2021] Deep Learning for Virus-Spreading Forecasting: a Brief Survey [paper]