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

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 information, 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.