2021 IEEE International Conference on Acoustics, Speech and Signal Processing

6-11 June 2021 • Toronto, Ontario, Canada

Extracting Knowledge from Information

2021 IEEE International Conference on Acoustics, Speech and Signal Processing

6-11 June 2021 • Toronto, Ontario, Canada

Extracting Knowledge from Information
Login Paper Search My Schedule Paper Index Help

My ICASSP 2021 Schedule

Note: Your custom schedule will not be saved unless you create a new account or login to an existing account.
  1. Create a login based on your email (takes less than one minute)
  2. Perform 'Paper Search'
  3. Select papers that you desire to save in your personalized schedule
  4. Click on 'My Schedule' to see the current list of selected papers
  5. Click on 'Printable Version' to create a separate window suitable for printing (the header and menu will appear, but will not actually print)

Paper Detail

Paper IDMLSP-13.5
Paper Title Energy Minimization for Federated Learning with IRS-Assisted Over-the-Air Computation
Authors Yuntao Hu, Ming Chen, Southeast University, China; Mingzhe Chen, Princeton University, United States; Zhaohui Yang, Mohammad Shikh-Bahaei, King's College London, United Kingdom; H. Vincent Poor, Princeton University, United States; Shuguang Cui, the Chinese University of Hong Kong, Hong Kong SAR China
SessionMLSP-13: Federated Learning 2
LocationGather.Town
Session Time:Wednesday, 09 June, 13:00 - 13:45
Presentation Time:Wednesday, 09 June, 13:00 - 13:45
Presentation Poster
Topic Machine Learning for Signal Processing: [MLR-DFED] Distributed/Federated learning
IEEE Xplore Open Preview  Click here to view in IEEE Xplore
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.