Over-the-Air Fair Federated Learning via Multi-Objective Optimization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Hamidi, Shayan Mohajer, Bereyhi, Ali, Asaad, Saba, Poor, H. Vincent
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909449712566272
author Hamidi, Shayan Mohajer
Bereyhi, Ali
Asaad, Saba
Poor, H. Vincent
author_facet Hamidi, Shayan Mohajer
Bereyhi, Ali
Asaad, Saba
Poor, H. Vincent
contents In federated learning (FL), heterogeneity among the local dataset distributions of clients can result in unsatisfactory performance for some, leading to an unfair model. To address this challenge, we propose an over-the-air fair federated learning algorithm (OTA-FFL), which leverages over-the-air computation to train fair FL models. By formulating FL as a multi-objective minimization problem, we introduce a modified Chebyshev approach to compute adaptive weighting coefficients for gradient aggregation in each communication round. To enable efficient aggregation over the multiple access channel, we derive analytical solutions for the optimal transmit scalars at the clients and the de-noising scalar at the parameter server. Extensive experiments demonstrate the superiority of OTA-FFL in achieving fairness and robust performance compared to existing methods.
format Preprint
id arxiv_https___arxiv_org_abs_2501_03392
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Over-the-Air Fair Federated Learning via Multi-Objective Optimization
Hamidi, Shayan Mohajer
Bereyhi, Ali
Asaad, Saba
Poor, H. Vincent
Machine Learning
Artificial Intelligence
In federated learning (FL), heterogeneity among the local dataset distributions of clients can result in unsatisfactory performance for some, leading to an unfair model. To address this challenge, we propose an over-the-air fair federated learning algorithm (OTA-FFL), which leverages over-the-air computation to train fair FL models. By formulating FL as a multi-objective minimization problem, we introduce a modified Chebyshev approach to compute adaptive weighting coefficients for gradient aggregation in each communication round. To enable efficient aggregation over the multiple access channel, we derive analytical solutions for the optimal transmit scalars at the clients and the de-noising scalar at the parameter server. Extensive experiments demonstrate the superiority of OTA-FFL in achieving fairness and robust performance compared to existing methods.
title Over-the-Air Fair Federated Learning via Multi-Objective Optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2501.03392