Deep transition network with gating mechanism for multivariate time series forecasting

Yimeng Wang,Shi Feng,Bing Wang,Jihong Ouyang
DOI: https://doi.org/10.1007/s10489-023-04503-w
IF: 5.3
2023-07-23
Applied Intelligence
Abstract:As an essential task in the machine learning community, multivariate time series forecasting has many real-world applications, such as PM2.5 forecasting, electricity price forecasting, and traffic flow forecasting. In multivariate time series data, one feature is chosen as the target feature and the other is seen as auxiliary information. These features have very complex spatial dependencies, and most of the existing models put all features together for spatial feature extraction, which is an undifferentiated feature extraction process. It can not distinguish the importance of different features to the target feature. Thus, the spatial feature extraction of data is insufficient. To solve the problems mentioned above, based on the Gated Recurrent Neural Network, we design a new architecture named Auxiliary Information-GRU (AI-GRU) to extract deep temporal and spatial features and combined it into Auxiliary Information-deep Transition Network (Al-DTN). AI-GRU added three gating units, auxiliary information gate controls the amount of auxiliary information flowing into AI-DTN, fusion information gate controls the degree of fusion of target feature and auxiliary information, linear transformation gate isolates the target features from the auxiliary information. Experimental results show that our proposed model outperforms the representative baseline models on the two benchmark datasets.
computer science, artificial intelligence
What problem does this paper attempt to address?