Physics-informed deep learning for infectious disease forecasting
Authors:
Ying Qian,
Kui Zhang,
Éric Marty,
Avranil Basu,
Eamon B. O'Dea,
Xianqiao Wang,
Spencer Fox,
Pejman Rohani,
John M. Drake,
He Li
Abstract:
Accurate forecasting of contagious diseases is critical for public health policymaking and pandemic preparedness. We propose a new infectious disease forecasting model based on physics-informed neural networks (PINNs), an emerging scientific machine learning approach. By embedding a compartmental model into the loss function, our method integrates epidemiological theory with data, helping to preve…
▽ More
Accurate forecasting of contagious diseases is critical for public health policymaking and pandemic preparedness. We propose a new infectious disease forecasting model based on physics-informed neural networks (PINNs), an emerging scientific machine learning approach. By embedding a compartmental model into the loss function, our method integrates epidemiological theory with data, helping to prevent model overfitting. We further enhance the model with a sub-network that accounts for covariates such as mobility and cumulative vaccine doses, which influence the transmission rate. Using state-level COVID-19 data from California, we demonstrate that the PINN model accurately predicts cases, deaths, and hospitalizations, aligning well with existing benchmarks. Notably, the PINN model outperforms naive baseline forecasts and several sequence deep learning models, including Recurrent Neural Networks (RNNs), Long Short-Term Memory (LSTM) networks, Gated Recurrent Units (GRUs), and Transformers. It also achieves performance comparable to a sophisticated Gaussian infection state forecasting model that combines compartmental dynamics, a data observation model, and parameter regression. However, the PINN model features a simpler structure and is easier to implement. In summary, we systematically evaluate the PINN model's ability to forecast infectious disease dynamics, demonstrating its potential as an efficient computational tool to strengthen forecasting capabilities.
△ Less
Submitted 29 April, 2025; v1 submitted 16 January, 2025;
originally announced January 2025.
Ebola cases and health system demand in Liberia
Authors:
John M. Drake,
RajReni B. Kaul,
Laura Alexander,
Suzanne M. O'Regan,
Andrew M. Kramer,
J. Tomlin Pulliam,
Matthew J. Ferrari,
Andrew W. Park
Abstract:
In 2014, a major epidemic of human Ebola virus disease emerged in West Africa, where human-to-human transmission has now been been sustained for greater than 10 months. In the summer of 2014, there was great uncertainty about the answers to several key policy questions concerning the path to containment. In recent years, epidemic models have been used to guide public health interventions. But, mod…
▽ More
In 2014, a major epidemic of human Ebola virus disease emerged in West Africa, where human-to-human transmission has now been been sustained for greater than 10 months. In the summer of 2014, there was great uncertainty about the answers to several key policy questions concerning the path to containment. In recent years, epidemic models have been used to guide public health interventions. But, model-based policy relies on high quality causal understanding of transmission, including the availability of appropriate dynamic transmission models and reliable reporting about the sequence of case incidence for model fitting, which were lacking for this epidemic. To investigate the range of potential transmission scenarios, we developed a multi-type branching process model that incorporates key heterogeneities and time-varying parameters to reflect changing human behavior and deliberate interventions. Ensembles of this model were evaluated at a set of parameters that were both epidemiologically plausible and capable of reproducing the observed trajectory. Results suggest that epidemic outcome depends on both hospital capacity and individual behavior. The model predicts that if hospital capacity is not increased soon, then transmission may outpace the rate of isolation and the ability to provide care for the ill, infectious, and dying. Similarly, containment will probably require individuals to adopt behaviors that increase the rates of case identification and isolation and secure burial of the deceased. Given current knowledge, it is uncertain that this epidemic will be contained even with 99% hospitalization rate at the currently projected hospital capacity.
△ Less
Submitted 30 October, 2014;
originally announced October 2014.
