Please wait a minute...
Data Analysis and Knowledge Discovery
Current Issue | Archive | Adv Search |
Predicting Major Infectious Diseases based on Grey Wolf Optimization and Multi-machine Learning: Case Study of COVID-19
Qu Zongxi,Sha Yongzhong,Li Yutong
(School of Management, Lanzhou University, Lanzhou 730000, China) (Research Center for Emergency Management, Lanzhou University, Lanzhou 730000, China)
Download:
Export: BibTeX | EndNote (RIS)      
Abstract  

[Objective] Building an accurate and effective forecasting model for major infectious diseases based on multi-machine learning can predict outbreak trends and help formulate countermeasures in advance.

[Methods] Based on the Gray Wolf Optimization algorithm, three machine learning optimal weight combinations of ANFIS, LSSVM and LSTM are searched to establish an ensemble prediction model. Experiments are designed to assess the model prediction performance by COVID-19 epidemic data.

[Results] The results show that ANFIS, LSSVM, and LSTM were suitable for confirmed case, death case, and recovery case scenarios, respectively; the R2 of the ensemble prediction model based on Gray Wolf Optimization reached 0.987, 0.993, and 0.987 for the three scenarios. The average RMSE was reduced by 38.79%, 64.40%, and 53.88% compared to the single model, respectively.

[Limitations] The model needs to be further verified by using other major infectious disease epidemic data sets.

[Conclusions] Different machine learning models have their own prediction performance, and the ensemble prediction model based on Gray Wolf Optimization can effectively merge the advantages of multiple machine learning models to obtain stable and accurate prediction results.


Key words Major infectious disease outbreak      Ensemble prediction      Grey Wolf Optimization      Machine learning      
Published: 20 June 2022
ZTFLH:  R183  
  TP181  

Cite this article:

Qu Zongxi, Sha Yongzhong, Li Yutong. Predicting Major Infectious Diseases based on Grey Wolf Optimization and Multi-machine Learning: Case Study of COVID-19 . Data Analysis and Knowledge Discovery, 0, (): 1-.

URL:

https://manu44.magtech.com.cn/Jwk_infotech_wk3/EN/10.11925/infotech.2096-3467.2021-1269     OR     https://manu44.magtech.com.cn/Jwk_infotech_wk3/EN/Y0/V/I/1

[1] Nie Hui, Wu Xiaoyan. Detecting Depression Factors with Gradient Boosting Tree and Explainable Machine Learning Model SHAP[J]. 数据分析与知识发现, 2024, 8(3): 41-52.
[2] Zhang Yunqiu, Huang Qifei, Zhu Xiang. Predicting Drug-Target Relationship Based on Relation Fusion and Bidirectional Mass Diffusion Model[J]. 数据分析与知识发现, 2024, 8(2): 155-167.
[3] Liu Zhifeng, Wang Jimin. Review of Interpretable Machine Learning for Information Resource Management[J]. 数据分析与知识发现, 2024, 8(1): 16-29.
[4] Liu Tianchang, Wang Lei, Zhu Qinghua. Predicting User Churn of Smart Home-based Care Services Based on SHAP Interpretation[J]. 数据分析与知识发现, 2024, 8(1): 40-54.
[5] Xu Chen, Zhang Wei. Detecting Crowdfunding Frauds Based on Textual and Imbalanced Data[J]. 数据分析与知识发现, 2023, 7(9): 125-135.
[6] Wei Huanan, Lei Ming, Wang Xuefeng, Yu Yin. Analyzing Evolution of Basic Research Funding Orientation: Case Study of NSF[J]. 数据分析与知识发现, 2023, 7(5): 10-20.
[7] Lin Weizhen, Liu Hongwei, Chen Yanjun, Wen Zhanming, Yi Minqi. Customer Satisfaction Modelling for Healthcare Wearable Devices Through Online Reviews[J]. 数据分析与知识发现, 2023, 7(5): 145-154.
[8] Jiang Linfu, Yuan Zhenming, Zhang Xingwei, Jiang Huaqiang, Sun Xiaoyan. Ten-Year Prediction of Coronary Heart Disease Based on PCHD-TabNet[J]. 数据分析与知识发现, 2023, 7(5): 133-144.
[9] Lv Qi, Shangguan Yanhong, Zhang Lin, Huang Ying. Interdisciplinary Measurement Based on Automatic Classification of Text Content[J]. 数据分析与知识发现, 2023, 7(4): 56-67.
[10] Qu Zongxi, Sha Yongzhong, Li Yutong. Predicting Major Infectious Diseases Based on Grey Wolf Optimization and Multi-machine Learning: Case Study of COVID-19[J]. 数据分析与知识发现, 2022, 6(8): 122-133.
[11] Zhao Yang, Yan Zhouzhou, Shen Qiqi, Li Zhonghang. Evaluating Privacy Policy for Mobile Health APPs with Machine Learning[J]. 数据分析与知识发现, 2022, 6(5): 112-126.
[12] Wang Lu, Le Xiaoqiu. Research Progress on Citation Analysis of Scientific Papers[J]. 数据分析与知识发现, 2022, 6(4): 1-15.
[13] Wang Ruojia, Yan Chengxi, Guo Fengying, Wang Jimin. Predicting Churners of Online Health Communities Based on the User Persona[J]. 数据分析与知识发现, 2022, 6(2/3): 80-92.
[14] Wu Jinhong, Mu Keliang. Automatic Identifying Abnormal Behaviors of International Journals[J]. 数据分析与知识发现, 2022, 6(2/3): 385-395.
[15] Hu Yamin, Wu Xiaoyan, Chen Fang. Review of Technology Term Recognition Studies Based on Machine Learning[J]. 数据分析与知识发现, 2022, 6(2/3): 7-17.
  Copyright © 2016 Data Analysis and Knowledge Discovery   Tel/Fax:(010)82626611-6626,82624938   E-mail:jishu@mail.las.ac.cn