[2006.11913] Finding Patient Zero: Learning Contagion Source with Graph Neural Networks
Abstract:Locating the source of an epidemic, or patient zero (P0), can provide critical insights into the infection's transmission course and allow efficient resource allocation. Existing methods use graph-theoretic centrality measures and expensive message-passing algorithms, requiring knowledge of the underlying dynamics and its parameters. In this paper, we revisit this problem using graph neural networks (GNNs) to learn P0. We establish a theoretical limit for the identification of P0 in a class of epidemic models. We evaluate our method against different epidemic models on both synthetic and a real-world contact network considering a disease with history and characteristics of COVID-19. % We observe that GNNs can identify P0 close to the theoretical bound on accuracy, without explicit input of dynamics or its parameters. In addition, GNN is over 100 times faster than classic methods for inference on arbitrary graph topologies. Our theoretical bound also shows that the epidemic is like a ticking clock, emphasizing the importance of early contact-tracing. We find a maximum time after which accurate recovery of the source becomes impossible, regardless of the algorithm used.
Abstract:Locating the source of an epidemic, or patient zero (P0), can provide critical insights into the infection's transmission course and allow efficient resource allocation. Existing methods use graph-theoretic centrality measures and expensive message-passing algorithms, requiring knowledge of the underlying dynamics and its parameters. In this paper, we revisit this problem using graph neural networks (GNNs) to learn P0. We establish a theoretical limit for the identification of P0 in a class of epidemic models. We evaluate our method against different epidemic models on both synthetic
Explore this link on the map →related reading
- Source detection on networks using spatial temporal graph convolutional networks | IEEE Conference Publication | IEEE Xploreieeexplore.ieee.org
- A Gentle Introduction to Graph Neural Networksdistill.pub
- Understanding Convolutions on Graphsdistill.pub
- Beyond Message Passing: a Physics-Inspired Paradigm for Graph Neural Networksthegradient.pub
- Centrality - Wikipediaen.wikipedia.org
- Paper Trailspapertrailshq.com
- Betweenness centrality - Wikipediaen.wikipedia.org
- Graph Convolutional Networks | Thomas Kipf | Google DeepMindtkipf.github.io
- Girvan–Newman algorithm - Wikipediaen.wikipedia.org
- What are graph algorithms? A comprehensive guideneo4j.com
- What Happens Next? COVID-19 Futures, Explained With Playable Simulationsncase.me
- Bayesian network - Wikipediaen.wikipedia.org