SPADE4: Sparsity and Delay Embedding based Forecasting of Epidemics

Abstract

Predicting the evolution of diseases is challenging, especially when the data availability is scarce and incomplete. The most popular tools for modelling and predicting infectious disease epidemics are compartmental models. They stratify the population into compartments according to health status and model the dynamics of these compartments using dynamical systems. However, these predefined systems may not capture the true dynamics of the epidemic due to the complexity of the disease transmission and human interactions. In order to overcome this drawback, we propose Sparsity and Delay Embedding based Forecasting (SPADE4) for predicting epidemics. SPADE4 predicts the future trajectory of an observable variable without the knowledge of the other variables or the underlying system. We use random features model with sparse regression to handle the data scarcity issue and employ Takens' delay embedding theorem to capture the nature of the underlying system from the observed variable. We show that our approach outperforms compartmental models when applied to both simulated and real data.

arXiv comment24 pages, 13 figures, 2 tables

References in corpus (16)

• Discovering governing equations from data: Sparse identification of nonlinear dynamical systems
• DeepXDE: A deep learning library for solving differential equations
• Artificial Neural Networks for Solving Ordinary and Partial Differential Equations
• Deep learning for universal linear embeddings of nonlinear dynamics
• Chaos as an Intermittently Forced Linear System
• Sparse identification of nonlinear dynamics for model predictive control in the low-data limit
• Data-Driven Forecasting of High-Dimensional Chaotic Systems with Long Short-Term Memory Networks
• Data Driven Governing Equations Approximation Using Deep Neural Networks
• Estimating the reproduction number of Ebola virus (EBOV) during the 2014 outbreak in West Africa
• Model selection for dynamical systems via sparse regression and information criteria
• Discovery of Nonlinear Multiscale Systems: Sampling Strategies and Embeddings
• The Random Feature Model for Input-Output Maps between Banach Spaces
• A Comparative Analysis of the Optimization and Generalization Property of Two-layer Neural Network and Random Feature Models Under Gradient Descent Dynamics
• HIV with contact-tracing: a case study in Approximate Bayesian Computation
• Discovery of Nonlinear Dynamical Systems using a Runge-Kutta Inspired Dictionary-based Sparse Regression Approach
• Generalization Bounds for Sparse Random Feature Expansions