Multi-Stage Federated Learning Mechanism with non-IID Data in Internet of Vehicles

被引:0
作者
Tang, Xiaolan [1 ]
Liang, Yuting [1 ]
Chen, Wenlong [1 ]
机构
[1] College of Information Engineering, Capital Normal University, Beijing
来源
Jisuanji Yanjiu yu Fazhan/Computer Research and Development | 2024年 / 61卷 / 09期
基金
中国国家自然科学基金;
关键词
federated learning (FL); Internet of vehicles (IoV); non-IID data; privacy preserving; transmission control;
D O I
10.7544/issn1000-1239.202330885
中图分类号
学科分类号
摘要
The Internet of vehicles (IoV) plays an indispensable role in the construction of smart cities, where cars are not just a means of transportation but also a crucial medium for information collection and transmission in the era of big data. With the rapid growth in the volume of data collected from vehicles and the increased awareness of privacy protection, ensuring users’ data security and preventing data breaches in IoV have become an urgent issue to address. Federated learning, as a “data-does-not-move, model-moves” approach, offers a feasible method for protecting user privacy while achieving excellent performance. However, because of the differences of devices, regions and individual habits, data collected from multiple vehicle typically exhibit non-independent and identically distributed (non-IID) characteristics. Traditional federated learning algorithms have slow model convergence when processing non-IID data. In response to this challenge, we propose a multi-stage federated learning algorithm with non-IID data in IoV, named FedWO. In Stage 1, FedWO utilizes the federated averaging algorithm to expedite the global model in reaching a basic level of accuracy. In Stage 2, FedWO employs weighted federated learning, where the weight of a vehicle in the global model is calculated based on its data characteristics. This aggregation results in an improved global model. Moreover, we design a transmission control strategy to reduce communication overhead caused by model transmission. The Stage 3 involves personalized computation, where each vehicle employs its own data for personalized learning, fine-tuning the local model to obtain a model more aligned with local data. We conducted experimental evaluations using a driving behavior dataset. The results demonstrate that, compared with traditional methods, FedWO preserves data privacy while improving the accuracy of algorithms in non-IID data scenarios. © 2024 Science Press. All rights reserved.
引用
收藏
页码:2170 / 2184
页数:14
相关论文
共 35 条
[11]  
Xiangjie Kong, Kailai Wang, Hou Mingliang, Et al., A federated learning-based license plate recognition scheme for 5G-enabled Internet of vehicles[J], IEEE Transactions on Industrial Informatics, 17, 12, pp. 8523-8530, (2021)
[12]  
Liang Feiyuan, Yang Qinglin, Liu Ruiqi, Et al., Semi-synchronous federated learning protocol with dynamic aggregation in Internet of vehicles, IEEE Transactions on Vehicular Technology, 71, 5, pp. 4677-4691, (2022)
[13]  
Lu Yunlong, Huang Xiaohong, Zhang Ke, Et al., Blockchain empowered asynchronous federated learning for secure data sharing in Internet of vehicles, IEEE Transactions on Vehicular Technology, 69, 4, pp. 4298-4311, (2020)
[14]  
Tang Xiaolan, Liang Yuting, Wang Guan, Et al., Assisted driving system based on federated reinforcement learning, Displays, 80, (2023)
[15]  
Parekh R, Patel N, Gupta R, Et al., Gefl: Gradient encryption-aided privacy preserved federated learning for autonomous vehicles[J], IEEE Access, 11, pp. 1825-1839, (2023)
[16]  
Qu Zhiguo, Tang Yang, Muhammad G, Et al., Privacy protection in intelligent vehicle networking: A novel federated learning algorithm based on information fusion, Information Fusion, 98, (2023)
[17]  
Zejun Li, Hao Wu, Yunlong Lu, Coalition based utility and efficiency optimization for multi-task federated learning in Internet of vehicles[J], Future Generation Computer Systems, 140, pp. 196-208, (2023)
[18]  
Jingyi He, Biyao Gong, Jiadi Yang, Et al., ASCFL: Accurate and speedy semi-supervised clustering federated learning[J], Tsinghua Science and Technology, 28, 5, (2023)
[19]  
Jiangang Shu, Tingting Yang, Xinying Liao, Et al., Clustered federated multitask learning on non-IID data with enhanced privacy[J], IEEE Internet of Things Journal, 10, 4, pp. 3453-3467, (2022)
[20]  
Pu Tian, Weixian Liao, Wei Yu, Et al., WSCC: A weight-similarity-based client clustering approach for non-IID federated learning[J], IEEE Internet of Things Journal, 9, 20, pp. 20243-20256, (2022)