Neural parameter calibration and uncertainty quantification for epidemic forecasting
File(s)journal.pone.0306704.pdf (2.26 MB)
Published version
Author(s)
Gaskin, Thomas
Conrad, Tim
Pavliotis, Grigorios A
Schütte, Christof
Type
Journal Article
Abstract
The recent COVID-19 pandemic has thrown the importance of accurately forecasting conta gion dynamics and learning infection parameters into sharp focus. At the same time, effec tive policy-making requires knowledge of the uncertainty on such predictions, in order, for
instance, to be able to ready hospitals and intensive care units for a worst-case scenario
without needlessly wasting resources. In this work, we apply a novel and powerful computa tional method to the problem of learning probability densities on contagion parameters and
providing uncertainty quantification for pandemic projections. Using a neural network, we
calibrate an ODE model to data of the spread of COVID-19 in Berlin in 2020, achieving both
a significantly more accurate calibration and prediction than Markov-Chain Monte Carlo
(MCMC)-based sampling schemes. The uncertainties on our predictions provide meaningful
confidence intervals e.g. on infection figures and hospitalisation rates, while training and
running the neural scheme takes minutes where MCMC takes hours. We show conver gence of our method to the true posterior on a simplified SIR model of epidemics, and also
demonstrate our method’s learning capabilities on a reduced dataset, where a complex
model is learned from a small number of compartments for which data is available
instance, to be able to ready hospitals and intensive care units for a worst-case scenario
without needlessly wasting resources. In this work, we apply a novel and powerful computa tional method to the problem of learning probability densities on contagion parameters and
providing uncertainty quantification for pandemic projections. Using a neural network, we
calibrate an ODE model to data of the spread of COVID-19 in Berlin in 2020, achieving both
a significantly more accurate calibration and prediction than Markov-Chain Monte Carlo
(MCMC)-based sampling schemes. The uncertainties on our predictions provide meaningful
confidence intervals e.g. on infection figures and hospitalisation rates, while training and
running the neural scheme takes minutes where MCMC takes hours. We show conver gence of our method to the true posterior on a simplified SIR model of epidemics, and also
demonstrate our method’s learning capabilities on a reduced dataset, where a complex
model is learned from a small number of compartments for which data is available
Editor(s)
Arunachalam, Viswanathan
Date Issued
2024-10
Date Acceptance
2024-06-21
Citation
PLoS One, 2024, 19 (10)
ISSN
1932-6203
Publisher
Public Library of Science (PLoS)
Journal / Book Title
PLoS One
Volume
19
Issue
10
Copyright Statement
© 2024 Gaskin et al. This is an open
access article distributed under the terms of the
Creative Commons Attribution License, which
permits unrestricted use, distribution, and
reproduction in any medium, provided the original
author and source are credited.
access article distributed under the terms of the
Creative Commons Attribution License, which
permits unrestricted use, distribution, and
reproduction in any medium, provided the original
author and source are credited.
License URL
Identifier
http://dx.doi.org/10.1371/journal.pone.0306704
Publication Status
Published
Article Number
e0306704
Date Publish Online
2024-10-17