Two-Layer Federated Learning With Heterogeneous Model Aggregation for 6G Supported Internet of Vehicles
Top Cited Papers
- 6 May 2021
- journal article
- research article
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Vehicular Technology
- Vol. 70 (6), 5308-5317
- https://doi.org/10.1109/tvt.2021.3077893
Abstract
The vision of the upcoming 6G technologies that have fast data rate, low latency, and ultra-dense network, draws great attentions to the Internet of Vehicles (IoV) and Vehicle-to-Everything (V2X) communication for intelligent transportation systems. There is an urgent need for distributed machine learning techniques that can take advantages of massive interconnected networks with explosive amount of heterogeneous data generated at the network edge. In this study, a two-layer federated learning model is proposed to take advantages of the distributed end-edge-cloud architecture typical in 6G environment, and to achieve a more efficient and more accurate learning while ensuring data privacy protection and reducing communication overheads. A novel multi-layer heterogeneous model selection and aggregation scheme is designed as a part of the federated learning process to better utilize the local and global contexts of individual vehicles and road side units (RSUs) in 6G supported vehicular networks. This context-aware distributed learning mechanism is then developed and applied to address intelligent object detection, which is one of the most critical challenges in modern intelligent transportation systems with autonomous vehicles. Evaluation results showed that the proposed method, which demonstrates a higher learning accuracy with better precision, recall and F1 score, outperforms other state-of-the-art methods under 6G network configuration by achieving faster convergence, and scales better with larger numbers of RSUs involved in the learning process.Keywords
Funding Information
- National Natural Science Foundation of China (72088101, 72091515, 62072171)
- National Key R&D Program of China (2017YFE0117500, 2019YFE0190500, 2019GK1010)
- Natural Science Foundation of Hunan Province (2019JJ40150)
This publication has 34 references indexed in Scilit:
- Online Multi-Workflow Scheduling under Uncertain Task Execution Time in IaaS CloudsIEEE Transactions on Cloud Computing, 2019
- Federated Machine LearningACM Transactions on Intelligent Systems and Technology, 2019
- Edge Preserving and Multi-Scale Contextual Neural Network for Salient Object DetectionIEEE Transactions on Image Processing, 2017
- A Roadmap to Safer Driving Through Advanced Driver Assistance SystemsAuto Tech Review, 2016
- Vehicular Ad Hoc Networks: Architectures, Research Issues, Methodologies, Challenges, and TrendsInternational Journal of Distributed Sensor Networks, 2015
- A Security and Privacy Review of VANETsIEEE Transactions on Intelligent Transportation Systems, 2015
- Multi-view traffic sign detection, recognition, and 3D localisationMachine Vision and Applications, 2011
- Vehicle-to-vehicle wireless communication protocols for enhancing highway traffic safetyIEEE Communications Magazine, 2006
- Multiagent traffic managementPublished by Association for Computing Machinery (ACM) ,2005
- Time-stable geocast for ad hoc networks and its application with virtual warning signsComputer Communications, 2004