arXiv:2509.06992cs.CV2025-09被引 1

提升联邦视觉语言模型的抗攻击能力,解决本地标签不一致问题

FedAPT: Federated Adversarial Prompt Tuning for Vision-Language Models

  • 用全局标签引导生成跨客户端对齐的视觉提示
  • 在多个数据集上对抗鲁棒性显著优于现有方法
  • 适合需要跨域泛化的实际联邦学习场景

联邦提示调优(FPT)是一种高效的方法,用于大型视觉-语言模型(VLMs)在跨客户端协作微调中的应用。然而,使用FPT训练的模型容易受到对抗攻击,导致下游任务中误分类。本文提出联邦对抗提示调优(FedAPT),旨在增强FPT的对抗鲁棒性。我们发现,在非独立同分布(non-IID)设置下,客户与全局模型之间存在关键的‘类别信息鸿沟’:客户仅依赖有限的本地标签生成对抗样本进行训练,而全局模型需防御来自全局标签的攻击。为此,我们提出一种‘类别感知提示生成器’,通过全局标签嵌入(作为‘灯塔’)编码跨客户端标签信息,生成更全局对齐的视觉提示。此外,我们还提出跨层生成器共享策略,增强模型各层间的提示耦合,进一步提升对抗鲁棒性。大量实验表明,FedAPT在多个图像分类数据集上显著优于现有方法,且在跨域和跨数据集场景中表现出优异的泛化能力,证明了其在真实应用中的有效性。

原文摘要 · Abstract (English)

Federated Prompt Tuning (FPT) is an efficient method for cross-client collaborative fine-tuning of large Vision-Language Models (VLMs). However, models tuned using FPT are vulnerable to adversarial attacks, leading to misclassification in downstream tasks. In this work, we introduce Federated Adversarial Prompt Tuning (\textbf{FedAPT}), a novel method designed to enhance the adversarial robustness of FPT. We identify a key issue in FedAPT under non-independent and identically distributed (non-IID) settings: a \textit{class information gap} between clients and the global model. Clients rely solely on limited local label information to generate adversarial samples for training, while the global model must defend against adversarial attacks from global labels. To address this issue, we propose a \textbf{class-aware prompt generator} that generates visual prompts from text prompts. This generator is guided by a \emph{Global Label Embedding} (serving as a ``beacon") which encodes cross-client label information to create more globally-aligned visual prompts. Additionally, we propose a \textbf{cross-layer generator sharing} strategy to enhance prompt coupling across different layers of the model, further boosting adversarial robustness. Extensive experiments on multiple image classification datasets demonstrate the superiority of FedAPT in improving adversarial robustness, outperforming existing methods by a large margin. FedAPT also exhibits exceptional generalization in cross-domain and cross-dataset scenarios, indicating its effectiveness in real-world applications.

联邦学习对抗鲁棒提示调优视觉语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。