PARDON: Privacy-Aware and Robust Federated Domain Generalization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Nguyen, Dung Thuy, Johnson, Taylor T., Leach, Kevin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910906827407360
author Nguyen, Dung Thuy
Johnson, Taylor T.
Leach, Kevin
author_facet Nguyen, Dung Thuy
Johnson, Taylor T.
Leach, Kevin
contents Federated Learning (FL) shows promise in preserving privacy and enabling collaborative learning. However, most current solutions focus on private data collected from a single domain. A significant challenge arises when client data comes from diverse domains (i.e., domain shift), leading to poor performance on unseen domains. Existing Federated Domain Generalization approaches address this problem but assume each client holds data for an entire domain, limiting their practicality in real-world scenarios with domain-based heterogeneity and client sampling. In addition, certain methods enable information sharing among clients, raising privacy concerns as this information could be used to reconstruct sensitive private data. To overcome this, we introduce FISC, a novel FedDG paradigm designed to robustly handle more complicated domain distributions between clients while ensuring security. FISC enables learning across domains by extracting an interpolative style from local styles and employing contrastive learning. This strategy gives clients multi-domain representations and unbiased convergent targets. Empirical results on multiple datasets, including PACS, Office-Home, and IWildCam, show FISC outperforms state-of-the-art (SOTA) methods. Our method achieves accuracy on unseen domains, with improvements ranging from 3.64% to 57.22% on unseen domains. Our code is available at https://github.com/judydnguyen/PARDON-FedDG.
format Preprint
id arxiv_https___arxiv_org_abs_2410_22622
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle PARDON: Privacy-Aware and Robust Federated Domain Generalization
Nguyen, Dung Thuy
Johnson, Taylor T.
Leach, Kevin
Machine Learning
Computer Vision and Pattern Recognition
Distributed, Parallel, and Cluster Computing
Federated Learning (FL) shows promise in preserving privacy and enabling collaborative learning. However, most current solutions focus on private data collected from a single domain. A significant challenge arises when client data comes from diverse domains (i.e., domain shift), leading to poor performance on unseen domains. Existing Federated Domain Generalization approaches address this problem but assume each client holds data for an entire domain, limiting their practicality in real-world scenarios with domain-based heterogeneity and client sampling. In addition, certain methods enable information sharing among clients, raising privacy concerns as this information could be used to reconstruct sensitive private data. To overcome this, we introduce FISC, a novel FedDG paradigm designed to robustly handle more complicated domain distributions between clients while ensuring security. FISC enables learning across domains by extracting an interpolative style from local styles and employing contrastive learning. This strategy gives clients multi-domain representations and unbiased convergent targets. Empirical results on multiple datasets, including PACS, Office-Home, and IWildCam, show FISC outperforms state-of-the-art (SOTA) methods. Our method achieves accuracy on unseen domains, with improvements ranging from 3.64% to 57.22% on unseen domains. Our code is available at https://github.com/judydnguyen/PARDON-FedDG.
title PARDON: Privacy-Aware and Robust Federated Domain Generalization
topic Machine Learning
Computer Vision and Pattern Recognition
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2410.22622