Short-term freeway traffic speed multistep prediction using an iTransformer model

Yajie Zou,Yubin Chen,Yajiao Xu,Hao Zhang,Siyang Zhang
DOI: https://doi.org/10.1016/j.physa.2024.130185
IF: 3.778
2024-11-08
Physica A Statistical Mechanics and its Applications
Abstract:Accurate prediction of the freeway traffic speed is important for enhancing intelligent transportation management and assisting with route planning. Current traffic speed prediction studies usually neglect predictions extending to an hour or longer. Thus, to address this gap, the inverted Transformer (iTransformer) model is utilized to predict freeway speeds across intervals from 5 to 150 minutes. The iTransformer model provides a global view by encapsulating the entire time series from each detector into variate tokens, allowing it to capture the complex patterns and dependencies that change over time. Additionally, a multi-head attention mechanism is employed to identify short-term and long-term speed patterns. This study validates the prediction performance of iTransformer by using the traffic speed data from an interstate freeway in Minnesota, comparing it against traditional traffic prediction methods and two other Transformer models. Results indicate that the iTransformer model outperforms these benchmark approaches, particularly when predicting speed over 60 minutes for peak hour periods.
physics, multidisciplinary
What problem does this paper attempt to address?