Federated Learning for Non-factorizable Models using Deep Generative Prior Approximations

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hassan, Conor, Bon, Joshua J, Semenova, Elizaveta, Mira, Antonietta, Mengersen, Kerrie
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909210398162944
author Hassan, Conor
Bon, Joshua J
Semenova, Elizaveta
Mira, Antonietta
Mengersen, Kerrie
author_facet Hassan, Conor
Bon, Joshua J
Semenova, Elizaveta
Mira, Antonietta
Mengersen, Kerrie
contents Federated learning (FL) allows for collaborative model training across decentralized clients while preserving privacy by avoiding data sharing. However, current FL methods assume conditional independence between client models, limiting the use of priors that capture dependence, such as Gaussian processes (GPs). We introduce the Structured Independence via deep Generative Model Approximation (SIGMA) prior which enables FL for non-factorizable models across clients, expanding the applicability of FL to fields such as spatial statistics, epidemiology, environmental science, and other domains where modeling dependencies is crucial. The SIGMA prior is a pre-trained deep generative model that approximates the desired prior and induces a specified conditional independence structure in the latent variables, creating an approximate model suitable for FL settings. We demonstrate the SIGMA prior's effectiveness on synthetic data and showcase its utility in a real-world example of FL for spatial data, using a conditional autoregressive prior to model spatial dependence across Australia. Our work enables new FL applications in domains where modeling dependent data is essential for accurate predictions and decision-making.
format Preprint
id arxiv_https___arxiv_org_abs_2405_16055
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Federated Learning for Non-factorizable Models using Deep Generative Prior Approximations
Hassan, Conor
Bon, Joshua J
Semenova, Elizaveta
Mira, Antonietta
Mengersen, Kerrie
Machine Learning
Computation
Methodology
Federated learning (FL) allows for collaborative model training across decentralized clients while preserving privacy by avoiding data sharing. However, current FL methods assume conditional independence between client models, limiting the use of priors that capture dependence, such as Gaussian processes (GPs). We introduce the Structured Independence via deep Generative Model Approximation (SIGMA) prior which enables FL for non-factorizable models across clients, expanding the applicability of FL to fields such as spatial statistics, epidemiology, environmental science, and other domains where modeling dependencies is crucial. The SIGMA prior is a pre-trained deep generative model that approximates the desired prior and induces a specified conditional independence structure in the latent variables, creating an approximate model suitable for FL settings. We demonstrate the SIGMA prior's effectiveness on synthetic data and showcase its utility in a real-world example of FL for spatial data, using a conditional autoregressive prior to model spatial dependence across Australia. Our work enables new FL applications in domains where modeling dependent data is essential for accurate predictions and decision-making.
title Federated Learning for Non-factorizable Models using Deep Generative Prior Approximations
topic Machine Learning
Computation
Methodology
url https://arxiv.org/abs/2405.16055