Personalized Federated Learning via Dual-Prompt Optimization and Cross Fusion

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Yuguang, Guo, Kuangpu, Lu, Zhihe, Wang, Yunbo, Liang, Jian
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911024189276160
author Zhang, Yuguang
Guo, Kuangpu
Lu, Zhihe
Wang, Yunbo
Liang, Jian
author_facet Zhang, Yuguang
Guo, Kuangpu
Lu, Zhihe
Wang, Yunbo
Liang, Jian
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2506_21144
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Personalized Federated Learning via Dual-Prompt Optimization and Cross Fusion
Zhang, Yuguang
Guo, Kuangpu
Lu, Zhihe
Wang, Yunbo
Liang, Jian
Machine Learning
Computer Vision and Pattern Recognition
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.
title Personalized Federated Learning via Dual-Prompt Optimization and Cross Fusion
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2506.21144