Adaptive Latent-Space Constraints in Personalized Federated Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Ayromlou, Sana, Tavakoli, Fatemeh, Emerson, D. B.
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917038142783488
author Ayromlou, Sana
Tavakoli, Fatemeh
Emerson, D. B.
author_facet Ayromlou, Sana
Tavakoli, Fatemeh
Emerson, D. B.
contents Federated learning (FL) is an effective and widely used approach to training deep learning models on decentralized datasets held by distinct clients. FL also strengthens both security and privacy protections for training data. Common challenges associated with statistical heterogeneity between distributed datasets have spurred significant interest in personalized FL (pFL) methods, where models combine aspects of global learning with local modeling specific to each client's unique characteristics. This work investigates the efficacy of theoretically supported, adaptive MMD measures in pFL, primarily focusing on the Ditto framework, a state-of-the-art technique for distributed data heterogeneity. The use of such measures significantly improves model performance across a variety of tasks, especially those with pronounced feature heterogeneity. Additional experiments demonstrate that such measures are directly applicable to other pFL techniques and yield similar improvements across a number of datasets. Finally, the results motivate the use of constraints tailored to the various kinds of heterogeneity expected in FL systems.
format Preprint
id arxiv_https___arxiv_org_abs_2505_07525
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Adaptive Latent-Space Constraints in Personalized Federated Learning
Ayromlou, Sana
Tavakoli, Fatemeh
Emerson, D. B.
Machine Learning
68T07
I.2.0; I.2.11; I.2.6
Federated learning (FL) is an effective and widely used approach to training deep learning models on decentralized datasets held by distinct clients. FL also strengthens both security and privacy protections for training data. Common challenges associated with statistical heterogeneity between distributed datasets have spurred significant interest in personalized FL (pFL) methods, where models combine aspects of global learning with local modeling specific to each client's unique characteristics. This work investigates the efficacy of theoretically supported, adaptive MMD measures in pFL, primarily focusing on the Ditto framework, a state-of-the-art technique for distributed data heterogeneity. The use of such measures significantly improves model performance across a variety of tasks, especially those with pronounced feature heterogeneity. Additional experiments demonstrate that such measures are directly applicable to other pFL techniques and yield similar improvements across a number of datasets. Finally, the results motivate the use of constraints tailored to the various kinds of heterogeneity expected in FL systems.
title Adaptive Latent-Space Constraints in Personalized Federated Learning
topic Machine Learning
68T07
I.2.0; I.2.11; I.2.6
url https://arxiv.org/abs/2505.07525