학술논문

Communication and Computation Efficient Federated Learning for Internet of Vehicles With a Constrained Latency
Document Type
Periodical
Source
IEEE Transactions on Vehicular Technology IEEE Trans. Veh. Technol. Vehicular Technology, IEEE Transactions on. 73(1):1038-1052 Jan, 2024
Subject
Transportation
Aerospace
Wireless communication
Convergence
Communication system security
Computational modeling
Resource management
Costs
Servers
Internet of Vehicles (IoV)
intelligent transportation system (ITS)
wireless federated learning
gradient compression
learning latency
convergence rate
Language
ISSN
0018-9545
1939-9359
Abstract
Considering the privacy and security issues in Internet of Vehicles (IoV), wireless federated learning (FL) can be adopted to facilitate various emerging vehicular applications. However, wireless FL would suffer from a large learning latency due to the limitation of bandwidth and computing power as well as the unreliable communication caused by vehicle mobility. To cope with these challenges, a new structure is designed in this article to facilitate the implementation of FL for IoV. First, we apply the gradient compression and mini-batch federated stochastic gradient descent to reduce the local gradient transmission and computation. Then, with theoretical analysis of the convergence rate and the learning latency, the learning performance can be improved by maximizing the convergence rate under a constrained latency. Accordingly, an optimization problem is formulated to jointly optimize compression ratio, batch size, and spectrum allocation. To solve this problem, an iterative algorithm is developed by problem decomposition. From the results, compression ratio and batch size should be adjusted according to the channel state information and computing power of the road side units to boost the learning efficiency at the cost of slight degradation on the learning accuracy. The superiority of the proposed algorithm is finally demonstrated through extensive simulations.