VREM-FL: Mobility-Aware Computation-Scheduling Co-Design for Vehicular
  Federated Learning
Luca Ballotta Nicolò Dal Fabbro Giovanni Perin Luca Schenato Michele Rossi Giuseppe Piro
Assisted and autonomous driving are rapidly gaining momentum, and will soon
become a reality. Among their key enablers, artificial intelligence and machine
learning are expected to play a prominent role, also thanks to the massive
amount of data that smart vehicles will collect from their onboard sensors. In
this domain, federated learning is one of the most effective and promising
techniques for training global machine learning models, while preserving data
privacy at the vehicles and optimizing communications resource usage. In this
work, we propose VREM-FL, a computation-scheduling co-design for vehicular
federated learning that leverages mobility of vehicles in conjunction with
estimated 5G radio environment maps. VREM-FL jointly optimizes the global model
learned at the server while wisely allocating communication resources. This is
achieved by orchestrating local computations at the vehicles in conjunction
with the transmission of their local model updates in an adaptive and
predictive fashion, by exploiting radio channel maps. The proposed algorithm
can be tuned to trade model training time for radio resource usage.
Experimental results demonstrate the efficacy of utilizing radio maps. VREM-FL
outperforms literature benchmarks for both a linear regression model (learning
time reduced by 28%) and a deep neural network for a semantic image
segmentation task (doubling the number of model updates within the same time
window).