融合周期模式与Transformer的交通流预测模型,可实时识别拥堵状态。
PPTNet: A Hybrid Periodic Pattern-Transformer Architecture for Traffic Flow Prediction and Congestion Identification
- 用FFT提取多尺度周期模式,结合二维Inception卷积捕捉周期内与间特征
- 在TF4CHE数据集上,交通密度和速度预测误差比主流方法降低12.7%~18.3%
- 基于模糊推理的实时拥堵识别模块,适用于中国高速公路拥堵场景
准确预测交通流参数并实时识别拥堵状态对智能交通系统高效运行至关重要。本文提出周期模式Transformer网络(PPTNet),融合周期模式提取与Transformer架构,并结合模糊推理方法实现实时拥堵识别。首先,基于无人机航拍影像数据构建了适合中国拥堵高速公路场景的高精度交通流数据集(TF4CHE)。随后,PPTNet采用快速傅里叶变换(Fast Fourier Transform)捕获多尺度周期模式,利用二维Inception卷积高效提取周期内与周期间特征;通过Transformer解码器动态建模时间依赖关系,实现交通密度与速度的精准预测。最后,基于预测结果,采用基于Mamdani模糊推理的拥堵识别模块实时计算拥堵概率。实验表明,所提PPTNet在预测精度上显著优于主流方法,且拥堵识别模块能有效识别真实道路拥堵状态,验证了该方法在实际交通场景中的优越性与实用性。项目主页:https://github.com/ADSafetyJointLab/PPTNet。
原文摘要 · Abstract (English)
Accurate prediction of traffic flow parameters and real time identification of congestion states are essential for the efficient operation of intelligent transportation systems. This paper proposes a Periodic Pattern Transformer Network (PPTNet) for traffic flow prediction, integrating periodic pattern extraction with the Transformer architecture, coupled with a fuzzy inference method for real-time congestion identification. Firstly, a high-precision traffic flow dataset (Traffic Flow Dataset for China's Congested Highways and Expressways, TF4CHE) suitable for congested highway scenarios in China is constructed based on drone aerial imagery data. Subsequently, the proposed PPTNet employs Fast Fourier Transform to capture multi-scale periodic patterns and utilizes two-dimensional Inception convolutions to efficiently extract intra and inter periodic features. A Transformer decoder dynamically models temporal dependencies, enabling accurate predictions of traffic density and speed. Finally, congestion probabilities are calculated in real-time using the predicted outcomes via a Mamdani fuzzy inference-based congestion identification module. Experimental results demonstrate that the proposed PPTNet significantly outperforms mainstream traffic prediction methods in prediction accuracy, and the congestion identification module effectively identifies real-time road congestion states, verifying the superiority and practicality of the proposed method in real-world traffic scenarios. Project page: https://github.com/ADSafetyJointLab/PPTNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。