1. TCLN: A Transformer-based Conv-LSTM network for multivariate time series forecasting.
- Author
-
Ma, Shusen, Zhang, Tianhao, Zhao, Yun-Bo, Kang, Yu, and Bai, Peng
- Subjects
CONVOLUTIONAL neural networks ,TRANSFORMER models ,TRAFFIC estimation ,FORECASTING ,AUTOREGRESSIVE models - Abstract
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). [ABSTRACT FROM AUTHOR]
- Published
- 2023
- Full Text
- View/download PDF