Epidemic Source Identification Based on Infection Graph Learning

Xingyun Hong,Ting Yu,Zhao Li,Ji Zhang
DOI: https://doi.org/10.1007/978-981-97-2303-4_16
2024-01-01
Abstract:Source identification plays a critical role in various domains, including patient zero tracing, social network rumor source detection, and more, as it enables tracing propagation processes and implementing effective measures to block transmission. However, most algorithms employed to identify propagation sources in networks heavily rely on prior knowledge of the underlying propagation models and associated parameters, as different propagation patterns yield diverse outcomes. To address this limitation, we present an approach called Infection Graph Learning for Sourcing (IGLS), which utilizes multiple snapshots to learn the propagation pattern. In this method, the source identification problem is redefined as a novel graph node classification scenario solved using Graph Convolutional Networks (GCN). The IGLS model takes both the state of nodes and the network’s structure into consideration while introducing a new loss function specifically designed for the task. Furthermore, to tackle the multiple source identification problem, multi-task learning is incorporated into the IGLS model structure, identifying the number of sources and their locations simultaneously. This represents the first application of a deep learning model to address problems involving an unknown number of sources. We conducted experiments on both synthetic and real-world networks, and the results demonstrate the effectiveness and superiority of our proposed method.
What problem does this paper attempt to address?