AutoFed无需调参,用自适应提示实现个性化交通预测。
AutoFed: Personalized Federated Traffic Prediction via Adaptive Prompt
- 基于提示学习设计联邦表示器,自动生成共享提示矩阵。
- 在多个真实数据集上优于现有方法,跨场景表现稳定。
- 适合注重隐私的交通系统部署,无需额外超参优化。
精准的交通预测对智能交通系统(如网约车调度、城市道路规划、车队管理)至关重要。然而,交通数据涉及隐私,多数方法依赖本地训练,导致数据孤岛和知识难以共享。联邦学习(FL)虽能实现隐私保护下的协作训练,但标准FL难以应对客户端间非独立同分布(non-IID)问题。为此,个性化联邦学习(PFL)成为新方向。但现有PFL框架在交通预测中仍需专门的图特征工程、数据处理与网络设计,且多数依赖跨数据集的超参数优化——这在真实场景中往往不可得,限制了实际应用。为此,我们提出AutoFed,一种无需人工调参的新型个性化联邦交通预测框架。受提示学习启发,AutoFed引入联邦表示器,通过客户端对齐的适配器将本地数据提炼为紧凑的全局共享提示矩阵,该提示再引导个性化预测器,使各客户端既能受益于跨客户端知识,又保留本地特性。在多个真实世界数据集上的实验表明,AutoFed在多样化场景中持续表现更优。代码已开源:https://github.com/RS2002/AutoFed。
原文摘要 · Abstract (English)
Accurate traffic prediction is essential for Intelligent Transportation Systems, including ride-hailing, urban road planning, and vehicle fleet management. However, due to significant privacy concerns surrounding traffic data, most existing methods rely on local training, resulting in data silos and limited knowledge sharing. Federated Learning (FL) offers an efficient solution through privacy-preserving collaborative training; however, standard FL struggles with the non-independent and identically distributed (non-IID) problem among clients. This challenge has led to the emergence of Personalized Federated Learning (PFL) as a promising paradigm. Nevertheless, current PFL frameworks require further adaptation for traffic prediction tasks, such as specialized graph feature engineering, data processing, and network architecture design. A notable limitation of many prior studies is their reliance on hyper-parameter optimization across datasets-information that is often unavailable in real-world scenarios-thus impeding practical deployment. To address this challenge, we propose AutoFed, a novel PFL framework for traffic prediction that eliminates the need for manual hyper-parameter tuning. Inspired by prompt learning, AutoFed introduces a federated representor that employs a client-aligned adapter to distill local data into a compact, globally shared prompt matrix. This prompt then conditions a personalized predictor, allowing each client to benefit from cross-client knowledge while maintaining local specificity. Extensive experiments on real-world datasets demonstrate that AutoFed consistently achieves superior performance across diverse scenarios. The code of this paper is provided at https://github.com/RS2002/AutoFed .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。