The prediction of traffic flow is a challenging yet crucial problem in spatial-temporal analysis, which has recently gained increasing interest. In addition to spatial-temporal correlations, the functionality of urban areas also plays a crucial role in traffic flow prediction. However, the exploration of regional functional attributes mainly focuses on adding additional topological structures, ignoring the influence of functional attributes on regional traffic patterns. Different from the existing works, we propose a novel module named POI-MetaBlock, which utilizes the functionality of each region (represented by Point of Interest distribution) as metadata to further mine different traffic characteristics in areas with different functions. Specifically, the proposed POI-MetaBlock employs a self-attention architecture and incorporates POI and time information to generate dynamic attention parameters for each region, which enables the model to fit different traffic patterns of various areas at different times. Furthermore, our lightweight POI-MetaBlock can be easily integrated into conventional traffic flow prediction models. Extensive experiments demonstrate that our module significantly improves the performance of traffic flow prediction and outperforms state-of-the-art methods that use metadata.
翻译:交通流预测是一项具有挑战性但又至关重要的时空分析问题,近年来越来越受到关注。除了时空相关性外,城市区域的功能性也在交通流预测中发挥着至关重要的作用。然而,对区域功能属性的探索主要集中在添加额外的拓扑结构上,忽略了功能属性对区域交通模式的影响。与现有的工作不同,我们提出了一个名为POI-MetaBlock的新模块,它利用每个区域的功能性(由兴趣点分布表示)作为元数据,进一步挖掘不同功能区域的不同交通特征。具体而言,所提出的POI-MetaBlock采用自我注意力结构,并结合POI和时间信息生成每个区域的动态注意力参数,使模型能够适应不同时间各个区域的不同交通模式。此外,我们的轻量级POI-MetaBlock可以轻松集成到常规交通流预测模型中。广泛的实验表明,我们的模块显著提高了交通流预测性能,并优于使用元数据的最先进方法。