中国航空学报(英文版)2024,Vol.37Issue(9) :328-346.DOI:10.1016/j.cja.2024.06.023

Client selection and resource scheduling in reliable federated learning for UAV-assisted vehicular networks

Hongbo ZHAO Liwei GENG Wenquan FENG Changming ZHOU
中国航空学报(英文版)2024,Vol.37Issue(9) :328-346.DOI:10.1016/j.cja.2024.06.023

Client selection and resource scheduling in reliable federated learning for UAV-assisted vehicular networks

Hongbo ZHAO 1Liwei GENG 1Wenquan FENG 1Changming ZHOU1
扫码查看

作者信息

  • 1. School of Elec-tronics and Information Engineering,Beihang University,Beijing 100191,China
  • 折叠

Abstract

Federated Learning(FL),a promising deep learning paradigm extensively deployed in Vehicular Edge Computing Networks(VECN),allows a distributed approach to train datasets of nodes locally,e.g.,for mobile vehicles,and exchanges model parameters to obtain an accurate model without raw data transmission.However,the existence of malicious vehicular nodes as well as the inherent heterogeneity of the vehicles hinders the attainment of accurate models.Moreover,the local model training and model parameter transmission during FL exert a notable energy burden on vehicles constrained in resources.In view of this,we investigate FL client selection and resource management problems in FL-enabled UAV-assisted Vehicular Networks(FLVN).We first devise a novel reputation-based client selection mechanism by integrating both data quality and computation capability metrics to enlist reliable high-performance vehicles.Further,to fortify the FL reliability,we adopt the consortium blockchain to oversee the reputation informa-tion,which boasts tamper-proof and interference-resistant qualities.Finally,we formulate the resource scheduling problem by jointly optimizing the computation capability,the transmission power,and the number of local training rounds,aiming to minimize the cost of clients while guaranteeing accuracy.To this end,we propose a reinforcement learning algorithm employing an asynchronous parallel network structure to achieve an optimized scheduling strategy.Simulation results show that our proposed client selection mechanism and scheduling algorithm can realize reliable FL with an accuracy of 0.96 and consistently outperform the baselines in terms of delay and energy consumption.

Key words

Federated learning/Vehicular edge computing/Resource management/Reinforcement learning/Optimization techniques

引用本文复制引用

基金项目

National Natural Science Foundation of China(61901015)

National Natural Science Foundation of China(62301017)

出版年

2024
中国航空学报(英文版)
中国航空学会

中国航空学报(英文版)

CSTPCDCSCDEI
影响因子:0.847
ISSN:1000-9361
参考文献量3
段落导航相关论文