Multiscale Traffic Dynamics Representation for Forecasting via MEMD-Guided Dual-Branch Recurrent Networks
Abstract
1. Introduction
- A MEMD-guided trend–residual framework is used to reconstruct frequency-aligned components into compact trend and residual representations.
- A heterogeneous dual-branch architecture models the reconstructed components with component-specific encoders: LSTM for trend evolution and Bi-GRU for residual fluctuations.
- Experiments and ablations on PeMS04 and PeMS08 show competitive results, suggesting that the gains mainly come from trend–residual representation learning and component-specific temporal modeling.
2. Related Work
3. Methodology
3.1. Overview of the Proposed Framework
3.2. Spectral Decomposition via MEMD
3.3. Component Reconstruction and Heterogeneous Modeling
3.3.1. Adaptive Component Reconstruction of MEMD Modes
| Algorithm 1 MEMD-Based Component Reconstruction |
|
3.3.2. Computational Cost Discussion
3.3.3. Trend Representation Learning Based on LSTM
3.3.4. Residual Dynamics Modeling Based on Bi-GRU
3.4. Adaptive Fusion and Optimization
3.4.1. Lightweight Fusion and Prediction Head
3.4.2. End-to-End Optimization
| Algorithm 2 Training Procedure of the Proposed Hybrid Framework |
|
4. Experiments
4.1. Experimental Setup
4.2. Experimental Results
4.3. Qualitative Forecasting Behavior
5. Ablation Study
5.1. Ablation Study on Decomposition
5.2. Ablation Study on IMF Reconstruction
5.3. Ablation Study on Branch Assignment
5.4. Ablation Study on Temporal Variables
5.5. Ablation Study on Local Transition Behavior
6. Conclusions and Future Work
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Conflicts of Interest
References
- Zhou, Y. An urban traffic flow prediction method based on multi-source data fusion. In Proceedings of the 2025 International Conference on Software Engineering and Computer Applications; Association for Computing Machinery: New York, NY, USA, 2025; pp. 308–313. [Google Scholar]
- Diao, C.; Zhang, D.; Liang, W.; Jiang, M.; Li, K. A novel attention-based dynamic multi-graph spatial-temporal graph neural network model for traffic prediction. IEEE Trans. Emerg. Top. Comput. Intell. 2024, 9, 1910–1923. [Google Scholar] [CrossRef]
- Chen, Y.; Shu, T.; Zhou, X.; Zheng, X.; Kawai, A.; Fueda, K.; Yan, Z.; Liang, W.; Wang, K.I.K. Graph attention network with spatial-temporal clustering for traffic flow forecasting in intelligent transportation system. IEEE Trans. Intell. Transp. Syst. 2022, 24, 8727–8737. [Google Scholar] [CrossRef]
- Ye, B.L.; Zhang, M.; Li, L.; Liu, C.; Wu, W. A survey of traffic flow prediction methods based on long short-term memory networks. IEEE Intell. Transp. Syst. Mag. 2024, 16, 87–112. [Google Scholar] [CrossRef]
- Gao, D.; Li, P.; Wang, M.; Liang, Y.; Liu, S.; Zhou, J.; Wang, L.; Zhang, Y. CSF-GTNet: A novel multi-dimensional feature fusion network based on Convnext-GeLU-BiLSTM for EEG-signals-enabled fatigue driving detection. IEEE J. Biomed. Health Inform. 2023, 28, 2558–2568. [Google Scholar] [CrossRef]
- Qian, W.; Zhao, Y.; Zhang, D.; Chen, B.; Zheng, K.; Zhou, X. Towards a unified understanding of uncertainty quantification in traffic flow forecasting. IEEE Trans. Knowl. Data Eng. 2023, 36, 2239–2256. [Google Scholar] [CrossRef]
- Zeng, A.; Chen, M.; Zhang, L.; Xu, Q. Are transformers effective for time series forecasting? In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Washington, DC, USA, 2023; Volume 37, pp. 11121–11128. [Google Scholar]
- Hu, H.X.; Hu, Q.; Tan, G.; Zhang, Y.; Lin, Z.Z. A multi-layer model based on transformer and deep learning for traffic flow prediction. IEEE Trans. Intell. Transp. Syst. 2023, 25, 443–451. [Google Scholar] [CrossRef]
- Zhang, J.; Mao, S.; Yang, L.; Ma, W.; Li, S.; Gao, Z. Physics-informed deep learning for traffic state estimation based on the traffic flow model and computational graph method. Inf. Fusion 2024, 101, 101971. [Google Scholar] [CrossRef]
- Huang, Y.; Hasan, N.; Deng, C.; Bao, Y. Multivariate empirical mode decomposition based hybrid model for day-ahead peak load forecasting. Energy 2022, 239, 122245. [Google Scholar] [CrossRef]
- Chen, Z.; Lu, Z.; Chen, Q.; Zhong, H.; Zhang, Y.; Xue, J.; Wu, C. Spatial–temporal short-term traffic flow prediction model based on dynamical-learning graph convolution mechanism. Inf. Sci. 2022, 611, 522–539. [Google Scholar] [CrossRef]
- Al-Selwi, H.F.; Abd Aziz, A.B.; Abas, F.S.; Hamzah, N.A.A.; Mahmud, A.B. The impact of weather data on traffic flow prediction models. IAES Int. J. Artif. Intell. 2022, 11, 1223. [Google Scholar] [CrossRef]
- Fan, J.; Zhu, F.; Weng, W.; Zhang, X.; Jiang, H.; Tian, H.; Wu, H. Dynamic modeling and analysis of Bi-directional traffic flows through a deep spatio-temporal graph neural network. IEEE Trans. Big Data 2025, 11, 3016–3028. [Google Scholar] [CrossRef]
- Wang, C.; Zhang, K.; Wang, H.; Chen, B. Auto-STGCN: Autonomous spatial-temporal graph convolutional network search. ACM Trans. Knowl. Discov. Data 2023, 17, 1–21. [Google Scholar] [CrossRef]
- Huang, Y.; Weng, Y.; Yu, S.; Chen, X. Diffusion convolutional recurrent neural network with rank influence learning for traffic forecasting. In Proceedings of the 2019 18th IEEE International Conference on Trust, Security and Privacy in Computing and Communications/13th IEEE International Conference on Big Data Science and Engineering (TrustCom/BigDataSE); IEEE: Piscataway, NJ, USA, 2019; pp. 678–685. [Google Scholar]
- Wu, Z.; Pan, S.; Long, G.; Jiang, J.; Zhang, C. Graph WaveNet for Deep Spatial-Temporal Graph Modeling. In Proceedings of the 28th International Joint Conference on Artificial Intelligence (IJCAI-19); International Joint Conference on Artificial Intelligence (IJCAI); AAAI Press: Washington, DC, USA, 2019. [Google Scholar]
- Zhao, L.; Song, Y.; Zhang, C.; Liu, Y.; Wang, P.; Lin, T.; Deng, M.; Li, H. T-GCN: A temporal graph convolutional network for traffic prediction. IEEE Trans. Intell. Transp. Syst. 2019, 21, 3848–3858. [Google Scholar] [CrossRef]
- Song, C.; Lin, Y.; Guo, S.; Wan, H. Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data Forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Washington, DC, USA, 2020; pp. 914–921. [Google Scholar]
- Bai, L.; Yao, L.; Li, C.; Wang, X.; Wang, C. Adaptive graph convolutional recurrent network for traffic forecasting. Adv. Neural Inf. Process. Syst. 2020, 33, 17804–17815. [Google Scholar]
- Guo, S.; Lin, Y.; Feng, N.; Song, C.; Wan, H. Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Washington, DC, USA, 2019; pp. 922–929. [Google Scholar]
- Zheng, C.; Fan, X.; Wang, C.; Qi, J. GMAN: A Graph Multi-Attention Network for Traffic Prediction. In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Washington, DC, USA, 2020; pp. 1234–1241. [Google Scholar]
- Cai, L.; Janowicz, K.; Mai, G.; Yan, B.; Zhu, R. Traffic transformer: Capturing the continuity and periodicity of time series for traffic forecasting. Trans. GIS 2020, 24, 736–755. [Google Scholar] [CrossRef]
- Wu, H.; Xu, J.; Wang, J.; Long, M. Autoformer: Decomposition transformers with auto-correlation for long-term series forecasting. Adv. Neural Inf. Process. Syst. 2021, 34, 22419–22430. [Google Scholar]
- Jiang, J.; Han, C.; Zhao, W.X.; Wang, J. PDFormer: Propagation Delay-Aware Dynamic Long-Range Transformer for Traffic Flow Prediction. In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Washington, DC, USA, 2023; pp. 4365–4373. [Google Scholar]
- Liu, H.; Dong, Z.; Jiang, R.; Deng, J.; Deng, J.; Chen, Q.; Song, X. Spatio-temporal adaptive embedding makes vanilla transformer sota for traffic forecasting. In Proceedings of the 32nd ACM International Conference on Information and Knowledge Management; Association for Computing Machinery: New York, NY, USA, 2023; pp. 4125–4129. [Google Scholar]
- Pan, Z.; Liang, Y.; Wang, W.; Yu, Y.; Zheng, Y.; Zhang, J. Urban traffic prediction from spatio-temporal data using deep meta learning. In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining; Association for Computing Machinery: New York, NY, USA, 2019; pp. 1720–1730. [Google Scholar]
- Wang, Y.; Zhang, D.; Liu, Y.; Dai, B.; Lee, L.H. Enhancing transportation systems via deep learning: A survey. Transp. Res. Part C Emerg. Technol. 2019, 99, 144–163. [Google Scholar] [CrossRef]
- Tedjopurnomo, D.A.; Bao, Z.; Zheng, B.; Choudhury, F.M.; Qin, A.K. A survey on modern deep neural network for traffic prediction: Trends, methods and challenges. IEEE Trans. Knowl. Data Eng. 2020, 34, 1544–1561. [Google Scholar] [CrossRef]
- Fafoutellis, P.; Vlahogianni, E.I. A theory-informed multivariate causal framework for trustworthy short-term urban traffic forecasting. Transp. Res. Part C Emerg. Technol. 2025, 170, 104945. [Google Scholar] [CrossRef]
- Wang, R.; Xin, Y.; Zhang, Y.; Perez-Cruz, F.; Raubal, M. Counterfactual explanations for deep learning-based traffic forecasting. Commun. Transp. Res. 2025, 5, 100176. [Google Scholar] [CrossRef]
- Kong, L.; Yang, H.; Li, W.; Zhang, Y.; Guan, J.; Zhou, S. Traffexplainer: A framework toward gnn-based interpretable traffic prediction. IEEE Trans. Artif. Intell. 2024, 6, 559–573. [Google Scholar] [CrossRef]
- Chen, J.; Zheng, L.; Hu, Y.; Wang, W.; Zhang, H.; Hu, X. Traffic flow matrix-based graph neural network with attention mechanism for traffic flow prediction. Inf. Fusion 2024, 104, 102146. [Google Scholar] [CrossRef]
- Yang, H.F.; Chen, Y.P.P. Hybrid deep learning and empirical mode decomposition model for time series applications. Expert Syst. Appl. 2019, 120, 128–138. [Google Scholar] [CrossRef]
- Tan, Z.; Shi, Y.; Zhang, Y. Traffic Flow Prediction Based on Multimodal Spatio-Temporal Bayesian Neural Network. In Proceedings of the International Conference on Information, Computing and Technology; Springer: Cham, Switzerland, 2025; pp. 201–211. [Google Scholar]
- Sun, R.; Cheng, N.; Li, C.; Quan, W.; Zhou, H.; Wang, Y.; Zhang, W.; Shen, X. A comprehensive survey of knowledge-driven deep learning for intelligent wireless network optimization in 6G. IEEE Commun. Surv. Tutor. 2025, 28, 1099–1135. [Google Scholar] [CrossRef]
- Wang, L.; He, H.; Dong, Y.; Li, X.; Gan, W.; Zhang, X. Predicting street-level distribution of bike-sharing traffic volume in metro station areas using integrated generative adversarial networks. J. Transp. Geogr. 2026, 130, 104456. [Google Scholar] [CrossRef]
- Ma, C.; Zhao, Y.; Dai, G.; Xu, X.; Wong, S.C. A novel STFSA-CNN-GRU hybrid model for short-term traffic speed prediction. IEEE Trans. Intell. Transp. Syst. 2022, 24, 3728–3737. [Google Scholar] [CrossRef]
- Deng, C.; Huang, Y.; Hasan, N.; Bao, Y. Multi-step-ahead stock price index forecasting using long short-term memory model with multivariate empirical mode decomposition. Inf. Sci. 2022, 607, 297–321. [Google Scholar] [CrossRef]
- Ur Rehman, N.; Mandic, D.P. Filter bank property of multivariate empirical mode decomposition. IEEE Trans. Signal Process. 2011, 59, 2421–2426. [Google Scholar] [CrossRef]
- Naheliya, B.; Redhu, P.; Kumar, K. Bi-directional long short term memory neural network for short-term traffic speed prediction using gravitational search algorithm. Int. J. Intell. Transp. Syst. Res. 2024, 22, 316–327. [Google Scholar] [CrossRef]
- Ma, X.; Tao, Z.; Wang, Y.; Yu, H.; Wang, Y. Long short-term memory neural network for traffic speed prediction using remote microwave sensor data. Transp. Res. Part C Emerg. Technol. 2015, 54, 187–197. [Google Scholar] [CrossRef]
- Xiao, J.; Huang, Y. Traffic state identification method based on GA-EWFCM. In Proceedings of the Tenth International Conference on Electromechanical Control Technology and Transportation (ICECTT 2025); SPIE: Washington, DC, USA, 2025; Volume 13781, pp. 656–661. [Google Scholar]
- Wu, L.; Li, S.; Li, H.; Huang, J.; Lei, X.; Jiang, H. Spatio-temporal Transfer Learning for Urban Data Modeling. In Proceedings of the 2025 IEEE 28th International Conference on Computational Science and Engineering (CSE); IEEE: Piscataway, NJ, USA, 2025; pp. 30–37. [Google Scholar]
- Lin, S.; Lin, W.; Wu, W.; Zhao, F.; Mo, R.; Zhang, H. Segrnn: Segment recurrent neural network for long-term time series forecasting. IEEE Internet Things J. 2025, 13, 9861–9871. [Google Scholar] [CrossRef]
- Lei, Z.; Dong, Y.; Li, J.; Chen, C. St-fit: Inductive spatial-temporal forecasting with limited training data. In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Washington, DC, USA, 2025; Volume 39, pp. 12031–12039. [Google Scholar]
- Sims, C.A. Macroeconomics and reality. In Econometrica: Journal of the Econometric Society; The Econometric Society: New York, NY, USA, 1980; pp. 1–48. [Google Scholar]
- Greff, K.; Srivastava, R.K.; Koutník, J.; Steunebrink, B.R.; Schmidhuber, J. LSTM: A search space odyssey. IEEE Trans. Neural Netw. Learn. Syst. 2016, 28, 2222–2232. [Google Scholar] [CrossRef]
- Gao, J.; Zhang, X.; Tian, L.; Liu, Y.; Wang, J.; Li, Z.; Hu, X. MTGNN: Multi-task graph neural network based few-shot learning for disease similarity measurement. Methods 2022, 198, 88–95. [Google Scholar] [CrossRef]
- Li, M.; Zhu, Z. Spatial-Temporal Fusion Graph Neural Networks for Traffic Flow Forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Washington, DC, USA, 2021; pp. 4189–4196. [Google Scholar]
- Choi, J.; Choi, H.; Hwang, J.; Park, N. Graph Neural Controlled Differential Equations for Traffic Forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Washington, DC, USA, 2022; pp. 6367–6374. [Google Scholar]
- Han, L.; Du, B.; Sun, L.; Fu, Y.; Lv, Y.; Xiong, H. Dynamic and multi-faceted spatio-temporal deep learning for traffic speed forecasting. In Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining; Association for Computing Machinery: New York, NY, USA, 2021; pp. 547–555. [Google Scholar]
- Shao, Z.; Zhang, Z.; Wei, W.; Wang, F.; Xu, Y.; Cao, X.; Jensen, C.S. Decoupled dynamic spatial-temporal graph neural network for traffic forecasting. Proc. VLDB Endow. 2022, 15, 2733–2746. [Google Scholar] [CrossRef]
- Shao, Z.; Zhang, Z.; Wang, F.; Wei, W.; Xu, Y. Spatial-temporal identity: A simple yet effective baseline for multivariate time series forecasting. In Proceedings of the 31st ACM International Conference on Information & Knowledge Management; Association for Computing Machinery: New York, NY, USA, 2022; pp. 4454–4458. [Google Scholar]






| Method | PeMS04 | PeMS08 | ||||
|---|---|---|---|---|---|---|
| MAE ↓ | RMSE ↓ | MAPE (%) ↓ | MAE ↓ | RMSE ↓ | MAPE (%) ↓ | |
| HA [45] | 24.50 | 39.83 | 16.60 | 21.19 | 36.64 | 13.82 |
| VAR [46] | 21.87 | 32.26 | 15.70 | 18.66 | 27.35 | 12.81 |
| LSTM [47] | 21.37 | 33.31 | 15.21 | 17.38 | 26.27 | 12.63 |
| DCRNN [15] | 24.71 | 38.12 | 17.12 | 17.86 | 28.83 | 12.45 |
| STGCN [14] | 22.70 | 35.55 | 14.59 | 18.61 | 28.16 | 13.12 |
| Graph WaveNet [16] | 20.65 | 33.08 | 14.66 | 16.23 | 25.02 | 12.43 |
| MTGNN [48] | 20.08 | 32.56 | 13.96 | 16.39 | 25.93 | 10.17 |
| STFGNN [49] | 20.83 | 32.09 | 14.02 | 16.46 | 25.81 | 10.92 |
| STGNCDE [50] | 20.21 | 32.09 | 13.76 | 16.45 | 25.81 | 10.92 |
| DMSTGCN [51] | 23.59 | 36.83 | 16.43 | 18.65 | 29.14 | 12.01 |
| D2STGNN [52] | 20.55 | 32.99 | 13.82 | 16.69 | 26.41 | 11.17 |
| ASTGCN [20] | 22.93 | 35.22 | 16.56 | 18.61 | 28.16 | 13.08 |
| GMAN [21] | 20.14 | 32.60 | 14.20 | 16.31 | 25.92 | 11.13 |
| STID [53] | 20.58 | 32.79 | 14.38 | 16.58 | 26.89 | 11.33 |
| Traffic Transformer [22] | 19.75 | 32.35 | 12.82 | 15.79 | 24.88 | 9.93 |
| Proposed (Ours) | 19.67 | 31.59 | 12.95 | 15.51 | 24.43 | 9.86 |
| Model | PeMS04 | PeMS08 | ||
|---|---|---|---|---|
| RMSE ↓ | MAE ↓ | RMSE ↓ | MAE ↓ | |
| Baseline | 38.71 | 23.79 | 33.25 | 21.40 |
| Trend Only | 39.11 | 24.21 | 33.95 | 22.01 |
| Fluctuation Only | 49.30 | 30.05 | 44.46 | 30.08 |
| Decomp Concat | 38.35 | 23.34 | 33.03 | 21.03 |
| Decomp Gated | 38.57 | 23.56 | 33.12 | 20.92 |
| Decomp Attention | 39.26 | 24.30 | 33.25 | 21.29 |
| Reconstruction Strategy | MAE ↓ | RMSE ↓ | MAPE (%) ↓ |
|---|---|---|---|
| Raw Input | |||
| IMF-wise Modeling | |||
| Two Groups (Ours) | |||
| Three Groups | |||
| Four Groups |
| Architecture | MAE ↓ | RMSE ↓ | MAPE (%) ↓ |
|---|---|---|---|
| LSTM–LSTM | |||
| GRU–GRU | |||
| Bi-GRU–Bi-GRU | |||
| LSTM–GRU | |||
| GRU–Bi-GRU | |||
| Transformer–Transformer | |||
| LSTM–Bi-GRU (Ours) |
| Input Setting | MAE ↓ | RMSE ↓ | MAPE (%) ↓ |
|---|---|---|---|
| Flow Only | |||
| Flow + Time-of-day | |||
| Flow + Day-of-week | |||
| Flow + Time + Day-of-week | |||
| Flow + Time + Day-of-week + Weekend |
| Regime | Model | MAE ↓ | RMSE ↓ | MAPE (%) ↓ |
|---|---|---|---|---|
| Stable | Baseline | 15.39 | 26.28 | 22.61 |
| Ours | 12.47 | 21.10 | 19.41 | |
| Gain (%) | 18.96 | 19.71 | 14.15 | |
| Peak | Baseline | 41.96 | 57.77 | 10.30 |
| Ours | 34.04 | 47.88 | 8.42 | |
| Gain (%) | 18.88 | 17.13 | 18.30 | |
| Rapid Transition | Baseline | 41.24 | 56.34 | 17.32 |
| Ours | 31.79 | 45.07 | 12.91 | |
| Gain (%) | 22.91 | 20.01 | 25.44 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Qian, Y.; Kang, T.; Zhang, S.; Li, C.; Wang, X.; Zhao, S. Multiscale Traffic Dynamics Representation for Forecasting via MEMD-Guided Dual-Branch Recurrent Networks. Sensors 2026, 26, 3369. https://doi.org/10.3390/s26113369
Qian Y, Kang T, Zhang S, Li C, Wang X, Zhao S. Multiscale Traffic Dynamics Representation for Forecasting via MEMD-Guided Dual-Branch Recurrent Networks. Sensors. 2026; 26(11):3369. https://doi.org/10.3390/s26113369
Chicago/Turabian StyleQian, Yichen, Taiming Kang, Shengduo Zhang, Chaoneng Li, Xiaolong Wang, and Shuxu Zhao. 2026. "Multiscale Traffic Dynamics Representation for Forecasting via MEMD-Guided Dual-Branch Recurrent Networks" Sensors 26, no. 11: 3369. https://doi.org/10.3390/s26113369
APA StyleQian, Y., Kang, T., Zhang, S., Li, C., Wang, X., & Zhao, S. (2026). Multiscale Traffic Dynamics Representation for Forecasting via MEMD-Guided Dual-Branch Recurrent Networks. Sensors, 26(11), 3369. https://doi.org/10.3390/s26113369

