基于谱域超图卷积网络的交通流预测模型
Traffic Flow Prediction Model Based on Spectral Hypergraph Convolutional Network
-
摘要: 针对传统图结构难以对节点间的隐含复杂关联关系建模的问题, 利用超图对交通流数据进行高阶表示, 提出基于谱域超图卷积网络的交通流预测方法。首先, 通过动态超边刻画数据特征层面的关系, 利用谱域超图卷积, 包括基于傅里叶和图小波的超图卷积及门控时序卷积, 在多尺度上提取交通流的时空特征, 实现端到端的节点级交通流预测。然后, 采用北京市以及美国加利福尼亚州真实历史数据集进行预测实验。消融实验通过孤立和重构网络模型验证了所提方法的有效性。全时段和早高峰交通流预测的实验结果表明, 该方法预测准确率高于目前主流交通流预测模型。Abstract: The traditional graph structure ignores the implicit complex relationship between nodes to a certain extent. Aiming at the problem, hypergraph was used to represent traffic flow data at a high level, and a traffic flow prediction method was proposed based on hypergraph convolutional network in spectral domain. First, spectral domain hypergraph convolution and gated temporal convolution were used to extract the spatiotemporal characteristics of traffic flow at multiple scales by describing the relationship at the data feature level through dynamic hyperedges, and end-to-end node-level traffic prediction was realized. Afterward, the real historical data sets of Beijing and California were used to conduct prediction experiments. The ablation experiments verify the effectiveness of the proposed method by isolating and reconstructing the network model; the full-time and morning peak traffic flow prediction experiments show that the prediction accuracy of the proposed method is higher than that of the current mainstream traffic forecasting models.