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...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:Proceedings of ... IEEE International Conference on Computer and Communications (Online) s. 2191 - 2197
Hlavní autoři: Ji, Juncheng, Lam, Chan-Tong, Wang, Ke, Ng, Benjamin K.
Médium: Konferenční příspěvek
Jazyk:angličtina
Vydáno: IEEE 08.12.2023
Témata:
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
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí: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