• • 下一篇
李晓繁,苏美红*,荀亚玲,张继福,马洋
摘要: 图卷积循环网络(GCRN)作为交通流量预测中的关键技术之一,可有效提升交通流量预测性能,但由于交通数据在时空维度上表现出强烈的异质性,现有的GCRN通常为每个节点或时间步独立生成参数,导致模型参数庞大且冗余度高。为缓解上述问题,并且充分利用元节点减少参数冗余的优势,以及超网络适配时空异质性、平衡性能与效率的特性,构建基于自适应元节点学习的图卷积循环网络(Adaptive Meta-node Graph Convolutional Recurrent Network,AMGCRN)。该模型首先通过引入少量可学习的元节点以凝练差异化交通模式,并利用超网络动态生成各节点在各时间步的专属卷积参数,不仅可以保证时空异质性,而且有效避免参数冗余;其次,通过多头交叉注意力机制,确保对全局时间依赖的捕捉能力。在多个真实交通数据集上的实验结果表明,相较于元注意力图卷积循环网络(MAGCRN),所提模型在平均绝对误差(MAE)指标上的相对降幅约为1%~3%。进一步地,在PeMSD8数据集上的效率评估中,所提模型的参数量相较于MAGCRN减少了83%,FLOPs降低了22%,实现了模型容量、计算复杂度与预测性能的综合优化。
中图分类号: