Long-term Forecasting using Tensor-Train RNNs

By Rose Yu, Stephan Zheng, Anima Anandkumar, Yisong Yue. 2017

146 citations recorded by the source corpus.

We present Tensor-Train RNN (TT-RNN), a novel family of neural sequence architectures for multivariate forecasting in environments with nonlinear dynamics. Long-term forecasting in such systems is highly challenging, since there exist long-term temporal dependencies, higher-order correlations and sensitivity to error propagation. Our proposed tensor recurrent architecture addresses these issues by learning the nonlinear dynamics directly using higher order moments and high-order state transition functions. Furthermore, we decompose the higher-order structure using the tensor-train (TT) decomposition to reduce the number of parameters while preserving the model performance. We theoretically establish the approximation properties of Tensor-Train RNNs for general sequence inputs, and such guarantees are not available for usual RNNs. We also demonstrate significant long-term prediction improvements over general RNN and LSTM architectures on a range of simulated environments with nonlinear dynamics, as well on real-world climate and traffic data.

Subfields: Artificial Intelligence, Computational Mathematics, Electrical and Electronic Engineering

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Mingsheng Long
  2. Jianmin Wang
  3. Haixu Wu
  4. Hui Xiong
  5. Kate Saenko
  6. Dacheng Tao
  7. Lijun Sun
  8. Xinyu Chen

In Computational Mathematics

  1. Lijun Sun
  2. Xinyu Chen
  3. Jean Kossaifi
  4. Maja Pantic
  5. Christos Faloutsos
  6. Yannis Panagakis
  7. Yu-Xiang Wang
  8. Haixu Wu

In Electrical and Electronic Engineering

  1. Haixu Wu
  2. Mingsheng Long
  3. Jianmin Wang
  4. Lijun Sun
  5. Xinyu Chen
  6. Maja Pantic
  7. J. Urgen Schmidhuber
  8. Yu-Xiang Wang

Citations in the corpus