Unifying spatiotemporal and frequential attention for traffic prediction
Abstract Intelligent transportation systems heavily rely on forecasting urban traffic flow, and a variety of approaches have been developed for this purpose. However, most current methods focus on exploring spatial and temporal dependencies in historical traffic data, while often overlooking the inh...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Nature Portfolio
2025-01-01
|
| Series: | Scientific Reports |
| Subjects: | |
| Online Access: | https://doi.org/10.1038/s41598-024-82759-z |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841544725928083456 |
|---|---|
| author | Qi Guo Qi Tan Jun Tang Benyun Shi |
| author_facet | Qi Guo Qi Tan Jun Tang Benyun Shi |
| author_sort | Qi Guo |
| collection | DOAJ |
| description | Abstract Intelligent transportation systems heavily rely on forecasting urban traffic flow, and a variety of approaches have been developed for this purpose. However, most current methods focus on exploring spatial and temporal dependencies in historical traffic data, while often overlooking the inherent spectral characteristics hidden in traffic time series. In this paper, we introduce an approach to analyzing traffic flow in the frequency domain. By integrating attention mechanisms, we comprehensively capture the hidden correlations among space, time, and frequency dimensions. By leveraging deep learning to capture spatial correlations in traffic flow and applying spectral analysis to fuse time series data with underlying periodic correlations in both the time and frequency domains, we develop an innovative traffic prediction model called the Space-Time-Frequency Attention Network (STFAN). The core of this network lies in the application of attention mechanisms, which project the hidden states of current traffic features across the space, time, and frequency domains onto future hidden states. This approach enables a comprehensive learning of the relationships between each dimension and the future states, ultimately allowing for accurate predictions of future traffic flow. We carry out experiments on two publicly available datasets from the California Department of Transportation, PeMS04 and PeMS08, to assess the performance of the proposed model. The results demonstrate that the proposed model outperforms existing baseline models in terms of predictive accuracy, particularly for mid- and long-term traffic flow forecasting. Finally, the ablation study confirmed that the frequency domain characteristics of traffic flow significantly influence future traffic conditions, demonstrating the practical effectiveness of the model. |
| format | Article |
| id | doaj-art-308d4f2788144ca8a6bf2f5eb360a1af |
| institution | Kabale University |
| issn | 2045-2322 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Nature Portfolio |
| record_format | Article |
| series | Scientific Reports |
| spelling | doaj-art-308d4f2788144ca8a6bf2f5eb360a1af2025-01-12T12:19:14ZengNature PortfolioScientific Reports2045-23222025-01-0115111410.1038/s41598-024-82759-zUnifying spatiotemporal and frequential attention for traffic predictionQi Guo0Qi Tan1Jun Tang2Benyun Shi3College of Computer and Information Engineering, Nanjing Tech UniversityCollege of Computer and Information Engineering, Nanjing Tech UniversityCollege of Computer and Information Engineering, Nanjing Tech UniversityCollege of Computer and Information Engineering, Nanjing Tech UniversityAbstract Intelligent transportation systems heavily rely on forecasting urban traffic flow, and a variety of approaches have been developed for this purpose. However, most current methods focus on exploring spatial and temporal dependencies in historical traffic data, while often overlooking the inherent spectral characteristics hidden in traffic time series. In this paper, we introduce an approach to analyzing traffic flow in the frequency domain. By integrating attention mechanisms, we comprehensively capture the hidden correlations among space, time, and frequency dimensions. By leveraging deep learning to capture spatial correlations in traffic flow and applying spectral analysis to fuse time series data with underlying periodic correlations in both the time and frequency domains, we develop an innovative traffic prediction model called the Space-Time-Frequency Attention Network (STFAN). The core of this network lies in the application of attention mechanisms, which project the hidden states of current traffic features across the space, time, and frequency domains onto future hidden states. This approach enables a comprehensive learning of the relationships between each dimension and the future states, ultimately allowing for accurate predictions of future traffic flow. We carry out experiments on two publicly available datasets from the California Department of Transportation, PeMS04 and PeMS08, to assess the performance of the proposed model. The results demonstrate that the proposed model outperforms existing baseline models in terms of predictive accuracy, particularly for mid- and long-term traffic flow forecasting. Finally, the ablation study confirmed that the frequency domain characteristics of traffic flow significantly influence future traffic conditions, demonstrating the practical effectiveness of the model.https://doi.org/10.1038/s41598-024-82759-zTraffic flow predictionGraph neural networksAttention mechanismTemporal-frequential attention |
| spellingShingle | Qi Guo Qi Tan Jun Tang Benyun Shi Unifying spatiotemporal and frequential attention for traffic prediction Scientific Reports Traffic flow prediction Graph neural networks Attention mechanism Temporal-frequential attention |
| title | Unifying spatiotemporal and frequential attention for traffic prediction |
| title_full | Unifying spatiotemporal and frequential attention for traffic prediction |
| title_fullStr | Unifying spatiotemporal and frequential attention for traffic prediction |
| title_full_unstemmed | Unifying spatiotemporal and frequential attention for traffic prediction |
| title_short | Unifying spatiotemporal and frequential attention for traffic prediction |
| title_sort | unifying spatiotemporal and frequential attention for traffic prediction |
| topic | Traffic flow prediction Graph neural networks Attention mechanism Temporal-frequential attention |
| url | https://doi.org/10.1038/s41598-024-82759-z |
| work_keys_str_mv | AT qiguo unifyingspatiotemporalandfrequentialattentionfortrafficprediction AT qitan unifyingspatiotemporalandfrequentialattentionfortrafficprediction AT juntang unifyingspatiotemporalandfrequentialattentionfortrafficprediction AT benyunshi unifyingspatiotemporalandfrequentialattentionfortrafficprediction |