TCLN: A Transformer-based Conv-LSTM network for multivariate time series forecasting

被引:3
|
作者
Ma, Shusen [1 ]
Zhang, Tianhao [2 ]
Zhao, Yun-Bo [1 ,2 ,3 ]
Kang, Yu [1 ,2 ,3 ]
Bai, Peng [2 ]
机构
[1] USTC, Inst Adv Technol, Hefei 230031, Anhui, Peoples R China
[2] USTC, Dept Automat, Hefei 230031, Anhui, Peoples R China
[3] Hefei Comprehens Natl Sci Ctr, Inst Artificial Intelligence, Hefei 230031, Anhui, Peoples R China
基金
中国国家自然科学基金;
关键词
MTSF; Transformer; CNN; LSTM; Autoregressive model;
D O I
10.1007/s10489-023-04980-z
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The study of multivariate time series forecasting (MTSF) problems has high significance in many areas, such as industrial forecasting and traffic flow forecasting. Traditional forecasting models pay more attention to the temporal features of variables and lack depth in extracting spatial and spatiotemporal features between variables. In this paper, a novel model based on the Transformer, convolutional neural network (CNN), and long short-term memory (LSTM) network is proposed to address the issues. The model first extracts the spatial feature vectors through the proposed Multi-kernel CNN. Then it fully extracts the temporal information by the Encoder layer that consists of the Transformer encoder layer and the LSTM network, which can also obtain the potential spatiotemporal correlation. To extract more feature information, we stack multiple Encoder layers. Finally, the output is decoded by the Decoder layer composed of the ReLU activation function and the Linear layer. To further improve the model's robustness, we also integrate an autoregressive model. In model evaluation, the proposed model achieves significant performance improvements over the current benchmark methods for MTSF tasks on four datasets. Further experiments demonstrate that the model can be used for long-horizon forecasting and achieve satisfactory results on the yield forecasting of test items (our private dataset, TIOB).
引用
收藏
页码:28401 / 28417
页数:17
相关论文
共 50 条
  • [11] Rankformer: Leveraging Rank Correlation for Transformer-based Time Series Forecasting
    Ouyang, Zuokun
    Jabloun, Meryem
    Ravier, Philippe
    [J]. 2023 IEEE STATISTICAL SIGNAL PROCESSING WORKSHOP, SSP, 2023, : 85 - 89
  • [12] Enhanced Linear and Vision Transformer-Based Architectures for Time Series Forecasting
    Alharthi, Musleh
    Mahmood, Ausif
    [J]. BIG DATA AND COGNITIVE COMPUTING, 2024, 8 (05)
  • [13] AGCNT: Adaptive Graph Convolutional Network for Transformer-based Long Sequence Time-Series Forecasting
    Su, Hongyang
    Wang, Xiaolong
    Qin, Yang
    [J]. PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, CIKM 2021, 2021, : 3439 - 3442
  • [14] Multi-Scale Spatiotemporal Conv-LSTM Network for Video Saliency Detection
    Tang, Yi
    Zou, Wenbin
    Jin, Zhi
    Li, Xia
    [J]. ICMR '18: PROCEEDINGS OF THE 2018 ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA RETRIEVAL, 2018, : 362 - 369
  • [15] Spatial-Temporal Convolutional Transformer Network for Multivariate Time Series Forecasting
    Huang, Lei
    Mao, Feng
    Zhang, Kai
    Li, Zhiheng
    [J]. SENSORS, 2022, 22 (03)
  • [16] Transformer-Based Hybrid Forecasting Model for Multivariate Renewable Energy
    Padilha, Guilherme Afonso Galindo
    Ko, JeongRyun
    Jung, Jason J.
    Gomes de Mattos Neto, Paulo Salgado
    [J]. APPLIED SCIENCES-BASEL, 2022, 12 (21):
  • [17] Using Conv-LSTM to Refine Features for Lightweight Image Super-Resolution Network
    Zhang, Jiangtao
    Qu, Yanyun
    Chen, Liang
    [J]. IMAGE AND GRAPHICS (ICIG 2021), PT III, 2021, 12890 : 230 - 240
  • [18] Unsupervised Anomaly Detection in Multivariate Time Series through Transformer-based Variational Autoencoder
    Zhang, Hongwei
    Xia, Yuanqing
    Yan, Tijin
    Liu, Guiyang
    [J]. PROCEEDINGS OF THE 33RD CHINESE CONTROL AND DECISION CONFERENCE (CCDC 2021), 2021, : 281 - 286
  • [19] Short-Term Wind Speed Forecasting With Principle-Subordinate Predictor Based on Conv-LSTM and Improved BPNN
    Chen, Gonggui
    Li, Lijun
    Zhang, Zhizhong
    Li, Shuaiyong
    [J]. IEEE ACCESS, 2020, 8 : 67955 - 67973
  • [20] Time-Aware Transformer-based Network for Clinical Notes Series Prediction
    Zhang, Dongyu
    Thadajarassiri, Jidapa
    Sen, Cansu
    Rundensteiner, Elke
    [J]. MACHINE LEARNING FOR HEALTHCARE CONFERENCE, VOL 126, 2020, 126 : 566 - 587