arXiv:2411.14046cs.LG2024-11被引 9

提出轻量级联邦在线学习框架,提升交通流预测精度并降低通信计算开销。

REFOL: Resource-Efficient Federated Online Learning for Traffic Flow Forecasting

  • 基于数据驱动机制动态决定客户端参与,实时检测概念漂移。
  • 自适应在线优化策略减少无效更新,保障预测性能。
  • 利用图卷积评估空间相关性贡献,零额外开销实现高效聚合。

针对交通流预测中集中式方法带来的隐私泄露与高传输成本问题,已有联邦学习方法多采用离线学习,但在概念漂移(历史与未来数据分布变化)情况下表现不佳。在线学习可实时检测漂移,更适用于交通流预测,但现有联邦在线学习方法在应对概念漂移时效率低下,且带来巨大计算与通信开销。为此,本文提出资源高效的联邦在线学习框架REFOL,确保预测性能的同时实现轻量通信与高效计算。设计数据驱动的客户端参与机制以检测概念漂移并判断参与必要性;提出自适应在线优化策略,保证预测性能并避免无意义模型更新;设计基于图卷积的模型聚合机制,通过空间相关性评估参与者贡献,无需增加客户端的通信与计算负担。在真实数据集上进行了大量实验,验证了REFOL在预测精度提升与资源节约方面的显著优势。

原文摘要 · Abstract (English)

Multiple federated learning (FL) methods are proposed for traffic flow forecasting (TFF) to avoid heavy-transmission and privacy-leaking concerns resulting from the disclosure of raw data in centralized methods. However, these FL methods adopt offline learning which may yield subpar performance, when concept drift occurs, i.e., distributions of historical and future data vary. Online learning can detect concept drift during model training, thus more applicable to TFF. Nevertheless, the existing federated online learning method for TFF fails to efficiently solve the concept drift problem and causes tremendous computing and communication overhead. Therefore, we propose a novel method named Resource-Efficient Federated Online Learning (REFOL) for TFF, which guarantees prediction performance in a communication-lightweight and computation-efficient way. Specifically, we design a data-driven client participation mechanism to detect the occurrence of concept drift and determine clients' participation necessity. Subsequently, we propose an adaptive online optimization strategy, which guarantees prediction performance and meanwhile avoids meaningless model updates. Then, a graph convolution-based model aggregation mechanism is designed, aiming to assess participants' contribution based on spatial correlation without importing extra communication and computing consumption on clients. Finally, we conduct extensive experiments on real-world datasets to demonstrate the superiority of REFOL in terms of prediction improvement and resource economization.

联邦学习交通预测在线学习资源优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。