We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.SI

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Social and Information Networks

Title: Transfer Graph Neural Networks for Pandemic Forecasting

Abstract: The recent outbreak of COVID-19 has affected millions of individuals around the world and has posed a significant challenge to global healthcare. From the early days of the pandemic, it became clear that it is highly contagious and that human mobility contributes significantly to its spread. In this paper, we study the impact of population movement on the spread of COVID-19, and we capitalize on recent advances in the field of representation learning on graphs to capture the underlying dynamics. Specifically, we create a graph where nodes correspond to a country's regions and the edge weights denote human mobility from one region to another. Then, we employ graph neural networks to predict the number of future cases, encoding the underlying diffusion patterns that govern the spread into our learning model. Furthermore, to account for the limited amount of training data, we capitalize on the pandemic's asynchronous outbreaks across countries and use a model-agnostic meta-learning based method to transfer knowledge from one country's model to another's. We compare the proposed approach against simple baselines and more traditional forecasting techniques in 3 European countries. Experimental results demonstrate the superiority of our method, highlighting the usefulness of GNNs in epidemiological prediction. Transfer learning provides the best model, highlighting its potential to improve the accuracy of the predictions in case of secondary waves, if data from past/parallel outbreaks is utilized.
Comments: 9 pages, AAAI 2021
Subjects: Social and Information Networks (cs.SI); Machine Learning (cs.LG); Machine Learning (stat.ML)
Cite as: arXiv:2009.08388 [cs.SI]
  (or arXiv:2009.08388v5 [cs.SI] for this version)

Submission history

From: George Panagopoulos [view email]
[v1] Thu, 10 Sep 2020 13:23:52 GMT (1201kb,D)
[v2] Sat, 7 Nov 2020 16:00:31 GMT (2551kb,D)
[v3] Mon, 16 Nov 2020 18:43:17 GMT (2664kb,D)
[v4] Fri, 11 Dec 2020 10:32:36 GMT (2604kb,D)
[v5] Mon, 12 Apr 2021 14:53:57 GMT (2604kb,D)

Link back to: arXiv, form interface, contact.