kth.sePublications
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
IRS Assisted Federated Learning: A Broadband Over-the-Air Aggregation Approach
KTH, School of Electrical Engineering and Computer Science (EECS), Intelligent systems, Information Science and Engineering. Beihang Univ, Sch Elect & Informat Engn, Beijing 100191, Peoples R China.ORCID iD: 0000-0001-9621-561X
KTH, School of Electrical Engineering and Computer Science (EECS), Intelligent systems, Information Science and Engineering.ORCID iD: 0000-0002-5407-0835
KTH, School of Electrical Engineering and Computer Science (EECS), Intelligent systems. ABB Corp Res, S-72226 Västerås, Sweden.ORCID iD: 0000-0002-7474-4294
KTH, School of Industrial Engineering and Management (ITM), Production engineering, Industrial Production Systems.ORCID iD: 0000-0001-8679-8049
Show others and affiliations
2024 (English)In: IEEE Transactions on Wireless Communications, ISSN 1536-1276, E-ISSN 1558-2248, Vol. 23, no 5, p. 4069-4082Article in journal (Refereed) Published
Abstract [en]

We consider a broadband over-the-air computation empowered model aggregation approach for wireless federated learning (FL) systems and propose to leverage an intelligent reflecting surface (IRS) to combat wireless fading and noise. We first investigate the conventional node-selection based framework, where a few edge nodes are dropped in model aggregation to control the aggregation error. We analyze the performance of this node-selection based framework and derive an upper bound on its performance loss, which is shown to be related to the selected edge nodes. Then, we seek to minimize the mean-squared error (MSE) between the desired global gradient parameters and the actually received ones by optimizing the selected edge nodes, their transmit equalization coefficients, the IRS phase shifts, and the receive factors of the cloud server. By resorting to the matrix lifting technique and difference-of-convex programming, we successfully transform the formulated optimization problem into a convex one and solve it using off-the-shelf solvers. To improve learning performance, we further propose a weight-selection based FL framework. In such a framework, we assign each edge node a proper weight coefficient in model aggregation instead of discarding any of them to reduce the aggregation error, i.e., amplitude alignment of the received local gradient parameters from different edge nodes is not required. We also analyze the performance of this weight-selection based framework and derive an upper bound on its performance loss, followed by minimizing the MSE via optimizing the weight coefficients of the edge nodes, their transmit equalization coefficients, the IRS phase shifts, and the receive factors of the cloud server. Furthermore, we use the MNIST dataset for simulations to evaluate the performance of both node-selection and weight-selection based FL frameworks.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE) , 2024. Vol. 23, no 5, p. 4069-4082
Keywords [en]
Servers, Atmospheric modeling, Computational modeling, Performance evaluation, Industrial Internet of Things, Wireless networks, Propagation losses, Federated learning, intelligent reflecting surface, over-the-air computation, OFDM
National Category
Communication Systems
Identifiers
URN: urn:nbn:se:kth:diva-348619DOI: 10.1109/TWC.2023.3313968ISI: 001244908800092Scopus ID: 2-s2.0-85185382129OAI: oai:DiVA.org:kth-348619DiVA, id: diva2:1877703
Note

QC 20240626

Available from: 2024-06-26 Created: 2024-06-26 Last updated: 2024-06-26Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Zhang, DeyouXiao, MingPang, ZhiboWang, Lihui

Search in DiVA

By author/editor
Zhang, DeyouXiao, MingPang, ZhiboWang, LihuiPoor, H. Vincent
By organisation
Information Science and EngineeringIntelligent systemsIndustrial Production Systems
In the same journal
IEEE Transactions on Wireless Communications
Communication Systems

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 110 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf