ALeRT-COVID: Attentive Lockdown-awaRe Transfer Learning for Predicting COVID-19 Pandemics in Different Countries
Tóm tắt
Countries across the world are in different stages of COVID-19 trajectory, among which many have implemented lockdown measures to prevent its spread. Although the lockdown is effective in such prevention, it may put the economy into a depression. Predicting the epidemic progression with the government switching the lockdown on or off is critical. We propose a transfer learning approach called ALeRT-COVID using attention-based recurrent neural network (RNN) architecture to predict the epidemic trends for different countries. A source model was trained on the pre-defined source countries and then transferred to each target country. The lockdown measure was introduced to our model as a predictor and the attention mechanism was utilized to learn the different contributions of the confirmed cases in the past days to the future trend. Results demonstrated that the transfer learning strategy is helpful especially for early-stage countries. By introducing the lockdown predictor and the attention mechanism, ALeRT-COVID showed a significant improvement in the prediction performance. We predicted the confirmed cases in 1 week when extending and easing lockdown separately. Our results show that lockdown measures are still necessary for several countries. We expect our research can help different countries to make better decisions on the lockdown measures.
Tài liệu tham khảo
COVID-19 Coronavirus Tracker. Available at https://www.kff.org/coronavirus-covid-19/fact-sheet/coronavirus-tracker/. Accessed 1 June 2020
Lechien JR, Chiesa-Estomba CM, De Siati DR, Horoi M, Le Bon SD, Rodriguez A et al (2020) Olfactory and gustatory dysfunctions as a clinical presentation of mil-to-moderate forms of the coronavirus disease (COVID-19): a multicenter European study. Eur Arch Otorhinolaryngol 277(8):2251–2261
Bedford J, Enria D, Giesecke J, Heymann DL, Ihekweazu C, Kobinger G et al (2020) COVID-19: towards controlling of a pandemic. Lancet 395(10229):1015–1018
Pan A, Liu L, Wang C, Guo H, Hao X, Wang Q et al (2020) Association of public health interventions with the epidemiology of the COVID-19 outbreak in Wuhan, China. JAMA 323(19):1915–1923
Peak CM, Childs LM, Grad YH, Buckee CO (2017) Comparing nonpharmaceutical interventions for containing emerging epidemics. Proc Natl Acad Sci 114(15):4023–4028
Prem K, Liu Y, Russell TW, Kucharski AJ, Eggo RM, Davies N et al (2020) The effect of control strategies to reduce social mixing on outcomes of the COVID-19 epidemic in Wuhan, China: a modelling study. Lancet Public Health 5(5):e261–e270
Habibi R, Burci GL, de Campos TC, Chirwa D, Cinà M, Dagron S et al (2020) Do not violate the International Health Regulations during the COVID-19 outbreak. Lancet 395(10225):664–666
Hossain M, Junus A, Zhu X, Jia P, Wen TH, Pfeiffer D, Yuan HY (2020) The effects of border control and quarantine measures on global spread of COVID-19. Epidemics. https://doi.org/10.1016/j.epidem.2020.100397
Kraemer MU, Yang CH, Gutierrez B, Wu CH, Klein B, Pigott DM et al (2020) The effect of human mobility and control measures on the COVID-19 epidemic in China. Science 368(6490):493–497
Chinazzi M, Davis JT, Ajelli M, Gioannini C, Litvinova M, Merler S et al (2020) The effect of travel restrictions on the spread of the 2019 novel coronavirus (COVID-19) outbreak. Science 368(6489):395–400
Bertuzzo E, Mari L, Pasetto D, Miccoli S, Casagrandi R, Gatto M, Rinaldo A (2020) The geography of COVID-19 spread in Italy and implications for the relaxation of confinement measures. medRxiv
Dev SM, Sengupta R (2020) Covid-19: impact on the Indian economy. Indira Gandhi Institute of Development Research, Mumbai
Singh S, Parmar KS, Kumar J, Makkhan SJS (2020) Development of new hybrid model of discrete wavelet decomposition and autoregressive integrated moving average (ARIMA) models in application to one month forecast the casualties cases of COVID-19. Chaos, Solitons & Fractals 135. https://doi.org/10.1016/j.chaos.2020.109866
Zhao Z, Li X, Liu F, Zhu G, Ma C, Wang L (2020) Prediction of the COVID-19 spread in African countries and implications for prevention and controls: a case study in South Africa, Egypt, Algeria, Nigeria, Senegal and Kenya. Sci Total Environ 138959
Zheng N, Du S, Wang J, Zhang H, Cui W, Kang et al (2020) Predicting covid-19 in china using hybrid AI model. IEEE Trans Cybern 50(7):2891–2904
Singh RK, Rani M, Bhagavathula AS, Sah R, Rodriguez-Morales AJ, Kalita H et al (2020) Prediction of the COVID-19 pandemic for the top 15 affected countries: advanced autoregressive integrated moving average (ARIMA) model. JMIR Public Health Surveill 6(2):e19115
Verma V, Vishwakarma RK, Verma A, Nath DC, Khan HT (2020) Time-to-death approach in revealing chronicity and severity of COVID-19 across the world. PLoS One 15(5):e0233074
Chen FH (2006) A susceptible-infected epidemic model with voluntary vaccinations. J Math Biol 53(2):253–272
Kermack WO, McKendrick AG (1927) A contribution to the mathematical theory of epidemics. Proc R Soc Lond Ser A, Containing papers of a Mathematical and Physical Character 115(772):700–721
Li MY, Graef JR, Wang L, Karsai J (1999) Global dynamics of a SEIR model with varying total population size. Math Biosci 160(2):191–213
Yang Z, Zeng Z, Wang K, Wong SS, Liang W, Zanin M et al (2020) Modified SEIR and AI prediction of the epidemics trend of COVID-19 in China under public health interventions. J Thorac Dis 12(3):165
López L, Rodo X (2020) A modified SEIR model to predict the COVID-19 outbreak in Spain and Italy: simulating control scenarios and multi-scale epidemics. Results Phys. https://doi.org/10.1016/j.rinp.2020.103746
Zhan C, Chi KT, Fu Y, Lai Z, Zhang H (2020) Modeling and prediction of the 2019 coronavirus disease spreading in China incorporating human migration data. medRxiv
Jia W, Wan Y, Li Y, Tan K, Lei W, Hu Y et al (2019) Integrating multiple data sources and learning models to predict infectious diseases in China. AMIA Summits Transl Sci Proc 680:2019
Goodfellow I, Bengio Y, Courville A, Bengio Y (2016) Deep Learning (Vol. 1). MIT press, Cambridge
Chen XW, Lin X (2014) Big data deep learning: challenges and perspectives. IEEE Access 2:514–525
Coronavirus Update (Live): 6,573,585 cases and 388,041 deaths from COVID-19 virus pandemic - Worldometer. Available at https://www.worldometers.info/coronavirus/ (2020). Accessed 14 May 2020
Coronavirus (COVID-19) Lockdown Tracker | Aura Vision. Available at https://auravision.ai/covid19-lockdown-tracker/ (2020). Accessed 14 May 2020
Wikipedia. List of countries and dependencies by population. Available at https://en.wikipedia.org/w/index.php?title=List_of_countries_and_dependencies_by_population&oldid=960653268 (2020). Accessed 14 May 2020
Hochreiter S, & Schmidhuber J (1997) Long short-term memory. Neural Comput 9(8): 1735–1780
Bahdanau D, Cho K, Bengio Y (2014) Neural machine translation by jointly learning to align and translate. arXiv:1409.0473
Gulli A, Pal S (2017) Deep learning with Keras. Packt Publishing Ltd
Kingma DP, Ba J (2014) Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980
Ruder S (2016) An overview of gradient descent optimization algorithms. arXiv preprint arXiv:1609.04747
Wang G, Huang NE, & Qiao F (2020) Quantitative evaluation on control measures for an epidemic: a case study of COVID-19. Kexue Tongbao/Chinese Science Bulletin 65(11)
Pan SJ, Yang Q (2009) A survey on transfer learning. IEEE Trans Knowl Data Eng 22(10):1345–1359
Aldhyani TH, Alrasheed M, Alzahrani MY, Ahmed H (2020) Deep learning and Holt-trend algorithms for predicting COVID-19 pandemic. medRxiv
Huang NE, Qiao F (2020) A data driven time-dependent transmission rate for tracking an epidemic: a case study of 2019-nCoV. Sci Bull 65(6):425–427
Konečný J, Liu J, Richtárik P, & Takáč M (2015) Mini-batch semi-stochastic gradient descent in the proximal setting. IEEE Journal of Selected Topics in Signal Processing, 10(2):242–255