| Title | FedSTN: Graph Representation Driven Federated Learning for Edge Computing Enabled Urban Traffic Flow Prediction |
|---|---|
| ID_Doc | 42714 |
| Authors | Yuan, XM; Chen, JH; Yang, JY; Zhang, N; Yang, TT; Han, T; Taherkordi, A |
| Title | FedSTN: Graph Representation Driven Federated Learning for Edge Computing Enabled Urban Traffic Flow Prediction |
| Year | 2023 |
| Published | Ieee Transactions On Intelligent Transportation Systems, 24, 8 |
| Abstract | Predicting traffic flow plays an important role in reducing traffic congestion and improving transportation efficiency for smart cities. Traffic Flow Prediction (TFP) in the smart city requires efficient models, highly reliable networks, and data privacy. As traffic data, traffic trajectory can be transformed into a graph representation, so as to mine the spatio-temporal information of the graph for TFP. However, most existing work adopt a central training mode where the privacy problem brought by the distributed traffic data is not considered. In this paper, we propose a Federated Deep Learning based on the Spatial-Temporal Long and Short-Term Networks (FedSTN) algorithm to predict traffic flow by utilizing observed historical traffic data. In FedSTN, each local TFP model deployed in an edge computing server includes three main components, namely Recurrent Long-term Capture Network (RLCN) module, Attentive Mechanism Federated Network (AMFN) module, and Semantic Capture Network (SCN) module. RLCN can capture the long-term spatial-temporal information in each area. AMFN shares short-term spatio-temporal hidden information when it trains its local TFP model by the additive homomorphic encryption approach based on Vertical Federated Learning (VFL). We employ SCN to capture semantic features such as irregular non-Euclidean connections and Point of Interest (POI). Compared with existing baselines, several simulations are conducted on practical data sets and the results prove the effectiveness of our algorithm. |
Similar Articles
| ID | Score | Article |
|---|---|---|
36339
|
Djenouri, Y; Michalak, TP; Lin, JCW Federated deep learning for smart city edge-based applications(2023) | |
| 37439
|
Wang, Z; Hu, J; Min, GY; Zhao, ZW; Wang, J Data-Augmentation-Based Cellular Traffic Prediction in Edge-Computing-Enabled Smart City(2021)Ieee Transactions On Industrial Informatics, 17.0, 6 | |
| 42056
|
Valente, R; Senna, C; Rito, P; Sargento, S Embedded Federated Learning for VANET Environments(2023)Applied Sciences-Basel, 13, 4 | |
| 36511
|
Zheng, G; Chai, WK; Katos, V An Ensemble Model for Short-Term Traffic Prediction in Smart City Transportation System(2019) | |
| 37296
|
Zhou, SH; Wei, CF; Song, CF; Pan, X; Chang, WB; Yang, LC Short-Term Traffic Flow Prediction of the Smart City Using 5G Internet of Vehicles Based on Edge Computing(2023)Ieee Transactions On Intelligent Transportation Systems, 24.0, 2 | |
| 39021
|
Wang, F; Li, GS; Wang, YL; Rafique, W; Khosravi, MR; Liu, GF; Liu, YW; Qi, LY Privacy-Aware Traffic Flow Prediction Based on Multi-Party Sensor Data with Zero Trust in Smart City(2023)Acm Transactions On Internet Technology, 23, 3 | |
| 40977
|
Wang, JQ; Yang, SQ; Gao, Y; Wang, J; Alfarraj, O Road Network Traffic Flow Prediction Method Based on Graph Attention Networks(2024) |