用双提示词提升跨域联邦学习中CLIP的泛化能力
FedDEAP: Adaptive Dual-Prompt Tuning for Multi-Domain Federated Learning
- 分离语义与领域特征,避免微调丢失关键信息
- 双提示设计平衡全局共享与本地个性化知识
- 对齐图文表征,保持语义与领域一致性
联邦学习(FL)允许多个客户端在不暴露本地数据的前提下协同训练模型,兼顾性能与隐私。然而,客户端间存在的领域偏移和标签异质性常导致聚合模型泛化能力下降。近期的大规模视觉语言模型如CLIP展现出强大的零样本分类能力,引发如何在联邦设置下有效微调CLIP的问题。本文提出自适应联邦提示调优框架FedDEAP,以增强CLIP在多领域场景下的泛化能力。方法包含三个关键组件:(1) 为缓解标签监督微调造成的领域特异性信息丢失,通过无偏映射的语义与领域变换网络解耦图像的语义与领域特征;(2) 为在全局提示聚合过程中保留领域特异性知识,引入双提示设计,包含全局语义提示与本地领域提示,平衡共享与个性化信息;(3) 为最大化图像生成文本特征中语义与领域信息的保留,基于两个学习到的变换对齐文本与视觉表征,确保语义与领域一致性。理论分析与四个数据集上的大量实验表明,该方法显著提升了CLIP在跨域联邦图像识别中的泛化性能。
原文摘要 · Abstract (English)
Federated learning (FL) enables multiple clients to collaboratively train machine learning models without exposing local data, balancing performance and privacy. However, domain shift and label heterogeneity across clients often hinder the generalization of the aggregated global model. Recently, large-scale vision-language models like CLIP have shown strong zero-shot classification capabilities, raising the question of how to effectively fine-tune CLIP across domains in a federated setting. In this work, we propose an adaptive federated prompt tuning framework, FedDEAP, to enhance CLIP's generalization in multi-domain scenarios. Our method includes the following three key components: (1) To mitigate the loss of domain-specific information caused by label-supervised tuning, we disentangle semantic and domain-specific features in images by using semantic and domain transformation networks with unbiased mappings; (2) To preserve domain-specific knowledge during global prompt aggregation, we introduce a dual-prompt design with a global semantic prompt and a local domain prompt to balance shared and personalized information; (3) To maximize the inclusion of semantic and domain information from images in the generated text features, we align textual and visual representations under the two learned transformations to preserve semantic and domain consistency. Theoretical analysis and extensive experiments on four datasets demonstrate the effectiveness of our method in enhancing the generalization of CLIP for federated image recognition across multiple domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。