skip to main content
10.1145/3631461.3631959acmotherconferencesArticle/Chapter ViewAbstractPublication PagesicdcnConference Proceedingsconference-collections
research-article

ATT-LSTM: An Interpretable Deep Learning Framework for COVID Outbreak Prediction

Published:22 January 2024Publication History

ABSTRACT

The outbreak of the new coronavirus (COVID-19), which causes respiratory infections in humans, has been a significant concern worldwide since December 2019. As of September 12, 2020, there had been 1 million reported deaths worldwide and 29 million certified cases. The urgent need to observe and predict the distribution of COVID-19 has been found to manage this spread. Among the various popular models of the COVID-19 forecast, statistical models have received a lot of media attention. However, it has been found that these statistical models have shown little accuracy with long-term forecasts due to insufficient data available and high levels of uncertainty. In this paper, we propose a comparison-based analysis of various deep learning models to predict the prevalence of COVID-19 as a more effective alternative to statistical methods. We proposed a new attention-based encoder-decoder model called Attention-Long Short-Term Memory (ATT-LSTM). The LSTM-based neural network layer architecture uses the idea of a fine-grained attention mechanism, which means paying attention to the hidden state rather than the hidden state vectors themselves. This can draw attention to the importance and contributions of each hidden state dimension. It is useful in locating and concentrating relevant temporary data, leading to a distinctly interpretable network. A learnable vector embedded for time has been implemented because time in a vector representation can be easily added with one or more architectures. This vector representation is known as Time2Vec. The proposed deep learning model predicts the trend and patterns of COVID-19 outbreaks in some of the worst-affected European countries, such as Italy, Spain, France, and Canada. We have used the COVID-19 database by Johns Hopkins University’s Center for Systems Science and Engineering (CSSE) to test and analyze the proposed model. The proposed model has higher predictive accuracy compared to other existing models.

References

  1. Nikita Agarwal, Shiva Reddy Koti, Sameer Saran, and A Senthil Kumar. 2018. Data mining techniques for predicting dengue outbreak in geospatial domain using weather parameters for New Delhi, India. Curr. Sci 114, 11 (2018), 2281–2291.Google ScholarGoogle ScholarCross RefCross Ref
  2. Sumiko Anno, Takeshi Hara, Hiroki Kai, Ming-An Lee, Yi Chang, Kei Oyoshi, Yousei Mizukami, and Takeo Tadono. 2019. Spatiotemporal dengue fever hotspots associated with climatic factors in taiwan including outbreak predictions based on machine-learning. Geospatial health 14, 2 (2019).Google ScholarGoogle Scholar
  3. Abdon Atangana. 2020. Modelling the spread of COVID-19 with new fractal-fractional operators: Can the lockdown save mankind before vaccination?Chaos, Solitons & Fractals 136 (2020), 109860.Google ScholarGoogle ScholarCross RefCross Ref
  4. Rachel M Burke, Minesh P Shah, Mary E Wikswo, Leslie Barclay, Anita Kambhampati, Zachary Marsh, Jennifer L Cannon, Umesh D Parashar, Jan Vinjé, and Aron J Hall. 2019. The norovirus epidemiologic triad: predictors of severe outcomes in US norovirus outbreaks, 2009–2016. The Journal of infectious diseases 219, 9 (2019), 1364–1372.Google ScholarGoogle ScholarCross RefCross Ref
  5. Long-ting Cao, Hong-hui Liu, Juan Li, Xiao-dong Yin, Yu Duan, and Jing Wang. 2020. Relationship of meteorological factors and human brucellosis in Hebei province, China. Science of The Total Environment 703 (2020), 135491.Google ScholarGoogle ScholarCross RefCross Ref
  6. Colin J Carlson, Eric Dougherty, Mike Boots, Wayne Getz, and Sadie J Ryan. 2018. Consensus and conflict among ecological forecasts of Zika virus outbreaks in the United States. Scientific reports 8, 1 (2018), 1–15.Google ScholarGoogle Scholar
  7. Zeynep Ceylan. 2020. Estimation of COVID-19 prevalence in Italy, Spain, and France. Science of The Total Environment (2020), 138817.Google ScholarGoogle Scholar
  8. Tanujit Chakraborty and Indrajit Ghosh. 2020. Real-time forecasts and risk assessment of novel coronavirus (COVID-19) cases: A data-driven analysis. Chaos, Solitons & Fractals (2020), 109850.Google ScholarGoogle ScholarCross RefCross Ref
  9. Jasper Fuk-Woo Chan, Shuofeng Yuan, Kin-Hang Kok, Kelvin Kai-Wang To, Hin Chu, Jin Yang, Fanfan Xing, Jieling Liu, Cyril Chik-Yan Yip, Rosana Wing-Shan Poon, 2020. A familial cluster of pneumonia associated with the 2019 novel coronavirus indicating person-to-person transmission: a study of a family cluster. The Lancet 395, 10223 (2020), 514–523.Google ScholarGoogle ScholarCross RefCross Ref
  10. Ye Chen, Kunkun Leng, Ying Lu, Lihai Wen, Ying Qi, Wei Gao, Huijie Chen, Lina Bai, Xiangdong An, Baijun Sun, 2020. Epidemiological features and time-series analysis of influenza incidence in urban and rural areas of Shenyang, China, 2010–2018. Epidemiology & Infection 148 (2020).Google ScholarGoogle Scholar
  11. Shima Shamkhali Chenar and Zhiqiang Deng. 2018. Development of artificial intelligence approach to forecasting oyster norovirus outbreaks along Gulf of Mexico coast. Environment international 111 (2018), 212–223.Google ScholarGoogle ScholarCross RefCross Ref
  12. Shima Shamkhali Chenar and Zhiqiang Deng. 2018. Development of genetic programming-based model for predicting oyster norovirus outbreak risks. Water research 128 (2018), 20–37.Google ScholarGoogle Scholar
  13. Vinay Kumar Reddy Chimmula and Lei Zhang. 2020. Time series forecasting of COVID-19 transmission in Canada using LSTM networks. Chaos, Solitons & Fractals (2020), 109864.Google ScholarGoogle Scholar
  14. Kyunghyun Cho, Bart Van Merriënboer, Dzmitry Bahdanau, and Yoshua Bengio. 2014. On the properties of neural machine translation: Encoder-decoder approaches. arXiv preprint arXiv:1409.1259 (2014).Google ScholarGoogle Scholar
  15. Heeyoul Choi, Kyunghyun Cho, and Yoshua Bengio. 2018. Fine-grained attention mechanism for neural machine translation. Neurocomputing 284 (2018), 171–176.Google ScholarGoogle ScholarCross RefCross Ref
  16. Salih Djilali and Behzad Ghanbari. 2020. Coronavirus pandemic: A predictive analysis of the peak outbreak epidemic in South Africa, Turkey, and Brazil. Chaos, Solitons & Fractals (2020), 109971.Google ScholarGoogle ScholarCross RefCross Ref
  17. Duccio Fanelli and Francesco Piazza. 2020. Analysis and forecast of COVID-19 spreading in China, Italy and France. Chaos, Solitons & Fractals 134 (2020), 109761.Google ScholarGoogle ScholarCross RefCross Ref
  18. Xinyu Fang, Wendong Liu, Jing Ai, Mike He, Ying Wu, Yingying Shi, Wenqi Shen, and Changjun Bao. 2020. Forecasting incidence of infectious diarrhea using random forest in Jiangsu Province, China. BMC Infectious Diseases 20, 1 (2020), 1–8.Google ScholarGoogle ScholarCross RefCross Ref
  19. Lisa E Gralinski and Vineet D Menachery. 2020. Return of the Coronavirus: 2019-nCoV. Viruses 12, 2 (2020), 135.Google ScholarGoogle ScholarCross RefCross Ref
  20. Sepp Hochreiter and Jürgen Schmidhuber. 1997. Long short-term memory. Neural computation 9, 8 (1997), 1735–1780.Google ScholarGoogle ScholarDigital LibraryDigital Library
  21. Chaolin Huang, Yeming Wang, Xingwang Li, Lili Ren, Jianping Zhao, Yi Hu, Li Zhang, Guohui Fan, Jiuyang Xu, Xiaoying Gu, 2020. Clinical features of patients infected with 2019 novel coronavirus in Wuhan, China. The lancet 395, 10223 (2020), 497–506.Google ScholarGoogle ScholarCross RefCross Ref
  22. Fumito Koike and Nobuo Morimoto. 2018. Supervised forecasting of the range expansion of novel non-indigenous organisms: Alien pest organisms and the 2009 H1N1 flu pandemic. Global Ecology and Biogeography 27, 8 (2018), 991–1000.Google ScholarGoogle ScholarCross RefCross Ref
  23. Vladimir Kurbalija, Miloš Radovanović, Mirjana Ivanović, Danilo Schmidt, Gabriela Lindemann von Trzebiatowski, Hans-Dieter Burkhard, and Carl Hinrichs. 2014. Time-series analysis in the medical domain: A study of Tacrolimus administration and influence on kidney graft function. Computers in biology and medicine 50 (2014), 19–31.Google ScholarGoogle Scholar
  24. Shengjie Lai, Isaac I Bogoch, Nick W Ruktanonchai, Alexander Watts, Xin Lu, Weizhong Yang, Hongjie Yu, Kamran Khan, and Andrew J Tatem. 2020. Assessing spread risk of Wuhan novel coronavirus within and beyond China, January-April 2020: a travel network-based modelling study. medRxiv (2020).Google ScholarGoogle Scholar
  25. Ruirui Liang, Yi Lu, Xiaosheng Qu, Qiang Su, Chunxia Li, Sijing Xia, Yongxin Liu, Qiang Zhang, Xin Cao, Qin Chen, 2020. Prediction for global African swine fever outbreaks based on a combination of random forest algorithms and meteorological data. Transboundary and Emerging Diseases 67, 2 (2020), 935–946.Google ScholarGoogle ScholarCross RefCross Ref
  26. Qiao Liu, Zhongqi Li, Ye Ji, Leonardo Martinez, Ui Haq Zia, Arshad Javaid, Wei Lu, and Jianming Wang. 2019. Forecasting the seasonality and trend of pulmonary tuberculosis in Jiangsu Province of China using advanced statistical time-series analyses. Infection and Drug Resistance 12 (2019), 2311.Google ScholarGoogle ScholarCross RefCross Ref
  27. Neeraj, Jimson Mathew, and Ranjan Kumar Behera. 2022. EMD-Att-LSTM: A Data-driven Strategy Combined with Deep Learning for Short-term Load Forecasting. Journal of Modern Power Systems and Clean Energy 10, 5 (2022), 1229–1240. https://doi.org/10.35833/MPCE.2020.000626Google ScholarGoogle ScholarCross RefCross Ref
  28. Neeraj, Jimson Mathew, and Ranjan Kumar Behera. 2022. Power load forecasting based on long short term memory-singular spectrum analysis. Energy Systems (2022), 1–23.Google ScholarGoogle Scholar
  29. Neeraj Neeraj, Jimson Mathew, Mayank Agarwal, and Ranjan Kumar Behera. 2021. Long short-term memory-singular spectrum analysis-based model for electric load forecasting. Electrical Engineering 103 (2021), 1067–1082.Google ScholarGoogle ScholarCross RefCross Ref
  30. Adam Paszke, Sam Gross, Francisco Massa, Adam Lerer, James Bradbury, Gregory Chanan, Trevor Killeen, Zeming Lin, Natalia Gimelshein, Luca Antiga, 2019. Pytorch: An imperative style, high-performance deep learning library. In Advances in neural information processing systems. 8026–8037.Google ScholarGoogle Scholar
  31. Sittisede Polwiang. 2020. The time series seasonal patterns of dengue fever and associated weather variables in Bangkok (2003-2017). BMC Infectious Diseases 20, 1 (2020), 1–10.Google ScholarGoogle ScholarCross RefCross Ref
  32. Dhesi Baha Raja, Rainier Mallol, Choo Yee Ting, Fadzilah Kamaludin, Rohani Ahmad, Suzilah Ismail, Vivek Jason Jayaraj, and Bala Murali Sundram. 2019. Artificial intelligence model as predictor for dengue outbreaks. Malaysian Journal of Public Health Medicine 19, 2 (2019), 103–108.Google ScholarGoogle ScholarCross RefCross Ref
  33. Catrin Sohrabi, Zaid Alsafi, Niamh O’Neill, Mehdi Khan, Ahmed Kerwan, Ahmed Al-Jabir, Christos Iosifidis, and Riaz Agha. 2020. World Health Organization declares global emergency: A review of the 2019 novel coronavirus (COVID-19). International Journal of Surgery (2020).Google ScholarGoogle ScholarCross RefCross Ref
  34. Leili Tapak, Omid Hamidi, Mohsen Fathian, and Manoochehr Karami. 2019. Comparative evaluation of time series models for predicting influenza outbreaks: Application of influenza-like illness data from sentinel sites of healthcare centers in Iran. BMC research notes 12, 1 (2019), 353.Google ScholarGoogle Scholar
  35. Madeleine C Thomson, Anna M Molesworth, Mamoudou H Djingarey, KR Yameogo, Francois Belanger, and Luis E Cuevas. 2006. Potential of environmental models to predict meningitis epidemics in Africa. Tropical Medicine & International Health 11, 6 (2006), 781–788.Google ScholarGoogle ScholarCross RefCross Ref
  36. O Torrealba-Rodriguez, RA Conde-Gutiérrez, and AL Hernández-Javier. 2020. Modeling and prediction of COVID-19 in Mexico applying mathematical and computational models. Chaos, Solitons & Fractals (2020), 109946.Google ScholarGoogle Scholar
  37. Rose Yu, Stephan Zheng, Anima Anandkumar, and Yisong Yue. 2017. Long-term forecasting using tensor-train RNNs. arXiv preprint arXiv:1711.00073 (2017).Google ScholarGoogle Scholar
  38. Xingyu Zhang, Yuanyuan Liu, Min Yang, Tao Zhang, Alistair A Young, and Xiaosong Li. 2013. Comparative study of four time series methods in forecasting typhoid fever incidence in China. PloS one 8, 5 (2013), e63116.Google ScholarGoogle ScholarCross RefCross Ref

Index Terms

  1. ATT-LSTM: An Interpretable Deep Learning Framework for COVID Outbreak Prediction

    Recommendations

    Comments

    Login options

    Check if you have access through your login credentials or your institution to get full access on this article.

    Sign in
    • Published in

      cover image ACM Other conferences
      ICDCN '24: Proceedings of the 25th International Conference on Distributed Computing and Networking
      January 2024
      423 pages
      ISBN:9798400716737
      DOI:10.1145/3631461

      Copyright © 2024 ACM

      Publication rights licensed to ACM. ACM acknowledges that this contribution was authored or co-authored by an employee, contractor or affiliate of a national government. As such, the Government retains a nonexclusive, royalty-free right to publish or reproduce this article, or to allow others to do so, for Government purposes only.

      Publisher

      Association for Computing Machinery

      New York, NY, United States

      Publication History

      • Published: 22 January 2024

      Permissions

      Request permissions about this article.

      Request Permissions

      Check for updates

      Qualifiers

      • research-article
      • Research
      • Refereed limited
    • Article Metrics

      • Downloads (Last 12 months)16
      • Downloads (Last 6 weeks)1

      Other Metrics

    PDF Format

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    HTML Format

    View this article in HTML Format .

    View HTML Format