用联邦学习构建城市动态图谱,提升交通预测精度
Fed-LDR: Federated Local Data-infused Graph Creation with Node-centric Model Refinement
- 通过动态调整城市节点连接关系,捕捉实时空间变化
- 在两个真实数据集上实现最低误差,较最优基线降81%误差
- 适合研究城市智能交通与隐私保护的算法开发者
全球城市化加速带来了提升城市基础设施与服务的新挑战。时空数据作为理解城市现象和推动可持续发展的关键工具,日益重要。在此背景下,联邦学习(FL)因其符合城市物联网环境的隐私要求而受到关注。然而,将传统与深度学习模型融入联邦框架面临显著挑战,尤其在捕捉复杂时空依赖性和适应多样城市条件方面。为此,本文提出联邦本地数据注入图构建与节点中心模型优化算法(Fed-LDR)。该算法结合联邦学习与图卷积网络(GCN),包含两个核心模块:(1) 本地数据注入图构建(LDIGC)模块,动态重构邻接矩阵以反映城市环境中不断演化的空间关系;(2) 节点中心模型优化(NoMoR)模块,为各城市节点定制模型参数以应对异构性。在PeMSD4和PeMSD8数据集上的评估表明,Fed-LDR优于六种基线方法。其在两数据集上的平均绝对误差(MAE)分别为20.15和17.30,均方根误差(RMSE)分别为32.30和27.15,相关系数高达0.96。值得注意的是,在PeMSD4数据集上,相比最优基线方法FedMedian,MAE和RMSE分别降低81%和78%。
原文摘要 · Abstract (English)
The rapid acceleration of global urbanization has introduced novel challenges in enhancing urban infrastructure and services. Spatio-temporal data, integrating spatial and temporal dimensions, has emerged as a critical tool for understanding urban phenomena and promoting sustainability. In this context, Federated Learning (FL) has gained prominence as a distributed learning paradigm aligned with the privacy requirements of urban IoT environments. However, integrating traditional and deep learning models into the FL framework poses significant challenges, particularly in capturing complex spatio-temporal dependencies and adapting to diverse urban conditions. To address these challenges, we propose the Federated Local Data-Infused Graph Creation with Node-centric Model Refinement (Fed-LDR) algorithm. Fed-LDR leverages FL and Graph Convolutional Networks (GCN) to enhance spatio-temporal data analysis in urban environments. The algorithm comprises two key modules: (1) the Local Data-Infused Graph Creation (LDIGC) module, which dynamically reconfigures adjacency matrices to reflect evolving spatial relationships within urban environments, and (2) the Node-centric Model Refinement (NoMoR) module, which customizes model parameters for individual urban nodes to accommodate heterogeneity. Evaluations on the PeMSD4 and PeMSD8 datasets demonstrate Fed-LDR's superior performance over six baseline methods. Fed-LDR achieved the lowest Mean Absolute Error (MAE) values of 20.15 and 17.30, and the lowest Root Mean Square Error (RMSE) values of 32.30 and 27.15, respectively, while maintaining a high correlation coefficient of 0.96 across both datasets. Notably, on the PeMSD4 dataset, Fed-LDR reduced MAE and RMSE by up to 81\% and 78\%, respectively, compared to the best-performing baseline FedMedian.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。