Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DWAFM: Dynamic Weighted Graph Structure Embedding Integrated with Attention and Frequency-Domain MLPs for Traffic Forecasting

About

Accurate traffic prediction is a key task for intelligent transportation systems. The core difficulty lies in accurately modeling the complex spatial-temporal dependencies in traffic data. In recent years, improvements in network architecture have failed to bring significant performance enhancements, while embedding technology has shown great potential. However, existing embedding methods often ignore graph structure information or rely solely on static graph structures, making it difficult to effectively capture the dynamic associations between nodes that evolve over time. To address this issue, this letter proposes a novel dynamic weighted graph structure (DWGS) embedding method, which relies on a graph structure that can truly reflect the changes in the strength of dynamic associations between nodes over time. By first combining the DWGS embedding with the spatial-temporal adaptive embedding, as well as the temporal embedding and feature embedding, and then integrating attention and frequency-domain multi-layer perceptrons (MLPs), we design a novel traffic prediction model, termed the DWGS embedding integrated with attention and frequency-domain MLPs (DWAFM). Experiments on five real-world traffic datasets show that the DWAFM achieves better prediction performance than some state-of-the-arts.

Sen Shi, Zhichao Zhang, Yangfan He• 2026

Related benchmarks

TaskDatasetResultRank
Traffic ForecastingPeMSD7 (test)
MAE2.59
95
Traffic Flow ForecastingPEMS08 (test)
MAE13.57
78
Traffic Flow ForecastingPEMS04 (test)
MAE18.14
78
Traffic Flow ForecastingPEMS03 (test)
MAE14.81
61
Traffic ForecastingPEMSD7 L (test)
MAE2.79
12
Showing 5 of 5 rows

Other info

Follow for update