arXiv:2507.09805cs.LGcs.AI2025-07中稿 · KDD被引 3

用图结构提升交通预测联邦学习效率,不增加计算负担。

Federated Learning with Graph-Based Aggregation for Traffic Forecasting

  • 基于图结构的轻量级聚合,利用邻居关系加权模型更新。
  • 在METR-LA和PEMS-BAY数据集上性能优于传统联邦平均方法。
  • 适合关注交通流预测与隐私保护结合的研究者。

在交通预测中,目标是利用各区域部署设备收集的历史数据,估算特定区域或路段的交通速度或流量。每个区域或路段可视为一个独立客户端,测量本地交通流,使联邦学习(FL)成为无需共享原始数据即可协同训练模型的合适方法。在集中式联邦学习中,中央服务器从多个客户端收集并聚合模型更新以构建共享模型,同时保护各客户端的数据隐私。标准联邦学习方法(如联邦平均,FedAvg)假设客户端之间相互独立,这在依赖空间关系的交通预测任务中可能限制性能。联邦图学习方法可在服务器端聚合过程中捕捉这些依赖关系,但通常引入显著计算开销。本文提出一种轻量级图感知的联邦学习方法,融合了FedAvg的简洁性与图学习的核心思想。该方法不训练完整模型,而是应用基本的邻域聚合原则来指导参数更新,根据图连接度对客户端模型进行加权。该方法有效捕捉空间关系,同时保持计算高效。我们在两个基准交通数据集METR-LA和PEMS-BAY上评估了该方法,结果表明其性能与标准基线及近期图基联邦学习技术相当。

原文摘要 · Abstract (English)

In traffic prediction, the goal is to estimate traffic speed or flow in specific regions or road segments using historical data collected by devices deployed in each area. Each region or road segment can be viewed as an individual client that measures local traffic flow, making Federated Learning (FL) a suitable approach for collaboratively training models without sharing raw data. In centralized FL, a central server collects and aggregates model updates from multiple clients to build a shared model while preserving each client's data privacy. Standard FL methods, such as Federated Averaging (FedAvg), assume that clients are independent, which can limit performance in traffic prediction tasks where spatial relationships between clients are important. Federated Graph Learning methods can capture these dependencies during server-side aggregation, but they often introduce significant computational overhead. In this paper, we propose a lightweight graph-aware FL approach that blends the simplicity of FedAvg with key ideas from graph learning. Rather than training full models, our method applies basic neighbourhood aggregation principles to guide parameter updates, weighting client models based on graph connectivity. This approach captures spatial relationships effectively while remaining computationally efficient. We evaluate our method on two benchmark traffic datasets, METR-LA and PEMS-BAY, and show that it achieves competitive performance compared to standard baselines and recent graph-based federated learning techniques.

交通预测联邦学习图神经网络

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