Approaching epidemiological dynamics of COVID-19 with physics-informed neural networks

Abstract

A physics-informed neural network (PINN) embedded with the susceptible-infected-removed (SIR) model is devised to understand the temporal evolution dynamics of infectious diseases. Firstly, the effectiveness of this approach is demonstrated on synthetic data as generated from the numerical solution of the susceptible-asymptomatic-infected-recovered-dead (SAIRD) model. Then, the method is applied to COVID-19 data reported for Germany and shows that it can accurately identify and predict virus spread trends. The results indicate that an incomplete physics-informed model can approach more complicated dynamics efficiently. Thus, the present work demonstrates the high potential of using machine learning methods, e.g., PINNs, to study and predict epidemic dynamics in combination with compartmental models.

arXiv comment23 pages, 14 figures

References in corpus (9)

• Adam: A Method for Stochastic Optimization
• The No-U-Turn Sampler: Adaptively Setting Path Lengths in Hamiltonian Monte Carlo
• Data Driven Governing Equations Approximation Using Deep Neural Networks
• Bayesian Physics-Informed Neural Networks for real-world nonlinear dynamical systems
• SEAIR epidemic spreading model of COVID-19
• On generalized residue network for deep learning of unknown dynamical systems
• Physics-Guided Deep Learning for Dynamical Systems: A Survey
• Dynamic tracking with model-based forecasting for the spread of the COVID-19 pandemic
• Data-driven approaches for predicting spread of infectious diseases through DINNs: Disease Informed Neural Networks