kth.sePublications KTH
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
Over-the-Air Federated Learning via Weighted Aggregation
KTH, School of Electrical Engineering and Computer Science (EECS), Computer Science, Network and Systems Engineering.ORCID iD: 0000-0002-7297-5953
Yale Univ, Dept Elect Engn, Inst Network Sci, New Haven, CT 06520 USA..
2024 (English)In: IEEE Transactions on Wireless Communications, ISSN 1536-1276, E-ISSN 1558-2248, Vol. 23, no 12, p. 18240-18253Article in journal (Refereed) Published
Abstract [en]

This paper introduces a new federated learning scheme that leverages over-the-air computation. A novel feature of this scheme is the proposal to employ adaptive weights during aggregation, a facet treated as predefined in other over-the-air schemes. This can mitigate the impact of wireless channel conditions on learning performance, without needing channel state information at transmitter side (CSIT). We provide a mathematical methodology to derive the convergence bound for the proposed scheme in the context of computational heterogeneity and general loss functions, supplemented with design insights. Accordingly, we propose aggregation cost metrics and efficient algorithms to find optimized weights for the aggregation. Finally, through numerical experiments, we validate the effectiveness of the proposed scheme. Even with the challenges posed by channel conditions and device heterogeneity, the proposed scheme surpasses other over-the-air strategies by an accuracy improvement of 15% over the scheme using CSIT and 30% compared to the one without CSIT.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE) , 2024. Vol. 23, no 12, p. 18240-18253
Keywords [en]
Wireless networks, Servers, Transmitters, Resource management, Receivers, Performance evaluation, Convergence, Federated learning, machine learning, fading multiple access channel, over-the-air computation, analog communications
National Category
Telecommunications
Identifiers
URN: urn:nbn:se:kth:diva-358712DOI: 10.1109/TWC.2024.3463754ISI: 001376971600030Scopus ID: 2-s2.0-85205520816OAI: oai:DiVA.org:kth-358712DiVA, id: diva2:1932138
Note

QC 20250128

Available from: 2025-01-28 Created: 2025-01-28 Last updated: 2025-01-28Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Azimi Abarghouyi, Seyed Mohammad

Search in DiVA

By author/editor
Azimi Abarghouyi, Seyed Mohammad
By organisation
Network and Systems Engineering
In the same journal
IEEE Transactions on Wireless Communications
Telecommunications

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

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