Scalable calibration of individual-based epidemic models through categorical approximations

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Rimella, Lorenzo, Whiteley, Nick, Jewell, Chris, Fearnhead, Paul, Whitehouse, Michael
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908787067060224
author Rimella, Lorenzo
Whiteley, Nick
Jewell, Chris
Fearnhead, Paul
Whitehouse, Michael
author_facet Rimella, Lorenzo
Whiteley, Nick
Jewell, Chris
Fearnhead, Paul
Whitehouse, Michael
contents Traditional compartmental models capture population-level dynamics but fail to characterize individual-level risk. The computational cost of exact likelihood evaluation for partially observed individual-based models, however, grows exponentially with the population size, necessitating approximate inference. Existing sampling-based methods usually require multiple simulations of the individuals in the population and rely on bespoke proposal distributions or summary statistics. We propose a deterministic approach to approximating the likelihood using categorical distributions. The approximate likelihood is amenable to automatic differentiation so that parameters can be estimated by maximization or posterior sampling using standard software libraries such as Stan or TensorFlow with little user effort. We prove the consistency of the maximum approximate likelihood estimator. We empirically test our approach on several classes of individual-based models for epidemiology: different sets of disease states, individual-specific transition rates, spatial interactions, under-reporting and misreporting. We demonstrate ground truth recovery and comparable marginal log-likelihood values at substantially reduced cost compared to competitor methods. Finally, we show the scalability and effectiveness of our approach with a real-world application on the 2001 UK Foot-and-Mouth outbreak, where the simplicity of the CAL allows us to include 162775 farms.
format Preprint
id arxiv_https___arxiv_org_abs_2501_03950
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Scalable calibration of individual-based epidemic models through categorical approximations
Rimella, Lorenzo
Whiteley, Nick
Jewell, Chris
Fearnhead, Paul
Whitehouse, Michael
Methodology
Traditional compartmental models capture population-level dynamics but fail to characterize individual-level risk. The computational cost of exact likelihood evaluation for partially observed individual-based models, however, grows exponentially with the population size, necessitating approximate inference. Existing sampling-based methods usually require multiple simulations of the individuals in the population and rely on bespoke proposal distributions or summary statistics. We propose a deterministic approach to approximating the likelihood using categorical distributions. The approximate likelihood is amenable to automatic differentiation so that parameters can be estimated by maximization or posterior sampling using standard software libraries such as Stan or TensorFlow with little user effort. We prove the consistency of the maximum approximate likelihood estimator. We empirically test our approach on several classes of individual-based models for epidemiology: different sets of disease states, individual-specific transition rates, spatial interactions, under-reporting and misreporting. We demonstrate ground truth recovery and comparable marginal log-likelihood values at substantially reduced cost compared to competitor methods. Finally, we show the scalability and effectiveness of our approach with a real-world application on the 2001 UK Foot-and-Mouth outbreak, where the simplicity of the CAL allows us to include 162775 farms.
title Scalable calibration of individual-based epidemic models through categorical approximations
topic Methodology
url https://arxiv.org/abs/2501.03950