让大模型在联邦学习中高效适配各客户端数据,尤其擅长医疗图像场景。
FAA-CLIP: Federated Adversarial Adaptation of CLIP
- 用轻量模块替代全模型传输,降低通信开销
- 通过对抗域适应实现跨客户端数据泛化,提升医学图像表现
- 冻结主模型只更新小模块,适合资源受限设备
尽管视觉语言模型(如CLIP)表现出色,但其庞大参数量成为联邦学习(FL)中的瓶颈——需频繁传输本地模型参数至全局服务器。此外,不同客户端数据异构性导致泛化性能下降,且预训练模型在医学数据上表现差,说明存在领域差距。为此,本文提出联邦对抗适配方法FAA-CLIP:采用轻量级特征适配模块(FAM)进行参数聚合,仅传输少量可训练参数,同时保持CLIP冻结以节省计算成本。通过引入域分类器,模型可学习跨客户端的域不变表示,直接缓解数据分布差异问题。在包含自然与医学图像的六个数据集上实验表明,相比现有联邦学习方法,FAA-CLIP在两类数据上均展现出更强泛化能力。代码已开源。
原文摘要 · Abstract (English)
Despite the remarkable performance of vision language models (VLMs) such as Contrastive Language Image Pre-training (CLIP), the large size of these models is a considerable obstacle to their use in federated learning (FL) systems where the parameters of local client models need to be transferred to a global server for aggregation. Another challenge in FL is the heterogeneity of data from different clients, which affects the generalization performance of the solution. In addition, natural pre-trained VLMs exhibit poor generalization ability in the medical datasets, suggests there exists a domain gap. To solve these issues, we introduce a novel method for the Federated Adversarial Adaptation (FAA) of CLIP. Our method, named FAA-CLIP, handles the large communication costs of CLIP using a light-weight feature adaptation module (FAM) for aggregation, effectively adapting this VLM to each client's data while greatly reducing the number of parameters to transfer. By keeping CLIP frozen and only updating the FAM parameters, our method is also computationally efficient. Unlike existing approaches, our FAA-CLIP method directly addresses the problem of domain shifts across clients via a domain adaptation (DA) module. This module employs a domain classifier to predict if a given sample is from the local client or the global server, allowing the model to learn domain-invariant representations. Extensive experiments on six different datasets containing both natural and medical images demonstrate that FAA-CLIP can generalize well on both natural and medical datasets compared to recent FL approaches. Our codes are available at https://github.com/AIPMLab/FAA-CLIP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。