WarmFed: Federated Learning with Warm-Start for Globalization and Personalization Via Personalized Diffusion Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Feng, Tao, Zhang, Jie, Li, Xiangjian, Huang, Rong, Liu, Huashan, Wang, Zhijie
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911536293871616
author Feng, Tao
Zhang, Jie
Li, Xiangjian
Huang, Rong
Liu, Huashan
Wang, Zhijie
author_facet Feng, Tao
Zhang, Jie
Li, Xiangjian
Huang, Rong
Liu, Huashan
Wang, Zhijie
contents Federated Learning (FL) stands as a prominent distributed learning paradigm among multiple clients to achieve a unified global model without privacy leakage. In contrast to FL, Personalized federated learning aims at serving for each client in achieving persoanlized model. However, previous FL frameworks have grappled with a dilemma: the choice between developing a singular global model at the server to bolster globalization or nurturing personalized model at the client to accommodate personalization. Instead of making trade-offs, this paper commences its discourse from the pre-trained initialization, obtaining resilient global information and facilitating the development of both global and personalized models. Specifically, we propose a novel method called WarmFed to achieve this. WarmFed customizes Warm-start through personalized diffusion models, which are generated by local efficient fine-tunining (LoRA). Building upon the Warm-Start, we advance a server-side fine-tuning strategy to derive the global model, and propose a dynamic self-distillation (DSD) to procure more resilient personalized models simultaneously. Comprehensive experiments underscore the substantial gains of our approach across both global and personalized models, achieved within just one-shot and five communication(s).
format Preprint
id arxiv_https___arxiv_org_abs_2503_03110
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle WarmFed: Federated Learning with Warm-Start for Globalization and Personalization Via Personalized Diffusion Models
Feng, Tao
Zhang, Jie
Li, Xiangjian
Huang, Rong
Liu, Huashan
Wang, Zhijie
Machine Learning
Computer Vision and Pattern Recognition
Federated Learning (FL) stands as a prominent distributed learning paradigm among multiple clients to achieve a unified global model without privacy leakage. In contrast to FL, Personalized federated learning aims at serving for each client in achieving persoanlized model. However, previous FL frameworks have grappled with a dilemma: the choice between developing a singular global model at the server to bolster globalization or nurturing personalized model at the client to accommodate personalization. Instead of making trade-offs, this paper commences its discourse from the pre-trained initialization, obtaining resilient global information and facilitating the development of both global and personalized models. Specifically, we propose a novel method called WarmFed to achieve this. WarmFed customizes Warm-start through personalized diffusion models, which are generated by local efficient fine-tunining (LoRA). Building upon the Warm-Start, we advance a server-side fine-tuning strategy to derive the global model, and propose a dynamic self-distillation (DSD) to procure more resilient personalized models simultaneously. Comprehensive experiments underscore the substantial gains of our approach across both global and personalized models, achieved within just one-shot and five communication(s).
title WarmFed: Federated Learning with Warm-Start for Globalization and Personalization Via Personalized Diffusion Models
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.03110