这篇论文用COVID-19数据实测,SBI比MCMC快几十倍,精度不输,适合需要快速迭代的流行病学建模场景。
该研究将基于模拟推理(SBI)的神经后验估计用于SECIR流行病学模型的贝叶斯校准,并使用2020年德国COVID-19 ICU入住数据。在31天推断窗口中,SBI恢复的后验分布与MCMC高度一致,准确再现ICU轨迹。在更具挑战的201天重构问题中,SBI保留了主要后验结构,尽管不确定性增加。SBI在单GPU上完成31天推断仅需60-70秒,而MCMC需要约1000秒;201天任务中SBI平均157秒,MCMC超过19000秒。结果表明SBI为快速近实时疫情分析提供了高效框架。
Simulation-based inference for rapid Bayesian parameter estimation in epidemiological models: a comparison with MCMC
Mechanistic epidemiological models are widely used to support infectious disease forecasting and public-health decision making. Bayesian calibration of such models is commonly performed using Markov chain Monte Carlo (MCMC), which can become computationally expensive for high-dimensional nonlinear systems and repeated near-real-time analyses. Here, we investigate simulation-based inference (SBI) using neural posterior estimation as a scalable alternative for Bayesian calibration of a mechanistic SECIR epidemiological model using COVID-19 intensive care unit (ICU) occupancy data from Germany during 2020. We compared SBI and MCMC across multiple epidemic phases using both 31-day inference windows and a substantially more challenging 201-day reconstruction problem involving multiple transmission change points. Posterior agreement was evaluated quantitatively using Wasserstein distances and Kullback-Leibler divergences together with posterior predictive checks. Across the 31-day windows, SBI recovered posterior distributions in strong agreement with MCMC while accurately reproducing observed ICU trajectories. In the 201-day setting, SBI preserved the dominant posterior structure despite increased uncertainty. SBI, by combining CPU and GPU resources, substantially reduced computational runtime compared with MCMC, which was restricted to running on CPUs. Whereas MCMC required approximately 1000 seconds for the 31-day inference problems, SBI achieved comparable posterior and predictive performance in approximately 60-70 seconds on a single GPU. For the 201-day inference problem, SBI required an average of 157 seconds, while the MCMC runs took over 19,000 seconds. Our results demonstrate that SBI provides a rapid and computationally efficient framework for Bayesian calibration of mechanistic epidemiological models, supporting repeated near-real-time inference and rapid outbreak analysis.