摘要
实时交通预测对于减少出行时间至关重要。为了预测交通状况,我们采用时空图神经网络(ST-GNN)将实时交通数据建模为时序图。尽管ST-GNN具有强大的功能,但在处理现实世界交通数据时,它往往难以实现高效的实时预测。鉴于实时数据的动态性,及时预测至关重要,因此我们采用知识蒸馏(KD)作为解决方案,以提高ST-GNN在交通预测中的执行时间。在本文中,我们引入了一种成本函数,旨在使用来自复杂网络(教师)的蒸馏数据来训练一个参数更少的网络(学生),同时保持其精度接近教师的精度。我们使用知识蒸馏,将教师网络中的时空相关性纳入其中,使学生能够学习教师感知的复杂模式。然而,一个挑战在于如何确定学生网络结构,而不是将其随意考虑。为了解决这一挑战,我们提出了一种算法,利用成本函数来计算剪枝分数,解决小型网络结构搜索问题,并使用KD联合微调每个剪枝阶段产生的网络。最终,我们在两个现实世界数据集PeMSD7和PeMSD8上评估了我们提出的想法。结果表明,即使仅保留3%的网络参数,我们的方法也能使学生的精度保持接近教师的精度。