FedMGP: Personalized Federated Learning with Multi-Group Text-Visual Prompts

TL;DR

FedMGP introduces a personalized federated learning framework using multi-group prompts for vision-language models, enhancing semantic diversity and communication efficiency, and achieving state-of-the-art results in federated vision-language tasks.

Contribution

It proposes a novel multi-group prompt paradigm with diversity loss and similarity-guided aggregation, improving personalization and robustness in federated vision-language learning.

Findings

Outperforms prior methods in personalization tasks

Reduces communication overhead significantly

Enhances domain generalization across benchmarks

Abstract

In this paper, we introduce FedMGP, a new paradigm for personalized federated prompt learning in vision-language models. FedMGP equips each client with multiple groups of paired textual and visual prompts, enabling the model to capture diverse, fine-grained semantic and instance-level cues. A diversity loss is introduced to drive each prompt group to specialize in distinct and complementary semantic aspects, ensuring that the groups collectively cover a broader range of local characteristics. During communication, FedMGP employs a dynamic prompt aggregation strategy based on similarity-guided probabilistic sampling: each client computes the cosine similarity between its prompt groups and the global prompts from the previous round, then samples s groups via a softmax-weighted distribution. This soft selection mechanism preferentially aggregates semantically aligned knowledge while still enabling exploration of underrepresented patterns effectively balancing the preservation of common knowledge with client-specific features. Notably, FedMGP maintains parameter efficiency by redistributing a fixed prompt capacity across multiple groups, achieving state-of-the-art performance with the lowest communication parameters among all federated prompt learning methods. Theoretical analysis shows that our dynamic aggregation strategy promotes robust global representation learning by reinforcing shared semantics while suppressing client-specific noise. Extensive experiments demonstrate that FedMGP consistently outperforms prior approaches in both personalization and domain generalization across diverse federated vision-language benchmarks. The code will be released on https://github.com/weihao-bo/FedMGP.git.