A Dynamic Negative Log-Likelihood Optimization Method for Device Selection in Federated Learning with Over-The-Air Computation

This paper addresses an optimization problem of device selections and aggregation errors in over-the-air computation with federated learning (AirCompFL) systems. To achieve this, we first propose an AirCompFL system model and then formulate the device selections problem as a multi-objective mixed in...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Proceedings of ... IEEE International Conference on Computer and Communications (Online) S. 2191 - 2197
Hauptverfasser: Ji, Juncheng, Lam, Chan-Tong, Wang, Ke, Ng, Benjamin K.
Format: Tagungsbericht
Sprache:Englisch
Veröffentlicht: IEEE 08.12.2023
Schlagworte:
aggregation error
Atmospheric modeling
communication efficiency
Computational modeling
device selection
Federated learning
Integer programming
Over-the-air computation
Performance evaluation
System performance
Wireless networks
ISSN:2837-7109
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:This paper addresses an optimization problem of device selections and aggregation errors in over-the-air computation with federated learning (AirCompFL) systems. To achieve this, we first propose an AirCompFL system model and then formulate the device selections problem as a multi-objective mixed integer programming problem. We then propose a dynamic negative log-likelihood weighted optimization decision (DNLWOD) approach to solve the above problem. The method selects a device based on multiple criteria for enhancing the overall performance, which optimizes the weight of each criterion to balance and minimize aggregation errors automatically and simultaneously. Experimental results show that the DNLWOD method can effectively reduce aggregation errors to enhance the performance of the AirCompFL system, outperforming the existing algorithms in terms of the overall performance, average performance and aggregated error. This work shows that in a wireless edge networking environment with the AirCompFL system, the proposed scheme can provide an effective strategy for selecting devices and optimizing aggregation to increase the communication efficiency and mitigate the aggregation errors.
ISSN:2837-7109
DOI:10.1109/ICCC59590.2023.10507265