Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Personalized Federated Learning via Dual-Prompt Optimization and Cross Fusion

About

Federated learning (FL) enables collaborative model training across decentralized clients without sharing local data, but is challenged by heterogeneity in data, computation, and communication. Pretrained vision-language models (VLMs), with their strong generalization and lightweight tuning via prompts, offer a promising solution. However, existing federated prompt-learning methods rely only on text prompts and overlook joint label-domain distribution shifts. In this paper, we propose a personalized FL framework based on dual-prompt learning and cross fusion, termed pFedDC. Specifically, each client maintains both global and local prompts across vision and language modalities: global prompts capture common knowledge shared across the federation, while local prompts encode client-specific semantics and domain characteristics. Meanwhile, a cross-fusion module is designed to adaptively integrate prompts from different levels, enabling the model to generate personalized representations aligned with each client's unique data distribution. Extensive experiments across nine datasets with various types of heterogeneity show that pFedDC consistently outperforms state-of-the-art methods.

Yuguang Zhang, Kuangpu Guo, Zhihe Lu, Yunbo Wang, Jian Liang• 2025

Related benchmarks

TaskDatasetResultRank
Image ClassificationDomainNet (test)
Average Accuracy87.11
266
Federated Few-shot Image ClassificationCIFAR10, CIFAR100, EuroSAT, Tiny-ImageNet, OxfordPet, Flower102, Caltech101, Caltech256, Food101 Local classes
Accuracy95.58
69
Image ClassificationOffice-Caltech-10 (test)
Average Accuracy97.4
58
Image ClassificationAggregate of 9 benchmarks (CIFAR10, CIFAR100, EuroSAT, OxfordPet, Flowers102, Food101, SUN397, DTD, Caltech101) Few-shot
Local Top-1 Accuracy96.28
35
Image ClassificationAggregate of 9 benchmarks (CIFAR10, CIFAR100, EuroSAT, OxfordPet, Flowers102, Food101, SUN397, DTD, Caltech101) Full-data
Average Local Top-1 Accuracy97.44
35
Federated Few-shot Image ClassificationCIFAR10, CIFAR100, EuroSAT, Tiny-ImageNet, OxfordPet, Flower102, Caltech101, Caltech256, Food101 Base classes
Accuracy47.66
18
Showing 6 of 6 rows

Other info

Follow for update