Deep evolutionary fusion neural network: a new prediction standard for infectious disease incidence rates

• Published in: BMC bioinformatics Vol. 25; no. 1; pp. 38 - 17
• Main Authors: Yao, Tianhua, Chen, Xicheng, Wang, Haojia, Gao, Chengcheng, Chen, Jia, Yi, Dali, Wei, Zeliang, Yao, Ning, Li, Yang, Yi, Dong, Wu, Yazhou
• Format: Journal Article
• Language: English
• Published: England BioMed Central Ltd 23.01.2024; BioMed Central; BMC
• Subjects: Accuracy; Algorithms; Autoregressive models; Benchmarks; Brucellosis; Communicable diseases; Data integration; Data mining; Datasets; Epidemics; Evolution; Hepatitis B; Hepatitis C; Infectious diseases; Internet; Keywords; Long short-term memory; Machine learning; Medical research; Medicine, Experimental; Meta-heuristic algorithm; Methods; Model accuracy; Neural network; Neural networks; Neuroevolution; Optimization; Optimization models; Prediction models; Search engines; Time series; Trends; Tuberculosis; China; Infectious diseases; Meta-heuristic algorithm; Neural network; Neuroevolution; Time series
• ISSN: 1471-2105; 1471-2105
• DOI: 10.1186/s12859-023-05621-5

Previously, many methods have been used to predict the incidence trends of infectious diseases. There are numerous methods for predicting the incidence trends of infectious diseases, and they have exhibited varying degrees of success. However, there are a lack of prediction benchmarks that integrate linear and nonlinear methods and effectively use internet data. The aim of this paper is to develop a prediction model of the incidence rate of infectious diseases that integrates multiple methods and multisource data, realizing ground-breaking research. The infectious disease dataset is from an official release and includes four national and three regional datasets. The Baidu index platform provides internet data. We choose a single model (seasonal autoregressive integrated moving average (SARIMA), nonlinear autoregressive neural network (NAR), and long short-term memory (LSTM)) and a deep evolutionary fusion neural network (DEFNN). The DEFNN is built using the idea of neural evolution and fusion, and the DEFNN + is built using multisource data. We compare the model accuracy on reference group data and validate the model generalizability on external data. (1) The loss of SA-LSTM in the reference group dataset is 0.4919, which is significantly better than that of other single models. (2) The loss values of SA-LSTM on the national and regional external datasets are 0.9666, 1.2437, 0.2472, 0.7239, 1.4026, and 0.6868. (3) When multisource indices are added to the national dataset, the loss of the DEFNN + increases to 0.4212, 0.8218, 1.0331, and 0.8575. We propose an SA-LSTM optimization model with good accuracy and generalizability based on the concept of multiple methods and multiple data fusion. DEFNN enriches and supplements infectious disease prediction methodologies, can serve as a new benchmark for future infectious disease predictions and provides a reference for the prediction of the incidence rates of various infectious diseases.