Transformer model-based multi-scale fine-grained identification and classification of regional traffic states

Abstract

To address the limitations in precision of conventional traffic state estimation methods, this article introduces a novel approach based on the Transformer model for traffic state identification and classification. Traditional methods commonly categorize traffic states into four or six classes; however, they often fail to accurately capture the nuanced transitions in traffic states before and after the implementation of traffic congestion reduction strategies. Many traffic congestion reduction strategies can alleviate congestion, but they often fail to effectively transition the traffic state from a congested condition to a free-flowing one. To address this issue, we propose a classification framework that divides traffic states into sixteen distinct categories. We design a Transformer model to extract features from traffic data. The k-means algorithm is then applied to these features to group similar traffic states. The resulting clusters are ranked by congestion level using non-dominated sorting, thereby dividing the data into 16 levels, from Level 1 (free-flowing) to Level 16 (congested). Extensive experiments are conducted using a large-scale simulated traffic dataset. The results demonstrate significant advancements in traffic state estimation achieved by our Transformer-based approach. Compared to baseline methods, our model exhibits marked improvements in both clustering quality and generalization capabilities.