Bayesian inference of heterogeneous epidemic models: Application to COVID-19 spread accounting for long-term care facilities

Abstract

We propose a high dimensional Bayesian inference framework for learning heterogeneous dynamics of a COVID-19 model, with a specific application to the dynamics and severity of COVID-19 inside and outside long-term care (LTC) facilities. We develop a heterogeneous compartmental model that accounts for the heterogeneity of the time-varying spread and severity of COVID-19 inside and outside LTC facilities, which is characterized by time-dependent stochastic processes and time-independent parameters in $\sim$1500 dimensions after discretization. To infer these parameters, we use reported data on the number of confirmed, hospitalized, and deceased cases with suitable post-processing in both a deterministic inversion approach with appropriate regularization as a first step, followed by Bayesian inversion with proper prior distributions. To address the curse of dimensionality and the ill-posedness of the high-dimensional inference problem, we propose use of a dimension-independent projected Stein variational gradient descent method, and demonstrate the intrinsic low-dimensionality of the inverse problem. We present inference results with quantified uncertainties for both New Jersey and Texas, which experienced different epidemic phases and patterns. Moreover, we also present forecasting and validation results based on the empirical posterior samples of our inference for the future trajectory of COVID-19.

References in corpus (13)

• Dimension-independent likelihood-informed MCMC
• Scalable and efficient algorithms for the propagation of uncertainty from data through inference to prediction for large-scale problems, with application to flow of the Antarctic ice sheet
• Likelihood-informed dimension reduction for nonlinear inverse problems
• Geometric MCMC for Infinite-Dimensional Inverse Problems
• Scalable posterior approximations for large-scale Bayesian inverse problems via likelihood-informed parameter and state reduction
• Solving Large-Scale PDE-constrained Bayesian Inverse Problems with Riemann Manifold Hamiltonian Monte Carlo
• Message Passing Stein Variational Gradient Descent
• Taylor approximation and variance reduction for PDE-constrained optimal control under uncertainty
• Stein Variational Gradient Descent With Matrix-Valued Kernels
• Hessian-based adaptive sparse quadrature for infinite-dimensional Bayesian inverse problems
• A Stein variational Newton method
• High-Dimensional Stochastic Design Optimization by Adaptive-Sparse Polynomial Dimensional Decomposition
• Projected Stein Variational Newton: A Fast and Scalable Bayesian Inference Method in High Dimensions

Cited by in corpus (3)

• Optimal design of acoustic metamaterial cloaks under uncertainty
• Residual-based error correction for neural operator accelerated infinite-dimensional Bayesian inverse problems
• Taylor approximation for chance constrained optimization problems governed by partial differential equations with high-dimensional random parameters