STELAR: Spatio-temporal Tensor Factorization with Latent Epidemiological Regularization

  title={STELAR: Spatio-temporal Tensor Factorization with Latent Epidemiological Regularization},
  author={Nikos Kargas and Cheng Qian and Nicholas D. Sidiropoulos and Cao Xiao and Lucas Glass and Jimeng Sun},
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… 

Figures and Tables from this paper

Data-Centric Epidemic Forecasting: A Survey

This survey delves into various data-driven methodological and practical advancements and introduces a conceptual framework to navigate through them, enumerate the large number of epidemiological datasets and novel data streams that are relevant to epidemic forecasting, and highlights some challenges and open problems found across the forecasting pipeline.

Explainability of the COVID-19 epidemiological model with nonnegative tensor factorization

A novel method to provide the explainability of an epidemiological model is proposed that represents the output of SEM as a tensor model and applies nonnegative tensor factorization (NTF) to identify patterns of global response behaviours of countries and cluster the countries based on these patterns.

Multiwave COVID-19 Prediction from Social Awareness Using Web Search and Mobility Data

A Social Awareness-Based Graph Neural Network (SAB-GNN) that considers the decay of symptom-related web search frequency to capture the changes in public awareness across multiple waves of COVID-19 pandemic prediction.

Accurate Online Tensor Factorization for Temporal Tensor Streams with Missing Values

STF (Streaming Tensor Factorization), an accurate online tensor factorization method for real-world temporal tensor streams with missing values, is proposed and an efficient online learning algorithm which allows each row of the temporal factor matrix to be updated from past and future information is proposed.

MTC: Multiresolution Tensor Completion from Partial and Coarse Observations

The proposed Multi-resolution Tensor Completion model (MTC) explores tensor mode properties and leverages the hierarchy of resolutions to recursively initialize an optimization setup, and optimizes on the coupled system using alternating least squares to ensure low computational and space complexity.

Multi-version Tensor Completion for Time-delayed Spatio-temporal Data

This work proposes a low-rank tensor model that can accurately predict the ground-truth values of many real-world tensors and extends the method to track the tensor data over time, leading to significant computational savings.



Examining COVID-19 Forecasting using Spatio-Temporal Graph Neural Networks

This work examines a novel forecasting approach for COVID-19 case prediction that uses Graph Neural Networks and mobility data, and demonstrates that the rich spatial and temporal information leveraged by the graph neural network allows the model to learn complex dynamics.

STAN: Spatio-Temporal Attention Network for Pandemic Prediction Using Real World Evidence

A spatio-temporal attention network (STAN) uses an attention-based graph convolutional network to capture geographical and temporal trends and predict the number of cases for a fixed number of days into the future, achieving up to 87% lower mean squared error compared to the best baseline prediction model.

DEFSI: Deep Learning Based Epidemic Forecasting with Synthetic Information

This work proposes DEFSI (Deep Learning Based Epidemic Forecasting with Synthetic Information), an epidemic forecasting framework that integrates the strengths of artificial neural networks and causal methods that significantly outperforms the other methods for short-term ILI forecasting at the state level.

Link Prediction on Evolving Data Using Matrix and Tensor Factorizations

This paper looks at bipartite graphs changing over time and considers matrix- and tensor-based methods for predicting links, and presents a weight-based method for collapsing multi-year data into a single matrix.

Temporal Link Prediction Using Matrix and Tensor Factorizations

This article considers bipartite graphs that evolve over time and considers matrix- and tensor-based methods for predicting future links and shows that Tensor- based techniques are particularly effective for temporal data with varying periodic patterns.

Time series forecasting of COVID-19 transmission in Canada using LSTM networks☆

TensorCast : Forecasting with Context using Coupled Tensors

TENSORCAST is a novel method that forecasts time-evolving networks more accurately than current state of the art methods by incorporating multiple data sources in coupled tensors.

Tensors for Data Mining and Data Fusion

This survey presents some of the most widely used tensor decompositions, providing the key insights behind them, and summarizing them from a practitioner’s point of view.

Modified SEIR and AI prediction of the epidemics trend of COVID-19 in China under public health interventions

It is found that the epidemic of China should peak by late February, showing gradual decline by end of April, and the implementation of control measures on January 23 2020 was indispensable in reducing the eventual COVID-19 epidemic size.

Prediction for the spread of COVID-19 in India and effectiveness of preventive measures