Federated Learning with Nonvacuous Generalisation Bounds

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Jobic, Pierre, Haddouche, Maxime, Guedj, Benjamin
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914579644153856
author Jobic, Pierre
Haddouche, Maxime
Guedj, Benjamin
author_facet Jobic, Pierre
Haddouche, Maxime
Guedj, Benjamin
contents We introduce a novel strategy to train randomised predictors in federated learning, where each node of the network aims at preserving its privacy by releasing a local predictor but keeping secret its training dataset with respect to the other nodes. We then build a global randomised predictor which inherits the properties of the local private predictors in the sense of a PAC-Bayesian generalisation bound. We consider the synchronous case where all nodes share the same training objective (derived from a generalisation bound), and the heterogenous and homogenous cases where each node may have its own personalised training objective. We show through a series of numerical experiments that our approach achieves a comparable predictive performance to that of the batch approach where all datasets are shared across nodes. Moreover the predictors are supported by numerically nonvacuous generalisation bounds while preserving privacy for each node. We explicitly compute the increment on predictive performance and generalisation bounds for our two federated settings, highlighting the price to pay to preserve privacy.
format Preprint
id arxiv_https___arxiv_org_abs_2310_11203
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Federated Learning with Nonvacuous Generalisation Bounds
Jobic, Pierre
Haddouche, Maxime
Guedj, Benjamin
Machine Learning
We introduce a novel strategy to train randomised predictors in federated learning, where each node of the network aims at preserving its privacy by releasing a local predictor but keeping secret its training dataset with respect to the other nodes. We then build a global randomised predictor which inherits the properties of the local private predictors in the sense of a PAC-Bayesian generalisation bound. We consider the synchronous case where all nodes share the same training objective (derived from a generalisation bound), and the heterogenous and homogenous cases where each node may have its own personalised training objective. We show through a series of numerical experiments that our approach achieves a comparable predictive performance to that of the batch approach where all datasets are shared across nodes. Moreover the predictors are supported by numerically nonvacuous generalisation bounds while preserving privacy for each node. We explicitly compute the increment on predictive performance and generalisation bounds for our two federated settings, highlighting the price to pay to preserve privacy.
title Federated Learning with Nonvacuous Generalisation Bounds
topic Machine Learning
url https://arxiv.org/abs/2310.11203