arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.05760cs.LGcs.AI

基于分层注意力LSTM(HierAttnLSTM)的网络级时空交通状态预测

Network Level Spatial Temporal Traffic State Forecasting with Hierarchical-Attention-LSTM (HierAttnLSTM)

University of Tennessee at Chattanooga(田纳西大学查塔努加分校)

Tianya Zhang

AI总结:

针对网络级交通状态预测需求,本文提出分层注意力LSTM(HierAttnLSTM)模型,整合不同层级LSTM状态并引入注意力机制,在PeMS数据集上优于基线模型,提升预测精度且能识别异常拥堵,数据代码开源。

摘要:

由无处不在的交通监测传感器收集的速度、流量和行程时间等交通状态数据,需要先进的网络级分析来预测和识别重要的交通模式。本文利用开放基准中加州交通局性能测量系统(PeMS)的多种交通状态数据集,与公认的时空模型相比取得了良好性能。受分层架构在各类人工智能(AI)任务中成功的启发,我们借鉴人类感知系统,通过注意力池化机制将低级到高级长短期记忆(LSTM)网络的细胞状态和隐藏状态进行整合。所开发的分层结构旨在处理不同时间尺度的依赖关系,捕捉网络级交通状态的时空相关性,实现对所有走廊而非单条链路或路线的交通状态预测。通过消融研究分析了所设计的基于注意力的LSTM的效率。与基线LSTM模型的对比结果表明,分层注意力LSTM(HierAttnLSTM)模型不仅预测精度更高,还能有效预测异常拥堵模式。数据和代码已公开,以支持可复现的科学研究。

英文摘要:

Traffic state data, such as speed, volume and travel time collected from ubiquitous traffic monitoring sensors require advanced network level analytics for forecasting and identifying significant traffic patterns. This paper leverages diverse traffic state datasets from the Caltrans Performance Measurement System (PeMS) hosted on the open benchmark and achieved promising performance compared to well recognized spatial-temporal models. Drawing inspiration from the success of hierarchical architectures in various Artificial Intelligence (AI) tasks, we integrate cell and hidden states from low-level to high-level Long Short-Term Memory (LSTM) networks with an attention pooling mechanism, similar to human perception systems. The developed hierarchical structure is designed to account for dependencies across different time scales, capturing the spatial-temporal correlations of network-level traffic states, enabling the prediction of traffic states for all corridors rather than a single link or route. The efficiency of designed attention-based LSTM is analyzed by ablation study. Comparative results with baseline LSTM models demonstrate that the Hierarchical Attention LSTM (HierAttnLSTM) model not only provides higher prediction accuracy but also effectively forecasts unusual congestion patterns. Data and code are made publicly available to support reproducible scientific research.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑