解决交通流预测中跨客户端动态空间依赖建模难题
Unlocking Dynamic Inter-Client Spatial Dependencies: A Federated Spatio-Temporal Graph Learning Method for Traffic Flow Forecasting
- 提出联邦图学习框架FedSTGD,动态建模跨客户端空间关系
- 在四个真实数据集上,RMSE/MAE/MAPE均优于现有方法
- 适合关注隐私保护下交通预测的工程师与研究者
时空图是建模交通时间序列复杂依赖关系的强大工具。然而,现实世界中交通数据分散在多个利益相关方,如何在遵守数据本地性约束的前提下建模和重构跨客户端的空间依赖关系,仍面临挑战。现有方法多关注静态依赖,忽视其动态特性,导致性能不佳。为此,我们提出联邦时空图动态跨客户端依赖建模方法(FedSTGD),该框架通过联邦非线性计算分解模块近似复杂图操作,并引入节点嵌入增强模块缓解分解带来的性能下降。客户端与服务器通过协同学习协议,将动态跨客户端空间依赖学习任务分解为轻量、可并行的子任务。在四个真实世界数据集上的大量实验表明,FedSTGD在RMSE、MAE和MAPE指标上均优于当前最优基线,接近集中式基线表现。消融实验证实各模块对动态跨客户端依赖建模的有效性,敏感性分析显示其对超参数变化具有鲁棒性。
原文摘要 · Abstract (English)
Spatio-temporal graphs are powerful tools for modeling complex dependencies in traffic time series. However, the distributed nature of real-world traffic data across multiple stakeholders poses significant challenges in modeling and reconstructing inter-client spatial dependencies while adhering to data locality constraints. Existing methods primarily address static dependencies, overlooking their dynamic nature and resulting in suboptimal performance. In response, we propose Federated Spatio-Temporal Graph with Dynamic Inter-Client Dependencies (FedSTGD), a framework designed to model and reconstruct dynamic inter-client spatial dependencies in federated learning. FedSTGD incorporates a federated nonlinear computation decomposition module to approximate complex graph operations. This is complemented by a graph node embedding augmentation module, which alleviates performance degradation arising from the decomposition. These modules are coordinated through a client-server collective learning protocol, which decomposes dynamic inter-client spatial dependency learning tasks into lightweight, parallelizable subtasks. Extensive experiments on four real-world datasets demonstrate that FedSTGD achieves superior performance over state-of-the-art baselines in terms of RMSE, MAE, and MAPE, approaching that of centralized baselines. Ablation studies confirm the contribution of each module in addressing dynamic inter-client spatial dependencies, while sensitivity analysis highlights the robustness of FedSTGD to variations in hyperparameters.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。