[2603.02349] Learning graph topology from metapopulation epidemic encoder-decoder

Computer Science > Machine Learning arXiv:2603.02349 (cs) [Submitted on 2 Mar 2026] Title:Learning graph topology from metapopulation epidemic encoder-decoder Authors:Xin Li, Jonathan Cohen, Shai Pilosof, Rami Puzis View a PDF of the paper titled Learning graph topology from metapopulation epidemic encoder-decoder, by Xin Li and 3 other authors View PDF HTML (experimental) Abstract:Metapopulation epidemic models are a valuable tool for studying large-scale outbreaks. With the limited availability of epidemic tracing data, it is challenging to infer the essential constituents of these models, namely, the epidemic parameters and the relevant mobility network between subpopulations. Either one of these constituents can be estimated while assuming the other; however, the problem of their joint inference has not yet been solved. Here, we propose two encoder-decoder deep learning architectures that infer metapopulation mobility graphs from time-series data, with and without the assumption of epidemic model parameters. Evaluation across diverse random and empirical mobility networks shows that the proposed approach outperforms the state-of-the-art topology inference. Further, we show that topology inference improves dramatically with data on additional pathogens. Our study establishes a robust framework for simultaneously inferring epidemic parameters and topology, addressing a persistent gap in modeling disease propagation. Subjects: Machine Learning (cs.LG) Cite as: arXiv:2603....