Multistage attention network for multivariate time series prediction

被引:98
作者
Hu, Jun [1 ,2 ]
Zheng, Wendong [1 ]
机构
[1] Hunan Univ, Coll Comp Sci & Elect Engn, Changsha, Peoples R China
[2] Nanjing Univ, State Key Lab Novel Software Technol, Nanjing, Peoples R China
关键词
Attention mechanism; Multivariate time series prediction; Long short-term memory; POWER;
D O I
10.1016/j.neucom.2019.11.060
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The deep learning model has been used to predict the variation rule of the target series of multivariate time series data. Based on the attention mechanism, the influence information of multiple non-predictive time series on target series in different time stages is processed as the same weight in the previous studies. However, on real-world datasets, multiple non-predictive time series has different influence (such as different mutation information) on target series in different time stages. Therefore, a new multistage attention network is designed to capture the different influence. The model is mainly composed of the influential attention mechanism and temporal attention mechanism. In the influential attention mechanism, the same and different time stage attention mechanisms are used to capture the influence information of different non-predictive time series on the target series over time. In the temporal attention mechanism, the variation law of data can be captured over time. Besides, the prediction performance of proposed model on two different real-world multivariate time series datasets is comprehensively evaluated. The results show that, the prediction performance of the proposed model beat all baseline models and SOTA models. In a word, the multistage attention network model can effectively learn the information of the influence of different non-predictive time series on the target series in different time stages in the historical data. (C) 2019 Elsevier B.V. All rights reserved.
引用
收藏
页码:122 / 137
页数:16
相关论文
共 33 条
[1]  
Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
[2]  
[Anonymous], P INT JOINT C NEUR N
[3]  
[Anonymous], P C EMP METH NAT LAN
[4]  
[Anonymous], 2017, P ICLR
[5]  
[Anonymous], NEUROCOMPUTING
[6]  
[Anonymous], P ICDM
[7]  
[Anonymous], 2016, P ACL
[8]  
Ba J., 2015, P INT C LEARNING REP
[9]  
Bahdanau D., ABS14090473 CORR
[10]  
Chollet F., 2015, KERAS