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)生成带有不确定性量化的概率预测,以辅助下游任务的风险管理与决策。基于真实数据集的广泛实验表明,我们的方法在预测任务中取得了显著成果,且不确定性量化结果与有无历史数据的位置高度相关。我们还展示了该模型可帮助支持交通领域的传感器部署任务,在有限的传感器部署预算下实现更高精度。