arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1806.06384cs.LGstat.ML
收藏

Multi-variable LSTM neural network for autoregressive exogenous model

ETH Zurich(苏黎世联邦理工学院);EPFL Lausanne(洛桑联邦理工学院)

Tian Guo, Tao Lin

英文摘要:

In this paper, we propose multi-variable LSTM capable of accurate forecasting and variable importance interpretation for time series with exogenous variables. Current attention mechanism in recurrent neural networks mostly focuses on the temporal aspect of data and falls short of characterizing variable importance. To this end, the multi-variable LSTM equipped with tensorized hidden states is developed to learn hidden states for individual variables, which give rise to our mixture temporal and variable attention. Based on such attention mechanism, we infer and quantify variable importance. Extensive experiments using real datasets with Granger-causality test and the synthetic dataset with ground truth demonstrate the prediction performance and interpretability of multi-variable LSTM in comparison to a variety of baselines. It exhibits the prospect of multi-variable LSTM as an end-to-end framework for both forecasting and knowledge discovery.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑