Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

SDFed: Bridging Local Global Discrepancy via Subspace Refinement and Divergence Control in Federated Prompt Learning

About

Vision-language pretrained models offer strong transferable representations, yet adapting them in privacy-sensitive multi-party settings is challenging due to the high communication cost of federated optimization and the limited local data on clients. Federated prompt learning mitigates this issue by keeping the VLPM backbone frozen and collaboratively training lightweight prompt parameters. However, existing approaches typically enforce a unified prompt structure and length across clients, which is inadequate under practical client heterogeneity in both data distributions and system resources, and may further introduce conflicts between globally shared and locally optimal knowledge. To address these challenges, we propose \textbf{SDFed}, a heterogeneous federated prompt learning framework that bridges Local-Global Discrepancy via Subspace Refinement and Divergence Control. SDFed maintains a fixed-length global prompt for efficient aggregation while allowing each client to learn a variable-length local prompt to better match its data characteristics and capacity. To mitigate local-global conflicts and facilitate effective knowledge transfer, SDFed introduces a subspace refinement method for local prompts and an information retention and divergence control strategy that preserves key local information while maintaining appropriate separability between global and local representations. Extensive experiments on several datasets demonstrate that SDFed consistently improves performance and robustness in heterogeneous federated settings.

Yicheng Di, Wei Yuan, Tieke He, Zhanjie Zhang, Ao Ma, Yuan Liu, Hongzhi Yin• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 (test)
Accuracy95.21
3381
Image ClassificationTiny ImageNet (test)
Accuracy56.81
265
Image ClassificationOffice-31
Average Accuracy94.68
261
Image ClassificationOffice-Home (test)
Mean Accuracy89.04
199
Image ClassificationDTD (test)
Accuracy93.56
181
Image ClassificationCaltech101 (test)
Accuracy99.12
121
Image ClassificationFood101 (test)
Accuracy96.38
87
Image ClassificationFlowers102 (test)
Accuracy99.35
68
Image ClassificationOxford Pets (test)
Accuracy99.36
35
Image ClassificationOffice-31 Amazon domain (test)
Accuracy91.05
20
Showing 10 of 12 rows

Other info

Follow for update