Energy-Latency Tradeoff for Joint Optimization of Vehicle Selection and Resource Allocation in UAV-Assisted Vehicular Edge Computing

Abstract

In Unmanned Aerial Vehicle (UAV)-assisted Vehicular Edge Computing (VEC), Federated Learning (FL) offers a means to protect user privacy during the training of models using multiple vehicle datasets. However, involving numerous vehicles in the training process can lead to significant communication overhead, thereby increasing FL latency and energy consumption. To address this issue, we propose an energy-latency tradeoff scheme for the joint optimization of vehicle selection and resource allocation in UAV-assisted VEC. Our investigation focuses on maximizing long-term training rewards for vehicle selection and resource allocation in FL, while considering constraints such as UAV energy consumption, vehicular energy consumption, bandwidth, and vehicle mobility. This problem is formulated as a Mixed-Integer Nonlinear Programming (MINLP) problem and modeled as a Markov Decision Process (MDP). We proposed an algorithm based on AdamW and Butterfly Optimization Algorithm (BOA) for Double-Depth Q-networks (AB-DDQN) to determine the optimal decisions. To expedite algorithm convergence, we replace the stochastic gradient descent (SGD) algorithm with AdamW algorithm and employ BOA to select hyperparameters, enhancing algorithm performance. Experimental validation using the GTSDB dataset demonstrates that our algorithm effectively reduces latency and energy consumption in FL.