arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.08658cs.LGcs.DC
收藏

基于时空相关性的在线联邦学习交通流预测

Online Spatio-Temporal Correlation-Based Federated Learning for Traffic Flow Forecasting

Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);University of Chinese Academy of Sciences(中国科学院大学);Zhongguancun Laboratory(中关村实验室);School of Computer and Information Technology, Beijing Jiaotong University(北京交通大学计算机与信息技术学院)

Qingxiang Liu, Sheng Sun, Min Liu, Yuwei Wang, Bo Gao

摘要:

交通流预测对智能交通系统的建设具有重要意义。为了减轻通信负担并解决集中式预测方法带来的隐私泄露问题,联邦学习(FL)被应用于TFF。然而,现有的基于FL的方法采用批量学习方式,这使得预先训练的模型不适用于后续的交通数据,从而表现出低于标准的预测性能。本文首先研究了FL框架下采用在线学习(Online Learning,OL)方式的交通流预测方法,然后提出了一种新的预测方法--基于时空相关的在线联邦联邦学习(Online Spatio-Temporal Correlation Based,Federated Learning,FedOSTC),旨在保证在不受流量波动影响的情况下获得性能增益。具体而言,客户端采用基于门控递归单元(GRU)的编码器来获得流量数据序列内的内部时间模式。然后,中央服务器通过图注意力网络(GAT)评估客户端之间的空间相关性,以适应交通波动引起的空间接近度的动态变化。此外,为了提高即将到来的流量数据的全局模型的泛化能力,提出了一个周期感知的聚合机制,聚合使用在线梯度下降(OGD)算法优化的本地模型在客户端。在两个真实数据集上进行了综合实验,验证了该方法的有效性,数值结果表明了FedOSTC的优越性.

英文摘要:

Traffic flow forecasting (TFF) is of great importance to the construction of Intelligent Transportation Systems (ITS). To mitigate communication burden and tackle with the problem of privacy leakage aroused by centralized forecasting methods, Federated Learning (FL) has been applied to TFF. However, existing FL-based approaches employ batch learning manner, which makes the pre-trained models inapplicable to subsequent traffic data, thus exhibiting subpar prediction performance. In this paper, we perform the first study of forecasting traffic flow adopting Online Learning (OL) manner in FL framework and then propose a novel prediction method named Online Spatio-Temporal Correlation-based Federated Learning (FedOSTC), aiming to guarantee performance gains regardless of traffic fluctuation. Specifically, clients employ Gated Recurrent Unit (GRU)-based encoders to obtain the internal temporal patterns inside traffic data sequences. Then, the central server evaluates spatial correlation among clients via Graph Attention Network (GAT), catering to the dynamic changes of spatial closeness caused by traffic fluctuation. Furthermore, to improve the generalization of the global model for upcoming traffic data, a period-aware aggregation mechanism is proposed to aggregate the local models which are optimized using Online Gradient Descent (OGD) algorithm at clients. We perform comprehensive experiments on two real-world datasets to validate the efficiency and effectiveness of our proposed method and the numerical results demonstrate the superiority of FedOSTC.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑