Spatio-temporal ecological models via physics-informed neural networks for studying chronic wasting disease

Abstract

To mitigate the negative effects of emerging wildlife diseases in biodiversity and public health it is critical to accurately forecast pathogen dissemination while incorporating relevant spatio-temporal covariates. Forecasting spatio-temporal processes can often be improved by incorporating scientific knowledge about the dynamics of the process using physical models. Ecological diffusion equations are often used to model epidemiological processes of wildlife diseases where environmental factors play a role in disease spread. Physics-informed neural networks (PINNs) are deep learning algorithms that constrain neural network predictions based on physical laws and therefore are powerful forecasting models useful even in cases of limited and imperfect training data. In this paper, we develop a novel ecological modeling tool using PINNs, which fits a feedforward neural network and simultaneously performs parameter identification in a partial differential equation (PDE) with varying coefficients. We demonstrate the applicability of our model by comparing it with the commonly used Bayesian stochastic partial differential equation method and traditional machine learning approaches, showing that our proposed model exhibits superior prediction and forecasting performance when modeling chronic wasting disease in deer in Wisconsin. Furthermore, our model provides the opportunity to obtain scientific insights into spatio-temporal covariates affecting spread and growth of diseases. This work contributes to future machine learning and statistical methodology development by studying spatio-temporal processes enhanced by prior physical knowledge.