Author: Kargas, Nikos; Qian, Cheng; Sidiropoulos, Nicholas D.; Xiao, Cao; Glass, Lucas M.; Sun, Jimeng
Title: STELAR: Spatio-temporal Tensor Factorization with Latent Epidemiological Regularization Cord-id: v6hgjkc1 Document date: 2020_12_8
ID: v6hgjkc1
Snippet: Accurate prediction of the transmission of epidemic diseases such as COVID-19 is crucial for implementing effective mitigation measures. In this work, we develop a tensor method to predict the evolution of epidemic trends for many regions simultaneously. We construct a 3-way spatio-temporal tensor (location, attribute, time) of case counts and propose a nonnegative tensor factorization with latent epidemiological model regularization named STELAR. Unlike standard tensor factorization methods whi
Document: Accurate prediction of the transmission of epidemic diseases such as COVID-19 is crucial for implementing effective mitigation measures. In this work, we develop a tensor method to predict the evolution of epidemic trends for many regions simultaneously. We construct a 3-way spatio-temporal tensor (location, attribute, time) of case counts and propose a nonnegative tensor factorization with latent epidemiological model regularization named STELAR. Unlike standard tensor factorization methods which cannot predict slabs ahead, STELAR enables long-term prediction by incorporating latent temporal regularization through a system of discrete-time difference equations of a widely adopted epidemiological model. We use latent instead of location/attribute-level epidemiological dynamics to capture common epidemic profile sub-types and improve collaborative learning and prediction. We conduct experiments using both county- and state-level COVID-19 data and show that our model can identify interesting latent patterns of the epidemic. Finally, we evaluate the predictive ability of our method and show superior performance compared to the baselines, achieving up to 21% lower root mean square error and 25% lower mean absolute error for county-level prediction.
Search related documents:
Co phrase search for related documents- long term epidemic prediction and machine epidemiological learning: 1
- long term prediction and lstm short term memory: 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12
- long term prediction and machine epidemiological: 1
- long term prediction and machine epidemiological learning: 1
- long term prediction and machine learning: 1, 2, 3, 4, 5, 6, 7, 8, 9, 10
- long term prediction and machine learning approach: 1
- long term prediction and machine learning model: 1, 2
- lstm apply and machine learning: 1
- lstm short term memory and machine learning: 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46
- lstm short term memory and machine learning approach: 1, 2, 3, 4, 5
- lstm short term memory and machine learning model: 1, 2, 3, 4, 5
- lstm short term memory and mae mean absolute error: 1, 2, 3, 4, 5, 6, 7, 8
- lstm short term memory and mae mean absolute error rmse: 1, 2, 3, 4, 5, 6, 7
- machine learning and mae mean absolute error: 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24
- machine learning and mae mean absolute error rmse: 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14
- machine learning approach and mae mean absolute error: 1, 2
- machine learning approach and mae mean absolute error rmse: 1
- machine learning model and mae mean absolute error: 1, 2, 3, 4
- machine learning model and mae mean absolute error rmse: 1, 2, 3
Co phrase search for related documents, hyperlinks ordered by date