A dual-stage attention-based Bi-LSTM network for multivariate time series prediction

被引:20
|
作者
Cheng, Qi [1 ]
Chen, Yixin [1 ]
Xiao, Yuteng [1 ]
Yin, Hongsheng [1 ]
Liu, Weidong [1 ]
机构
[1] China Univ Min & Technol, Sch Informat & Control Engn, Xuzhou 221116, Jiangsu, Peoples R China
来源
JOURNAL OF SUPERCOMPUTING | 2022年 / 78卷 / 14期
关键词
Multivariate time series; Attention mechanism; Bi-LSTM; Multistep prediction; MODELS;
D O I
10.1007/s11227-022-04506-3
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
In the context of the big data era, time series data present the characteristics of high dimensionality and nonlinearity, which bring great challenges to the prediction of multivariate time series (MTS). In view of the problems of insufficient feature extraction of MTS data and insufficient long-term dependency characteristics, we propose a novel model for MTS prediction named a dual-stage attention-based bidirectional long short-term memory (DABi-LSTM). Specifically, the input attention mechanism gives important driving series greater weight, effectively extracting the features of the driving series. Secondly, the bidirectional long short-term memory network (Bi-LSTM) network extracts the MTS features in two directions. Furthermore, the combination of the attention mechanism, Bi-LSTM and long short-term memory (LSTM) network can effectively solve the problem of insufficient long-term dependence in MTS prediction. Experiments based on typical datasets of finance, environment and energy determine the optimal window size and hidden size of the model, and demonstrate that the model gets the state-of-the-art performance for single-step prediction and multistep prediction compared other methods.
引用
收藏
页码:16214 / 16235
页数:22
相关论文
共 50 条
  • [1] A dual-stage attention-based Bi-LSTM network for multivariate time series prediction
    Qi Cheng
    Yixin Chen
    Yuteng Xiao
    Hongsheng Yin
    Weidong Liu
    [J]. The Journal of Supercomputing, 2022, 78 : 16214 - 16235
  • [2] A dual-stage attention-based Conv-LSTM network for spatio-temporal correlation and multivariate time series prediction
    Xiao, Yuteng
    Yin, Hongsheng
    Zhang, Yudong
    Qi, Honggang
    Zhang, Yundong
    Liu, Zhaoyang
    [J]. INTERNATIONAL JOURNAL OF INTELLIGENT SYSTEMS, 2021, 36 (05) : 2036 - 2057
  • [3] A Dual-Stage Attention-Based Recurrent Neural Network for Time Series Prediction
    Qin, Yao
    Song, Dongjin
    Cheng, Haifeng
    Cheng, Wei
    Jiang, Guofei
    Cottrell, Garrison W.
    [J]. PROCEEDINGS OF THE TWENTY-SIXTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2017, : 2627 - 2633
  • [4] Attention-Based Bi-LSTM for Anomaly Detection on Time-Series Data
    Mishra, Sanket
    Kshirsagar, Varad
    Dwivedula, Rohit
    Hota, Chittaranjan
    [J]. ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2021, PT I, 2021, 12891 : 129 - 140
  • [5] Attention-Based Bi-LSTM Network for Abusive Language Detection
    Nelatoori, Kiran Babu
    Kommanti, Hima Bindu
    [J]. IETE JOURNAL OF RESEARCH, 2023, 69 (11) : 7884 - 7892
  • [6] Adaptive prediction for effluent quality of wastewater treatment plant: Improvement with a dual-stage attention-based LSTM network
    An, Tong
    Feng, Kuanliang
    Cheng, Peijin
    Li, Ruojia
    Zhao, Zihao
    Xu, Xiangyang
    Zhu, Liang
    [J]. JOURNAL OF ENVIRONMENTAL MANAGEMENT, 2024, 359
  • [7] Real-time risk prediction of chemical processes based on attention-based Bi-LSTM
    Qianlin Wang
    Jiaqi Han
    Feng Chen
    Xin Zhang
    Cheng Yun
    Zhan Dou
    Tingjun Yan
    Guoan Yang
    [J]. Chinese Journal of Chemical Engineering, 2024, 75 (11) - 141
  • [8] Temperature Time Series Prediction Model Based on Time Series Decomposition and Bi-LSTM Network
    Zhang, Kun
    Huo, Xing
    Shao, Kun
    [J]. MATHEMATICS, 2023, 11 (09)
  • [9] DSTP-RNN: A dual-stage two-phase attention-based recurrent neural network for long-term and multivariate time series prediction
    Liu, Yeqi
    Gong, Chuanyang
    Yang, Ling
    Chen, Yingyi
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2020, 143 (143)
  • [10] Attention-based gating optimization network for multivariate time series prediction
    Geng, Xiulin
    He, Xiaoyu
    Xu, Lingyu
    Yu, Jie
    [J]. APPLIED SOFT COMPUTING, 2022, 126