Estimating the state of epidemics spreading with graph neural networks
Abhishek Tomy (Inria Grenoble Rhône-Alpes)
Matteo Razzanelli (Proxima Robotics srl)
Francesco Di Lauro (University of Oxford)
Daniela Rus (Massachusetts Institute of Technology)
C. Lieu (TU Delft - Learning & Autonomous Control, Deutsches Zentrum für Luft- und Raumfahrt (DLR))
More Info
expand_more
Other than for strictly personal use, it is not permitted to download, forward or distribute the text or part of it, without the consent of the author(s) and/or copyright holder(s), unless the work is under an open content license such as Creative Commons.
Abstract
When an epidemic spreads into a population, it is often impractical or impossible to continuously monitor all subjects involved. As an alternative, we propose using algorithmic solutions that can infer the state of the whole population from a limited number of measures. We analyze the capability of deep neural networks to solve this challenging task. We base our proposed architecture on Graph Convolutional Neural Networks. As such, it can reason on the effect of the underlying social network structure, which is recognized as the main component in spreading an epidemic. The proposed architecture can reconstruct the entire state with accuracy above 70%, as proven by two scenarios modeled on the CoVid-19 pandemic. The first is a generic homogeneous population, and the second is a toy model of the Boston metropolitan area. Note that no retraining of the architecture is necessary when changing the model.