Efficient Federated Learning in Wireless Communication Systems: A Multi-Objective Optimization Perspective

Yu Zhou, Xiaohui Zhao, Xuemei Liu, Lei You, Lei Lei

Research output: Chapter in Book/Report/Conference proceedingArticle in proceedingsResearchpeer-review

68 Downloads (Pure)

Abstract

This paper focuses on a federated learning (FL) system that employs a base station as a central server while clients with limited computation capabilities perform local training. The limited bandwidth leads to that only a portion of clients can participate in each FL training round, and picking different clients can impact the performance of FL systems, requiring effective allocation of their computing resources. In FL systems, both model convergence and energy consumption are important performances. To this end, we formulate a multi-objective optimization problem (MOP) to simultaneously speed up model convergence and reduce energy consumption. To address the MOP, we propose a multi-objective algorithm (MOA) for FL systems to obtain a Pareto optimal solution set, where Tchebycheff approach is adopted to divide MOP into multiple single- objective problems and optimize them by differential evolution. The extensive experiments on Fashion-MNIST dataset in both i.i.d and non-i.i.d data settings illustrates that MOA outperforms other algorithms.
Original languageEnglish
Title of host publicationProceedings of 2023 IEEE/CIC International Conference on Communications in China (ICCC)
Number of pages5
PublisherIEEE
Publication date2023
ISBN (Electronic)979-8-3503-4538-4
DOIs
Publication statusPublished - 2023
Event12th IEEE/CIC lnternational Conference on Communications in China - Dalian, China
Duration: 10 Aug 202312 Aug 2023
Conference number: 12

Conference

Conference12th IEEE/CIC lnternational Conference on Communications in China
Number12
Country/TerritoryChina
CityDalian
Period10/08/202312/08/2023

Keywords

  • Federated learning
  • Multi-objective algorithm
  • Differential evolution
  • Decomposition

Fingerprint

Dive into the research topics of 'Efficient Federated Learning in Wireless Communication Systems: A Multi-Objective Optimization Perspective'. Together they form a unique fingerprint.

Cite this