Trend-Season Dual-Branch Fusion Network with Harmonic Weighting for Traffic Prediction

multi-scale fusion long-term prediction traffic flow wavelet transform harmonic weighting

Authors

  • Yangyi LIU Intelligent Policing Key Laboratory of Sichuan Province, Sichuan Police College, Luzhou, China
  • Siyuan ZHANG School of Computer Science and Soft Engineering, Southwest Petroleum University, Chengdu, China
  • Wenjun ZHOU
    zhouwenjun@swpu.edu.cn
    School of Computer Science and Soft Engineering, Southwest Petroleum University, Chengdu, China
  • Yifan WANG School of Computer Science and Soft Engineering, Southwest Petroleum University, Chengdu, China
  • Quan ZHANG School of Computer Science and Soft Engineering, Southwest Petroleum University, Chengdu, China
  • Bo PENG School of Computer Science and Soft Engineering, Southwest Petroleum University, Chengdu, China

Downloads

Traffic flow prediction is an important component of intelligent transportation systems. Most existing studies adopt spatio-temporal methods for real-time short-term prediction. However, when dealing with complex and dynamically changing historical data in long-term prediction tasks, several challenges arise, such as high complexity, high computational cost and unstable spatial dependencies. In response to this deficiency, this paper proposes a trend-season dual-branch fusion network with harmonic weighting for long-sequence traffic flow prediction. This model decomposes the original sequence into trend and seasonal components. Specifically, the trend branch applies multi-scale sampling, and the trend components at different scales are fused in a top-down manner to enhance information representation. Finally, a linear layer is used to predict the trend results. The seasonal branch encodes the seasonal components through wavelet transform, and a harmonic weighting mechanism is designed to adaptively fuse the prediction results of both branches. Experiments conducted on multiple public transportation datasets demonstrate that the proposed model significantly outperforms existing mainstream methods in long-term predictions at time steps of 96, 192, 288 and 336, thereby verifying the effectiveness and robustness of time-series modelling approach in long-sequence prediction scenarios.