Enhancing the Performance of Global Model by Improving the Adaptability of Local Models in Federated Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhou, Wujun, Ding, Shu, Li, ZeLin, Wang, Wei
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916741726076928
author Zhou, Wujun
Ding, Shu
Li, ZeLin
Wang, Wei
author_facet Zhou, Wujun
Ding, Shu
Li, ZeLin
Wang, Wei
contents Federated learning enables the clients to collaboratively train a global model, which is aggregated from local models. Due to the heterogeneous data distributions over clients and data privacy in federated learning, it is difficult to train local models to achieve a well-performed global model. In this paper, we introduce the adaptability of local models, i.e., the average performance of local models on data distributions over clients, and enhance the performance of the global model by improving the adaptability of local models. Since each client does not know the data distributions over other clients, the adaptability of the local model cannot be directly optimized. First, we provide the property of an appropriate local model which has good adaptability on the data distributions over clients. Then, we formalize the property into the local training objective with a constraint and propose a feasible solution to train the local model. Extensive experiments on federated learning benchmarks demonstrate that our method significantly improves the adaptability of local models and achieves a well-performed global model that consistently outperforms the baseline methods.
format Preprint
id arxiv_https___arxiv_org_abs_2505_10125
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Enhancing the Performance of Global Model by Improving the Adaptability of Local Models in Federated Learning
Zhou, Wujun
Ding, Shu
Li, ZeLin
Wang, Wei
Machine Learning
Federated learning enables the clients to collaboratively train a global model, which is aggregated from local models. Due to the heterogeneous data distributions over clients and data privacy in federated learning, it is difficult to train local models to achieve a well-performed global model. In this paper, we introduce the adaptability of local models, i.e., the average performance of local models on data distributions over clients, and enhance the performance of the global model by improving the adaptability of local models. Since each client does not know the data distributions over other clients, the adaptability of the local model cannot be directly optimized. First, we provide the property of an appropriate local model which has good adaptability on the data distributions over clients. Then, we formalize the property into the local training objective with a constraint and propose a feasible solution to train the local model. Extensive experiments on federated learning benchmarks demonstrate that our method significantly improves the adaptability of local models and achieves a well-performed global model that consistently outperforms the baseline methods.
title Enhancing the Performance of Global Model by Improving the Adaptability of Local Models in Federated Learning
topic Machine Learning
url https://arxiv.org/abs/2505.10125