English

Parameter-Efficient and Personalized Federated Training of Generative Models at the Edge

Machine Learning 2025-12-30 v3 Distributed, Parallel, and Cluster Computing

Abstract

Large generative models (for example, language and diffusion models) enable high-quality text and image synthesis but are hard to train or adapt in cross-device federated settings due to heavy computation and communication and statistical/system heterogeneity. We propose FedGen-Edge, a framework that decouples a frozen, pre-trained global backbone from lightweight client-side adapters and federates only the adapters. Using Low-Rank Adaptation (LoRA) constrains client updates to a compact subspace, which reduces uplink traffic by more than 99 percent versus full-model FedAvg, stabilizes aggregation under non-IID data, and naturally supports personalization because each client can keep a locally tuned adapter. On language modeling (PTB) and image generation (CIFAR-10), FedGen-Edge achieves lower perplexity/FID and faster convergence than strong baselines while retaining a simple FedAvg-style server. A brief ablation shows diminishing returns beyond moderate LoRA rank and a trade-off between local epochs and client drift. FedGen-Edge offers a practical path toward privacy-preserving, resource-aware, and personalized generative AI on heterogeneous edge devices.

Keywords

Cite

@article{arxiv.2511.11585,
  title  = {Parameter-Efficient and Personalized Federated Training of Generative Models at the Edge},
  author = {Kabir Khan and Manju Sarkar and Anita Kar and Suresh Ghosh},
  journal= {arXiv preprint arXiv:2511.11585},
  year   = {2025}
}

Comments

37 pages, 8 figures

R2 v1 2026-07-01T07:37:56.411Z