Predicting COVID-19 in China Using Hybrid AI Model

• Published in: IEEE transactions on cybernetics Vol. 50; no. 7; pp. 2891 - 2904
• Main Authors: Zheng, Nanning, Du, Shaoyi, Wang, Jianji, Zhang, He, Cui, Wenting, Kang, Zijian, Yang, Tao, Lou, Bin, Chi, Yuting, Long, Hong, Ma, Mei, Yuan, Qi, Zhang, Shupei, Zhang, Dong, Ye, Feng, Xin, Jingmin
• Format: Journal Article
• Language: English
• Published: United States IEEE 01.07.2020; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Analytical models; Artificial Intelligence; Betacoronavirus; China - epidemiology; Coronavirus disease 2019 (COVID-19) prediction; Coronavirus Infections - epidemiology; Coronaviruses; COVID-19; Disease transmission; epidemic model; Epidemics; Humans; hybrid artificial-intelligence (AI) model; Hybrid systems; Infections; Models, Statistical; Natural Language Processing; natural language processing (NLP); Pandemics; Pneumonia, Viral - epidemiology; Predictive models; SARS-CoV-2; Viral diseases; China
• ISSN: 2168-2267; 2168-2275; 2168-2275
• DOI: 10.1109/TCYB.2020.2990162

The coronavirus disease 2019 (COVID-19) breaking out in late December 2019 is gradually being controlled in China, but it is still spreading rapidly in many other countries and regions worldwide. It is urgent to conduct prediction research on the development and spread of the epidemic. In this article, a hybrid artificial-intelligence (AI) model is proposed for COVID-19 prediction. First, as traditional epidemic models treat all individuals with coronavirus as having the same infection rate, an improved susceptible-infected (ISI) model is proposed to estimate the variety of the infection rates for analyzing the transmission laws and development trend. Second, considering the effects of prevention and control measures and the increase of the public's prevention awareness, the natural language processing (NLP) module and the long short-term memory (LSTM) network are embedded into the ISI model to build the hybrid AI model for COVID-19 prediction. The experimental results on the epidemic data of several typical provinces and cities in China show that individuals with coronavirus have a higher infection rate within the third to eighth days after they were infected, which is more in line with the actual transmission laws of the epidemic. Moreover, compared with the traditional epidemic models, the proposed hybrid AI model can significantly reduce the errors of the prediction results and obtain the mean absolute percentage errors (MAPEs) with 0.52%, 0.38%, 0.05%, and 0.86% for the next six days in Wuhan, Beijing, Shanghai, and countrywide, respectively.