Federated Domain Generalization with Latent Space Inversion

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Palakkadavath, Ragja, Le, Hung, Nguyen-Tang, Thanh, Venkatesh, Svetha, Gupta, Sunil
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914194059689984
author Palakkadavath, Ragja
Le, Hung
Nguyen-Tang, Thanh
Venkatesh, Svetha
Gupta, Sunil
author_facet Palakkadavath, Ragja
Le, Hung
Nguyen-Tang, Thanh
Venkatesh, Svetha
Gupta, Sunil
contents Federated domain generalization (FedDG) addresses distribution shifts among clients in a federated learning framework. FedDG methods aggregate the parameters of locally trained client models to form a global model that generalizes to unseen clients while preserving data privacy. While improving the generalization capability of the global model, many existing approaches in FedDG jeopardize privacy by sharing statistics of client data between themselves. Our solution addresses this problem by contributing new ways to perform local client training and model aggregation. To improve local client training, we enforce (domain) invariance across local models with the help of a novel technique, \textbf{latent space inversion}, which enables better client privacy. When clients are not \emph{i.i.d}, aggregating their local models may discard certain local adaptations. To overcome this, we propose an \textbf{important weight} aggregation strategy to prioritize parameters that significantly influence predictions of local models during aggregation. Our extensive experiments show that our approach achieves superior results over state-of-the-art methods with less communication overhead.
format Preprint
id arxiv_https___arxiv_org_abs_2512_10224
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Federated Domain Generalization with Latent Space Inversion
Palakkadavath, Ragja
Le, Hung
Nguyen-Tang, Thanh
Venkatesh, Svetha
Gupta, Sunil
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Federated domain generalization (FedDG) addresses distribution shifts among clients in a federated learning framework. FedDG methods aggregate the parameters of locally trained client models to form a global model that generalizes to unseen clients while preserving data privacy. While improving the generalization capability of the global model, many existing approaches in FedDG jeopardize privacy by sharing statistics of client data between themselves. Our solution addresses this problem by contributing new ways to perform local client training and model aggregation. To improve local client training, we enforce (domain) invariance across local models with the help of a novel technique, \textbf{latent space inversion}, which enables better client privacy. When clients are not \emph{i.i.d}, aggregating their local models may discard certain local adaptations. To overcome this, we propose an \textbf{important weight} aggregation strategy to prioritize parameters that significantly influence predictions of local models during aggregation. Our extensive experiments show that our approach achieves superior results over state-of-the-art methods with less communication overhead.
title Federated Domain Generalization with Latent Space Inversion
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2512.10224