Estimate Epidemiological Parameters given Partial Observations based on Algebraically Observable PINNs

Mizuka Komatsu
2024-07-17
Abstract:In this study, we considered the problem of estimating epidemiological parameters based on physics-informed neural networks (PINNs). In practice, not all trajectory data corresponding to the population estimated by epidemic models can be obtained, and some observed trajectories are noisy. Learning PINNs to estimate unknown epidemiological parameters using such partial observations is challenging. Accordingly, we introduce the concept of algebraic observability into PINNs. The validity of the proposed PINN, named as an algebraically observable PINNs, in terms of estimation parameters and prediction of unobserved variables, is demonstrated through numerical experiments.
Machine Learning,Dynamical Systems,Populations and Evolution
What problem does this paper attempt to address?
### What problem does this paper attempt to solve? This paper aims to address the challenge of estimating epidemiological parameters based on partially observed data. Specifically, the paper focuses on how to use Physics - Informed Neural Networks (PINNs) to accurately estimate the unknown parameters in epidemiological models and predict unobserved variables when only partial observed data are available and there is noise. #### Research Background In practical applications, epidemiological models (such as SIR and SEIR models) are used to describe the state changes of different populations during the spread of infectious diseases. These models rely on a set of parameters, such as the infection rate, the latent - period conversion rate, and the removal rate. However, in reality, we often cannot obtain the complete trajectory data of all population states, and the existing observed data may contain noise. This makes it very difficult to estimate these parameters using traditional methods. #### Main Problems 1. **Partially Observed Data**: Not all trajectory data related to the population can be obtained, and some observed trajectories also have noise. 2. **Challenges in Parameter Estimation**: When only partial observed data are available, it is very challenging to use traditional PINNs for parameter estimation. #### Solutions To solve the above problems, the author introduced the concept of algebraic observability and applied it to PINNs. Through this method, even when only partial observed data are available, the unknown parameters can be estimated more accurately and the unobserved variables can be predicted. #### Specific Methods - **SEIR Model**: As an example, the paper selected the SEIR model to describe the spread of infectious diseases: \[ \begin{aligned} \dot{S} &= -\beta SI, \\ \dot{E} &= \beta SI - \epsilon E, \\ \dot{I} &= \epsilon E - \gamma I, \\ \dot{R} &= \gamma I, \\ S + E + I + R &= 1. \end{aligned} \] where \( S \), \( E \), \( I \) and \( R \) represent the proportions of susceptibles, exposed, infected and removed respectively; \( \beta \), \( \epsilon \) and \( \gamma \) represent the infection rate, the latent - period conversion rate and the removal rate respectively. - **Algebraic Observability**: By introducing algebraic observability, polynomial equations can be constructed to represent the relationships between unobserved variables, observed variables and their derivatives. For example, for \( E \) in the SEIR model, its algebraic observability can be represented by the following equation: \[ E=\frac{\dot{I}+\gamma I}{\epsilon}. \] - **Improved PINNs**: Based on algebraic observability, Algebraically Observable PINNs were proposed. Through this improved PINNs, data of unobserved variables can be generated when only partial observed data are available, and parameter estimation can be further optimized. #### Experimental Results The effectiveness of the proposed method was verified through numerical experiments. The experimental results show that, compared with the baseline method (i.e., traditional PINNs), Algebraically Observable PINNs can estimate unknown parameters more accurately and better fit the observed data and predict unobserved variables. #### Summary This paper solves the problem of estimating epidemiological parameters under the condition of partially observed data by introducing the concept of algebraic observability, and improves the accuracy of parameter estimation and the model prediction ability.