Streaming Factor Trajectory Learning for Temporal Tensor Decomposition

Shikai Fang, Xin Yu, Shibo Li, Zheng Wang, R. Kirby +1 more
10/1/2023

No official code found. Best implementation: xuangu-fang/Streaming-Factor-Trajectory-Learning (7 stars, MIT, last commit 2023-10-27).

Abstract

Practical tensor data is often along with time information. Most existing temporal decomposition approaches estimate a set of fixed factors for the objects in each tensor mode, and hence cannot capture the temporal evolution of the objects' representation. More important, we lack an effective approach to capture such evolution from streaming data, which is common in real-world applications. To address these issues, we propose Streaming Factor Trajectory Learning (SFTL) for temporal tensor decomposition. We use Gaussian processes (GPs) to model the trajectory of factors so as to flexibly estimate their temporal evolution. To address the computational challenges in handling streaming data, we convert the GPs into a state-space prior by constructing an equivalent stochastic differential equation (SDE). We develop an efficient online filtering algorithm to estimate a decoupled running posterior of the involved factor states upon receiving new data. The decoupled estimation enables us to conduct standard Rauch-Tung-Striebel smoothing to compute the full posterior of all the trajectories in parallel, without the need for revisiting any previous data. We have shown the advantage of SFTL in both synthetic tasks and real-world applications.

View on arXivView PDF

Code Implementations(1)

MIT

Streaming Factor Trajectory Learning for Temporal Tensor Decomposition(NeurIPS 2023)

70Shell, PythonOct 4, 20232 years agoMIT

Similar Papers

Cite this paper

@article{fang2023streaming,
  title  = {Streaming Factor Trajectory Learning for Temporal Tensor Decomposition},
  author = {Shikai Fang and Xin Yu and Shibo Li and Zheng Wang and R. Kirby and Shandian Zhe},
  year   = {2023},
  eprint = {2310.17021},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2310.17021},
  journal = {NEURIPS 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2310.17021.svg)](https://codeofpaper.com/paper/2310.17021)

Discussion