Forecasting COVID-19 confirmed cases in South Korea using Spatio-Temporal Graph Neural Networks
1
조회
0
다운로드
0
인용
0
참고문헌
키워드
초록
Since the outbreak of the coronavirus disease 2019 (COVID-19) pandemic, a lot of efforts have been made in the field of data science to help combat against this disease. Among them, forecasting the number of cases of infection is a crucial problem to predict the development of the pandemic. Many deep learning-based models can be applied to solve this type of time series problem. In this research, we would like to take a step forward to incorporate spatial data (geography) with time series data to forecast the cases of region-level infection simultaneously. Specifically, we model a single spatio-temporal graph, in which nodes represent the geographic regions, spatial edges represent the distance between each pair of regions, and temporal edges indicate the node features through time. We evaluate this approach in COVID-19 in a Korean dataset, and we show a decrease of approximately 10% in both RMSE and MAE, and a significant boost to the training speed compared to the baseline models. Moreover, the training efficiency allows this approach to be extended for a large-scale spatio-temporal dataset.
Abstract
Since the outbreak of the coronavirus disease 2019 (COVID-19) pandemic, a lot of efforts have been made in the field of data science to help combat against this disease. Among them, forecasting the number of cases of infection is a crucial problem to predict the development of the pandemic. Many deep learning-based models can be applied to solve this type of time series problem. In this research, we would like to take a step forward to incorporate spatial data (geography) with time series data to forecast the cases of region-level infection simultaneously. Specifically, we model a single spatio-temporal graph, in which nodes represent the geographic regions, spatial edges represent the distance between each pair of regions, and temporal edges indicate the node features through time. We evaluate this approach in COVID-19 in a Korean dataset, and we show a decrease of approximately 10% in both RMSE and MAE, and a significant boost to the training speed compared to the baseline models. Moreover, the training efficiency allows this approach to be extended for a large-scale spatio-temporal dataset.
참고문헌 (34)
- Ensheng Dong, "An interactive web-based dashboard to track COVID-19 in real time", The Lancet Infectious Diseases, 20(5), pp.533~534 (2020), Elsevier BV[DOI]
- A. Kapoor, "Examining covid-19 forecasting using spatio-temporal graph neural networks", arXiv preprint arXiv:2007.03113, ~ (2020)
- I. COVID, "Forecasting COVID-19 impact on hospital bed-days, ICU-days, ventilator-days and deaths by US state in the next 4 months", MedRxiv (2020)[DOI]
- J. Durbin, "Time series analysis by state space methods" (2012), Oxford university press
- Zifeng Yang, "Modified SEIR and AI prediction of the epidemics trend of COVID-19 in China under public health interventions", Journal of Thoracic Disease, 12(3), pp.165~174 (2020), AME Publishing Company[DOI]
- Qingchun Guo, "Prediction of the confirmed cases and deaths of global COVID-19 using artificial intelligence", Environmental Science and Pollution Research, 28(9), pp.11672~11682 (2021), Springer Science and Business Media LLC[DOI]
- L. Wang, "Examining Deep Learning Models with Multiple Data Sources for COVID-19 Forecasting", arXiv preprint arXiv:2010.14491, ~ (2020)
- B. Yu, "Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting", Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence (2018)
- Y. Li, "Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting", International Conference on Learning Representations (2018)
- Alok Kumar Sahai, "ARIMA modelling & forecasting of COVID-19 in top five affected countries", Diabetes & Metabolic Syndrome: Clinical Research & Reviews, 14(5), pp.1419~1427 (2020), Elsevier BV[DOI]
- Andres Hernandez-Matamoros, "Forecasting of COVID19 per regions using ARIMA models and polynomial functions", Applied Soft Computing, 96, pp.106610~ (2020), Elsevier BV[DOI]
- David E. Rumelhart, "Learning representations by back-propagating errors", Nature, 323(6088), pp.533~536 (1986), Springer Science and Business Media LLC[DOI]
- A. Graves, "A Novel Connectionist System for Unconstrained Handwriting Recognition", IEEE Transactions on Pattern Analysis and Machine Intelligence, 31(5), pp.855~868 (2009), Institute of Electrical and Electronics Engineers (IEEE)[DOI]
- H. Sak, "Long short-term memory recurrent neural network architectures for large scale acoustic modeling", ~ (2014)
- X. Li, "Constructing long short-term memory based deep recurrent neural networks for large vocabulary speech recognition", 2015 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) (2015)
- Sepp Hochreiter, "Long Short-Term Memory", Neural Computation, 9(8), pp.1735~1780 (1997), MIT Press - Journals[DOI]
- A. Graves, "Supervised sequence labelling with recurrent neural networks", pp.37~45 (2012), Springer
- Vinay Kumar Reddy Chimmula, "Time series forecasting of COVID-19 transmission in Canada using LSTM networks", Chaos, Solitons & Fractals, 135, pp.109864~ (2020), Elsevier BV[DOI]
- Farah Shahid, "Predictions for COVID-19 with deep learning models of LSTM, GRU and Bi-LSTM", Chaos, Solitons & Fractals, 140, pp.110212~ (2020), Elsevier BV[DOI]
- I. Sutskever, "Sequence to sequence learning with neural networks", arXiv preprint arXiv:1409.3215, ~ (2014)
- D. Bahdanau, "Neural machine translation by jointly learning to align and translate", arXiv preprint arXiv:1409.0473, ~ (2014)
- M.-T. Luong, "Effective approaches to attention-based neural machine translation", arXiv preprint arXiv:1508.04025, ~ (2015)
- Jie Zhou, "Graph neural networks: A review of methods and applications", AI Open, 1, pp.57~81 (2020), Elsevier BV[DOI]
- J. Bruna, "Spectral networks and locally connected networks on graphs", arXiv preprint arXiv:1312.6203, ~ (2013)
- M. Defferrard, "Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering", Proceedings of the 30th International Conference on Neural Information Processing Systems (2016)
- T. N. Kipf, "Semi-Supervised Classification with Graph Convolutional Networks", arXiv preprint arXiv:1609.02907, ~ (2016)
- D. I. Shuman, "The emerging field of signal processing on graphs: Extending high-dimensional data analysis to networks and other irregular domains", IEEE Signal Processing Magazine, 30(3), pp.83~98 (2013), Institute of Electrical and Electronics Engineers (IEEE)[DOI]
- David K. Hammond, "Wavelets on graphs via spectral graph theory", Applied and Computational Harmonic Analysis, 30(2), pp.129~150 (2011), Elsevier BV[DOI]
- J. Chung, "Empirical evaluation of gated recurrent neural networks on sequence modeling", arXiv preprint arXiv:1412.3555, ~ (2014)
- D. P. Kingma, "Adam: A method for stochastic optimization", arXiv preprint arXiv:1412.6980, ~ (2014)