kth.sePublications
Planned maintenance
A system upgrade is planned for 10/12-2024, at 12:00-13:00. During this time DiVA will be unavailable.
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
Broadband Over-the-Air Computation for Federated Learning in Industrial IoT
KTH, School of Electrical Engineering and Computer Science (EECS), Intelligent systems, Information Science and Engineering.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, Information Science and Engineering.ORCID iD: 0000-0002-7474-4294
KTH, School of Industrial Engineering and Management (ITM), Production Engineering, Sustainable Production Systems.ORCID iD: 0000-0001-8679-8049
2022 (English)In: IECON 2022 – 48th Annual Conference of the IEEE Industrial Electronics Society, Institute of Electrical and Electronics Engineers (IEEE), 2022Conference paper, Published paper (Refereed)
Abstract [en]

We consider a broadband over-the-air computation empowered model aggregation scheme for federated learning (FL) in Industrial Internet of Things systems. Due to fading and communication noise, the received global gradient parameters inevitably become inaccurate, leading to a notable decrease of the learning performance. Instead of discarding any edge nodes to reduce the aggregation error, we propose to assign each of them a proper weight coefficient in the model aggregation procedures, i.e., amplitude alignment of the received local gradient parameters from different edge nodes is not required in this paper. We derive an upper bound on the performance loss of the proposed FL scheme, which is shown to be related to the weight coefficients of edge nodes and the mean-squared error (MSE) between the desired global gradient parameters and the actually received ones. Then, we derive a closed-form expression for MSE and use it as the objective function to formulate an optimization problem with respect to the edge nodes’ transmit equalization coefficients, their weight coefficients, and the receive scalars of the cloud server. We transform the formulated optimization problem into a convex one and solve it optimally using CVX. Last, we leverage the popular MNIST dataset and conduct experiments to evaluate the prediction accuracy of the proposed FL scheme. Simulation results demonstrate its superior performances.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2022.
National Category
Electrical Engineering, Electronic Engineering, Information Engineering
Identifiers
URN: urn:nbn:se:kth:diva-322455DOI: 10.1109/IECON49645.2022.9968873Scopus ID: 2-s2.0-85143898152OAI: oai:DiVA.org:kth-322455DiVA, id: diva2:1719477
Conference
IECON 2022–48th Annual Conference of the IEEE Industrial Electronics Society, 17-20 Oktober, Brussels, Belgium
Note

QC 20230621

Available from: 2022-12-15 Created: 2022-12-15 Last updated: 2023-06-21Bibliographically 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, Lihui
By organisation
Information Science and EngineeringSustainable Production Systems
Electrical Engineering, Electronic Engineering, Information Engineering

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 42 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