Journal
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS
Volume 24, Issue 5, Pages 5516-5526Publisher
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/TITS.2023.3243913
Keywords
Traffic flow prediction; hidden state extraction; spatio-temporal autoencoder
Ask authors/readers for more resources
Forecasting traffic flow is crucial in urban areas, and a novel method called Spatio-Temporal AutoEncoder (ST-AE) is proposed to more effectively incorporate various intrinsic patterns in real-world traffic flows. The method learns the patterns from traffic flow data and predicts future traffic flows by projecting the current hidden states to the future hidden states.
Forecasting traffic flow is an important task in urban areas, and a large number of methods have been proposed for traffic flow prediction. However, most of the existing methods follow a general technical route to aggregate historical information spatially and temporally. In this paper, we propose a different approach for traffic flow prediction. Our major motivation is to more effectively incorporate various intrinsic patterns in real-world traffic flows, such as fixed spatial distributions, topological correlations, and temporal periodicity. Along this line, we propose a novel autoencoder-based traffic flow prediction method, named Spatio-Temporal AutoEncoder (ST-AE). The core of our method is an autoencoder specially designed to learn the intrinsic patterns from traffic flow data, and encode the current traffic flow information into a low-dimensional representation. The prediction is made by simply projecting the current hidden states to the future hidden states, and then reconstructing the future traffic flows with the trained autoencoder. We have conducted extensive experiments on four real-world data sets. Our method outperforms existing methods in several settings, particularly for long-term traffic flow prediction.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available