arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04050cs.LGcs.AIcs.CV

连接点:面向黑盒视觉-语言模型的协作微调

Connecting the Dots: Collaborative Fine-tuning for Black-Box Vision-Language Models

University of Science and Technology of China(中国科学技术大学);Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);School of Artificial Intelligence, University of Chinese Academy of Sciences(中国科学院大学人工智能学院);Nanjing University(南京大学)

Zhengbo Wang, Jian Liang, Ran He, Zilei Wang, Tieniu Tan

AI总结:

本文提出CraFT方法,通过提示生成和预测精炼双模块及协作训练算法,在仅访问输入输出下实现黑盒视觉-语言模型微调,以极低内存和查询代价取得显著性能提升。

摘要:

随着预训练视觉-语言模型(VLMs)的出现,人们投入了大量精力对其进行微调以适应下游任务。尽管在设计高效的微调方法方面取得了进展,但此类方法需要访问模型的参数,由于模型所有者通常选择将其模型作为黑盒提供以保护模型所有权,这可能具有挑战性。本文提出了一种协作微调(Collaborative Fine-Tuning, CraFT)方法,用于将黑盒VLMs微调至下游任务,其中只能访问模型的输入提示和输出预测。CraFT包含两个模块:一个用于学习文本提示的提示生成模块,和一个以残差方式增强输出预测的预测精炼模块。此外,我们引入了辅助预测一致性损失,以促进这些模块之间的一致优化。这些模块通过一种新颖的协作训练算法进行优化。在15个数据集上的少样本分类的大量实验证明了CraFT的优越性。结果表明,在16-shot数据集和仅8,000次查询的情况下,CraFT实现了约12%的显著提升。此外,CraFT训练速度更快,部署时仅使用约1/80的内存占用,而与白盒方法相比仅牺牲了1.62%。

英文摘要:

With the emergence of pretrained vision-language models (VLMs), considerable efforts have been devoted to fine-tuning them for downstream tasks. Despite the progress made in designing efficient fine-tuning methods, such methods require access to the model's parameters, which can be challenging as model owners often opt to provide their models as a black box to safeguard model ownership. This paper proposes a \textbf{C}ollabo\textbf{ra}tive \textbf{F}ine-\textbf{T}uning (\textbf{CraFT}) approach for fine-tuning black-box VLMs to downstream tasks, where one only has access to the input prompts and the output predictions of the model. CraFT comprises two modules, a prompt generation module for learning text prompts and a prediction refinement module for enhancing output predictions in residual style. Additionally, we introduce an auxiliary prediction-consistent loss to promote consistent optimization across these modules. These modules are optimized by a novel collaborative training algorithm. Extensive experiments on few-shot classification over 15 datasets demonstrate the superiority of CraFT. The results show that CraFT achieves a decent gain of about 12\% with 16-shot datasets and only 8,000 queries. Moreover, CraFT trains faster and uses only about 1/80 of the memory footprint for deployment, while sacrificing only 1.62\% compared to the white-box method. Our code is publicly available at https://github.com/mrflogs/CraFT .

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑