Loading [MathJax]/extensions/TeX/ieee_stixext.js
MS-Net: Multi-Source Spatio-Temporal Network for Traffic Flow Prediction | IEEE Journals & Magazine | IEEE Xplore

MS-Net: Multi-Source Spatio-Temporal Network for Traffic Flow Prediction


Abstract:

Predicting urban traffic flow is a challenging task, due to the complicated spatio-temporal dependencies on traffic networks. Urban traffic flow usually has both short-te...Show More

Abstract:

Predicting urban traffic flow is a challenging task, due to the complicated spatio-temporal dependencies on traffic networks. Urban traffic flow usually has both short-term neighboring and long-term periodic temporal dependencies. It is also noticed that the spatial correlations over different traffic nodes are both local and non-local. What’s more, the traffic flow is affected by various external factors. To capture the non-local spatial correlations, we propose a Dilated Attentional Graph Convolution (DAGC). The DAGC utilizes a dilated graph convolution kernel to expand the nodes’ receptive field and exploit multi-order neighborhood. Technically, the lower-order neighborhood corresponds to local spatial dependencies, while the higher-order neighborhood corresponds to non-local spatial dependencies between nodes. Based on DAGC, a Multi-Source Spatio-Temporal Network (MS-Net) is designed, which suffices to integrate long-range historical traffic data as well as multi-modal external information. MS-Net consists of four components: a spatial feature extraction module, a temporal feature fusion module, an external factors embedding module, and a multi-source data fusion module. Extensive experiments on three real traffic datasets demonstrates that the proposed model performs well on both the public transportation networks, road networks, and can handle large-scale traffic networks in particular the Beijing bus network which has more than 4,000 traffic nodes.
Published in: IEEE Transactions on Intelligent Transportation Systems ( Volume: 23, Issue: 7, July 2022)
Page(s): 7142 - 7155
Date of Publication: 24 March 2021

ISSN Information:

Funding Agency:


References

References is not available for this document.