arXiv:2602.18439cs.CVcs.LG2026-02

用动态提示生成提升视觉语言模型在联邦学习中的跨类别泛化能力

Replication Study: Federated Text-Driven Prompt Generation for Vision-Language Models

  • 基于类别名称动态生成提示,避免静态提示的泛化瓶颈
  • 在6个数据集上平均准确率达74.58%(已见类)和76.00%(未见类)
  • 验证了联邦学习下不共享数据仍可实现高泛化性能

视觉语言模型如CLIP展现出强大的零样本能力,但在联邦学习场景下的泛化能力仍面临挑战,尤其对未见类别。原论文FedTPG提出一种文本驱动的提示生成网络,根据类别名动态生成提示,从而提升联邦设置下的跨类别泛化能力。本文对该方法进行忠实复现,在六个不同视觉数据集(Caltech101、Oxford Flowers、FGVC Aircraft、Oxford Pets、Food-101、DTD)上评估预训练模型,结果与原论文报告准确率相差不超过0.2%,在已见类上平均准确率为74.58%,未见类为76.00%,泛化性能提升1.43个百分点。实验验证了:(1)文本驱动提示生成相比静态提示学习具有更优的未见类别泛化能力;(2)在不共享私有数据的前提下,联邦训练提示生成器仍能在多种视觉领域保持高性能。复现成功证明了FedTPG方法的鲁棒性与可复现性。

原文摘要 · Abstract (English)

Vision-language models like CLIP have demonstrated remarkable zero-shot capabilities, yet their adaptation to federated learning scenarios presents significant challenges, particularly regarding generalization to unseen classes. The original FedTPG paper \cite{Qiu2024} addresses this limitation by introducing a text driven prompt generation network that dynamically creates prompts conditioned on class names, enabling better cross-class generalization in federated settings. In this work, we present a faithful replication study of FedTPG, evaluating the pre-trained model on six diverse vision datasets: Caltech101, Oxford Flowers, FGVC Aircraft, Oxford Pets, Food-101, and DTD. Our evaluation achieves results within 0.2\% of the original paper's reported accuracies, with an average accuracy of 74.58\% on seen (base) classes and 76.00\% on unseen (new) classes, demonstrating a +1.43 percentage point improvement in generalization. These results validate the original paper's core claims: (1) text-driven prompt generation enables superior generalization to unseen classes compared to static prompt learning methods, and (2) federated training of prompt generators maintains high performance across diverse visual domains without sharing private data. Our successful replication confirms the robustness and reproducibility of the FedTPG approach.

联邦学习提示生成零样本

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