OFDMA-F2L: Federated Learning With Flexible Aggregation Over an OFDMA Air Interface

Shuyan Hu, Xin Yuan, Wei Ni, Xin Wang, Ekram Hossain, H. Vincent Poor

Research output: Contribution to journalArticlepeer-review

Abstract

Federated learning (FL) can suffer from communication bottlenecks when deployed in mobile networks, limiting participating clients and deterring FL convergence. In this context, the impact of practical air interfaces with discrete modulation schemes on FL has not previously been studied in depth. This paper proposes a new paradigm of flexible aggregation-based FL (F2L) over an orthogonal frequency division multiple-access (OFDMA) air interface, termed as “OFDMA-F2L”, allowing selected clients to train local models for various numbers of iterations before uploading the models in each aggregation round. We optimize the selections of clients, subchannels and modulation scheme, adapting to channel conditions and computing power. Specifically, we derive an upper bound on the optimality gap of OFDMA-F2L capturing the impact of these selections, and show that the upper bound is minimized by maximizing the weighted sum rate of the clients per aggregation round. A Lagrange-dual based method is developed to solve this challenging mixed integer program of weighted sum rate maximization, revealing that a “winner-takes-all” policy provides the almost surely optimal client, subchannel, and modulation selections. Experiments on multilayer perceptrons and convolutional neural networks show that OFDMA-F2L with optimal selections can significantly improve the training convergence and accuracy, e.g., by about 18% and 5%, compared to potential alternatives.

Original languageEnglish (US)
Pages (from-to)1
Number of pages1
JournalIEEE Transactions on Wireless Communications
DOIs
StateAccepted/In press - 2024
Externally publishedYes

All Science Journal Classification (ASJC) codes

  • Computer Science Applications
  • Electrical and Electronic Engineering
  • Applied Mathematics

Keywords

  • Adaptation models
  • Atmospheric modeling
  • Computational modeling
  • Convergence
  • Federated learning
  • Modulation
  • Training
  • Wireless communication
  • channel allocation
  • client selection
  • convergence analysis
  • flexible aggregation
  • modulation selection

Fingerprint

Dive into the research topics of 'OFDMA-F2L: Federated Learning With Flexible Aggregation Over an OFDMA Air Interface'. Together they form a unique fingerprint.

Cite this