Personalized Multi-tier Federated Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Banerjee, Sourasekhar, Dadras, Ali, Yurtsever, Alp, Bhuyan, Monowar
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916330325671936
author Banerjee, Sourasekhar
Dadras, Ali
Yurtsever, Alp
Bhuyan, Monowar
author_facet Banerjee, Sourasekhar
Dadras, Ali
Yurtsever, Alp
Bhuyan, Monowar
contents The key challenge of personalized federated learning (PerFL) is to capture the statistical heterogeneity properties of data with inexpensive communications and gain customized performance for participating devices. To address these, we introduced personalized federated learning in multi-tier architecture (PerMFL) to obtain optimized and personalized local models when there are known team structures across devices. We provide theoretical guarantees of PerMFL, which offers linear convergence rates for smooth strongly convex problems and sub-linear convergence rates for smooth non-convex problems. We conduct numerical experiments demonstrating the robust empirical performance of PerMFL, outperforming the state-of-the-art in multiple personalized federated learning tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2407_14251
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Personalized Multi-tier Federated Learning
Banerjee, Sourasekhar
Dadras, Ali
Yurtsever, Alp
Bhuyan, Monowar
Machine Learning
Artificial Intelligence
Optimization and Control
The key challenge of personalized federated learning (PerFL) is to capture the statistical heterogeneity properties of data with inexpensive communications and gain customized performance for participating devices. To address these, we introduced personalized federated learning in multi-tier architecture (PerMFL) to obtain optimized and personalized local models when there are known team structures across devices. We provide theoretical guarantees of PerMFL, which offers linear convergence rates for smooth strongly convex problems and sub-linear convergence rates for smooth non-convex problems. We conduct numerical experiments demonstrating the robust empirical performance of PerMFL, outperforming the state-of-the-art in multiple personalized federated learning tasks.
title Personalized Multi-tier Federated Learning
topic Machine Learning
Artificial Intelligence
Optimization and Control
url https://arxiv.org/abs/2407.14251