arXiv · 2203.05018
Application of neural-network hybrid models in estimating the infection functions of nonlinear epidemic models
Abstract
Hybrid neural network models combine the advantages of a neural network's fitting functionality with differential equation models to reflect actual physical processes and are widely used in analyzing time-series data. Most related studies have focused on linear hybrid models, but only a few have examined nonlinear problems. In this work, we use a hybrid nonlinear epidemic neural network as the entry point to study its power in predicting the correct infection function of an epidemic model. To achieve this goal, we combine the bifurcation theory of the nonlinear differential model with the mean-squared error loss and design a novel loss function to ensure model trainability. Furthermore, we find the unique existence conditions supporting ordinary differential equations to estimate the correct infection function. Using the Runge Kutta method, we perform numerical experiments on our proposed model and verify its soundness. We also apply it to real COVID-19 data to accurately discover the change law of its infectivity.
Explore related subjects
Keep this discovery
Chentong Li, Changsheng Zhou, Junmin Liu, Yao Rong. 2022-03-05. Application of neural-network hybrid models in estimating the infection functions of nonlinear epidemic models. https://arxiv.org/abs/2203.05018
Cite the original work for its findings. Save a collection to share your selection of sources.