Energy minimization for federated learning with IRS-assisted over-the-air computation

Yuntao Hu, Ming Chen, Mingzhe Chen, Zhaohui Yang, Mohammad Shikh-Bahaei, H. Vincent Poor, Shuguang Cui

Research output: Contribution to journalConference articlepeer-review

Abstract

This paper investigates the deployment of federated learning (FL) over an over-the-air computation (AirComp) and intelligent reflecting surface (IRS) based wireless network. In the considered system, devices transmit locally trained machine learning (ML) models to the base station (BS) which aggregates the received ML models and generates a shared global ML model. The devices can directly transmit ML models to the BS or using IRS. Meanwhile, AirComp is used to aggregate ML models that are transmitted from the devices to the BS. To minimize the energy consumption of devices, an energy minimization problem is formulated, which jointly optimizes the device selection, phase shift matrix, decoding vector, and power control. To seek the solution, the original optimization problem is divided into four sub-problems. Then the fractional program, greedy algorithm, matrix derivation, and weighted minimum mean square error methods are used to compute the phase shift matrix, device selection vector, decoding vector, and transmit power, respectively. Simulation results show that the proposed algorithm can reduce 11.2% energy consumption of devices compared to an FL algorithm that is implemented at a network without any IRSs.

Original languageEnglish (US)
Pages (from-to)3105-3109
Number of pages5
JournalICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings
Volume2021-June
DOIs
StatePublished - 2021
Event2021 IEEE International Conference on Acoustics, Speech, and Signal Processing, ICASSP 2021 - Virtual, Toronto, Canada
Duration: Jun 6 2021Jun 11 2021

All Science Journal Classification (ASJC) codes

  • Software
  • Signal Processing
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Energy minimization for federated learning with IRS-assisted over-the-air computation'. Together they form a unique fingerprint.

Cite this