Predicting COVID-19 in China Using Hybrid AI Model

Nanning Zheng,Shaoyi Du,Jianji Wang,He Zhang,Wenting Cui,Zijian Kang,Tao Yang,Bin Lou,Yuting Chi,Hong Long,Mei Ma,Qi Yuan,Shupei Zhang,Dong Zhang,Feng Ye,Jingmin Xin
DOI: https://doi.org/10.1109/tcyb.2020.2990162
IF: 11.8
2020-01-01
IEEE Transactions on Cybernetics
Abstract: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.
What problem does this paper attempt to address?