中国物理B ›› 2026, Vol. 35 ›› Issue (6): 60204-060204.doi: 10.1088/1674-1056/ae3069

• • 上一篇    下一篇

Traffic flow prediction based on frequency-domain dynamic graph and Mamba

Yifei Zhang(张逸飞)1 and Jialin He(何嘉林)1,2,3,†   

  1. 1 School of Computer Science, China West Normal University, Nanchong 637009, China;
    2 Artificial Intelligence Key Laboratory of Nanchong, China West Normal University, Nanchong 637009, China;
    3 The Internet of Things Perception and Big Data Analysis Key Laboratory of Nanchong City, Nanchong 637009, China
  • 收稿日期:2025-10-08 修回日期:2025-12-21 接受日期:2025-12-23 发布日期:2026-06-18
  • 通讯作者: Jialin He E-mail:hejialin32@126.com
  • 基金资助:
    Project supported by the National Natural Science Foundation of China (Grant No. 62176217), the Program from the Sichuan Provincial Science and Technology, China (Grant No. 2018RZ0081), and the Fundamental Research Funds of China West Normal University (Grant No. 17E063).

Traffic flow prediction based on frequency-domain dynamic graph and Mamba

Yifei Zhang(张逸飞)1 and Jialin He(何嘉林)1,2,3,†   

  1. 1 School of Computer Science, China West Normal University, Nanchong 637009, China;
    2 Artificial Intelligence Key Laboratory of Nanchong, China West Normal University, Nanchong 637009, China;
    3 The Internet of Things Perception and Big Data Analysis Key Laboratory of Nanchong City, Nanchong 637009, China
  • Received:2025-10-08 Revised:2025-12-21 Accepted:2025-12-23 Published:2026-06-18
  • Contact: Jialin He E-mail:hejialin32@126.com
  • Supported by:
    Project supported by the National Natural Science Foundation of China (Grant No. 62176217), the Program from the Sichuan Provincial Science and Technology, China (Grant No. 2018RZ0081), and the Fundamental Research Funds of China West Normal University (Grant No. 17E063).

摘要: With the rapid advancement of intelligent transportation systems (ITS), urban traffic prediction faces significant challenges in effectively modeling complex spatio-temporal dynamics while maintaining computational efficiency. Existing approaches are often limited by static graph structures and the high computational cost of self-attention mechanisms when processing long sequences. To overcome these limitations, this paper proposes a novel framework, termed spatio-temporal frequency-domain mamba network (STFD-MambaNet). Specifically, the framework integrates a frequency-domain dynamic graph learner to capture evolving traffic topologies and employs the Mamba structured state space model to efficiently extract long-range temporal dependencies with linear complexity. Furthermore, a hierarchical spatial modeling module is developed to characterize multi-scale spatial correlations. Experiments conducted on four real-world datasets demonstrate that STFD-MambaNet consistently outperforms state-of-the-art methods in both accuracy and efficiency. The results further demonstrate the effective complementarity between frequency-domain dynamic graph learning and structured state space modeling, providing a robust solution for spatio-temporal traffic forecasting.

关键词: traffic flow forecasting, frequency-domain graph learning, Mamba, graph convolutional network(GCN)

Abstract: With the rapid advancement of intelligent transportation systems (ITS), urban traffic prediction faces significant challenges in effectively modeling complex spatio-temporal dynamics while maintaining computational efficiency. Existing approaches are often limited by static graph structures and the high computational cost of self-attention mechanisms when processing long sequences. To overcome these limitations, this paper proposes a novel framework, termed spatio-temporal frequency-domain mamba network (STFD-MambaNet). Specifically, the framework integrates a frequency-domain dynamic graph learner to capture evolving traffic topologies and employs the Mamba structured state space model to efficiently extract long-range temporal dependencies with linear complexity. Furthermore, a hierarchical spatial modeling module is developed to characterize multi-scale spatial correlations. Experiments conducted on four real-world datasets demonstrate that STFD-MambaNet consistently outperforms state-of-the-art methods in both accuracy and efficiency. The results further demonstrate the effective complementarity between frequency-domain dynamic graph learning and structured state space modeling, providing a robust solution for spatio-temporal traffic forecasting.

Key words: traffic flow forecasting, frequency-domain graph learning, Mamba, graph convolutional network(GCN)

中图分类号:  (Combinatorics; graph theory)

  • 02.10.Ox
45.70.Vn (Granular models of complex systems; traffic flow) 06.30.Ft (Time and frequency)