Performance Optimization of Federated Learning over Mobile Wireless Networks

Mingzhe Chen, H. Vincent Poor, Walid Saad, Shuguang Cui

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

In this paper, the problem of training federated learning (FL) algorithms over a wireless network with mobile users is studied. In the considered model, several mobile users and a network base station (BS) cooperatively perform an FL algorithm. In particular, the wireless mobile users train their local FL models and send the trained local FL model parameters to the BS. The BS will then integrate the received local FL models to generate a global FL model and send it back to all users. Due to the limited training time at each iteration, the number of users that can transmit their local FL models to the BS will be affected by changes in the users' locations and wireless channels. In this paper, this joint learning, user selection, and wireless resource allocation problem is formulated as an optimization problem whose goal is to minimize the FL loss function, which captures the FL performance, while meeting the transmission delay requirement. To solve this problem, a closed-form expression for the expected convergence rate of the FL algorithm is first derived to quantify the impact of the users' mobility and wireless factors on FL. Then, based on the expected FL convergence rate, the user selection and uplink resource allocation is optimized at each FL iteration so as to minimize the FL loss function while satisfying the FL parameter transmission delay requirement. Simulation results show that the proposed approach can reduce the FL loss function value by up to 20% compared to a standard FL algorithm.

Original languageEnglish (US)
Title of host publication2020 IEEE 21st International Workshop on Signal Processing Advances in Wireless Communications, SPAWC 2020
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781728154787
DOIs
StatePublished - May 2020
Event21st IEEE International Workshop on Signal Processing Advances in Wireless Communications, SPAWC 2020 - Atlanta, United States
Duration: May 26 2020May 29 2020

Publication series

NameIEEE Workshop on Signal Processing Advances in Wireless Communications, SPAWC
Volume2020-May

Conference

Conference21st IEEE International Workshop on Signal Processing Advances in Wireless Communications, SPAWC 2020
CountryUnited States
CityAtlanta
Period5/26/205/29/20

All Science Journal Classification (ASJC) codes

  • Electrical and Electronic Engineering
  • Computer Science Applications
  • Information Systems

Fingerprint Dive into the research topics of 'Performance Optimization of Federated Learning over Mobile Wireless Networks'. Together they form a unique fingerprint.

  • Cite this

    Chen, M., Poor, H. V., Saad, W., & Cui, S. (2020). Performance Optimization of Federated Learning over Mobile Wireless Networks. In 2020 IEEE 21st International Workshop on Signal Processing Advances in Wireless Communications, SPAWC 2020 [9154300] (IEEE Workshop on Signal Processing Advances in Wireless Communications, SPAWC; Vol. 2020-May). Institute of Electrical and Electronics Engineers Inc.. https://doi.org/10.1109/SPAWC48557.2020.9154300