Traffic prediction is a crucial topic because of its broad scope of applications in the transportation domain. Recently, various studies have achieved promising results. However, most studies assume the prediction locations have complete or at least partial historical records and cannot be extended to non-historical recorded locations. In real-life scenarios, the deployment of sensors could be limited due to budget limitations and installation availability, which makes most current models not applicable. Though few pieces of literature tried to impute traffic states at the missing locations, these methods need the data simultaneously observed at the locations with sensors, making them not applicable to prediction tasks. Another drawback is the lack of measurement of uncertainty in prediction, making prior works unsuitable for risk-sensitive tasks or involving decision-making. To fill the gap, inspired by the previous inductive graph neural network, this work proposed an uncertainty-aware framework with the ability to 1) extend prediction to missing locations with no historical records and significantly extend spatial coverage of prediction locations while reducing deployment of sensors and 2) generate probabilistic prediction with uncertainty quantification to help the management of risk and decision making in the down-stream tasks. Through extensive experiments on real-life datasets, the result shows our method achieved promising results on prediction tasks, and the uncertainty quantification gives consistent results which highly correlated with the locations with and without historical data. We also show that our model could help support sensor deployment tasks in the transportation field to achieve higher accuracy with a limited sensor deployment budget.
翻译:交通预测因其在交通领域的广泛应用而成为关键课题。近年来,多项研究已取得令人瞩目的成果。然而,大多数研究假设预测位置具有完整或至少部分历史记录,无法推广到无历史记录的位置。在实际场景中,由于预算限制和安装条件限制,传感器的部署可能受限,导致现有模型难以适用。尽管少数文献尝试对缺失位置的交通状态进行插补,但这些方法需要同时利用传感器所在位置的观测数据,因此不适用于预测任务。另一个缺陷是预测中缺乏不确定性度量,使得先前的工作不适用于风险敏感任务或需要决策支持的情况。为弥补这一空白,受先前归纳图神经网络的启发,本文提出了一种不确定性感知框架,具备以下能力:1)将预测扩展到无历史记录的缺失位置,显著扩大预测位置的空间覆盖范围,同时减少传感器部署;2)生成带有不确定性量化的概率预测,帮助下游任务中的风险管理和决策制定。通过在真实数据集上的广泛实验,结果表明我们的方法在预测任务上取得了令人满意的成果,且不确定性量化结果与有无历史数据的位置高度相关。我们还证明,该模型可帮助支持交通领域的传感器部署任务,在有限传感器部署预算下实现更高精度。