arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.01155cs.LG
收藏

CPT:面向黑盒优化的一致性代理调优

CPT: Consistent Proxy Tuning for Black-box Optimization

IHPC, A*STAR(新加坡科技研究局高性能计算研究院);NUS(新加坡国立大学);HIT(哈尔滨工业大学);MBZUAI(穆罕默德·本·扎耶德人工智能大学);ANU(澳大利亚国立大学)

Yuanyang He, Zitong Huang, Xinxing Xu, Rick Siow Mong Goh, Salman Khan, Wangmeng Zuo, Yong Liu, Chun-Mei Feng

AI总结:

本文提出一致性代理调优(CPT),通过同时利用冻结的大黑盒模型和小白盒模型,解决代理调优中训练与测试不一致的问题,在 LLMs 和 VLMs 的黑盒调优上取得更优性能。

摘要:

黑盒调优近年来受到关注,因为先进专有模型的结构或内部参数无法访问。代理调优(Proxy-tuning)为黑盒语言模型的调优提供了一种测试时输出调整方法。它利用调优前后较小的白盒“代理”模型输出 logits 的差异来改进黑盒模型。然而,该技术仅作为解码时算法使用,导致训练与测试之间不一致,从而可能限制整体性能。为解决这一问题,我们提出一致性代理调优(Consistent Proxy Tuning,CPT),一种简单而有效的黑盒调优方法。与代理调优不同,CPT 额外利用冻结的大型黑盒模型和另一个冻结的小型白盒模型,确保训练阶段优化目标与测试时代理之间的一致性。这种一致性有利于代理调优并提升模型性能。值得注意的是,我们的方法仅关注 logit 级别的计算,这使其具有模型无关性,并适用于任何涉及 logit 分类的任务。大量实验结果表明,我们的 CPT 在大型语言模型(LLMs)和视觉-语言模型(VLMs)的黑盒调优中,在多个数据集上均表现出优越性。代码可在 https://github.com/chunmeifeng/CPT 获取。

英文摘要:

Black-box tuning has attracted recent attention due to that the structure or inner parameters of advanced proprietary models are not accessible. Proxy-tuning provides a test-time output adjustment for tuning black-box language models. It applies the difference of the output logits before and after tuning a smaller white-box "proxy" model to improve the black-box model. However, this technique serves only as a decoding-time algorithm, leading to an inconsistency between training and testing which potentially limits overall performance. To address this problem, we introduce Consistent Proxy Tuning (CPT), a simple yet effective black-box tuning method. Different from Proxy-tuning, CPT additionally exploits the frozen large black-box model and another frozen small white-box model, ensuring consistency between training-stage optimization objective and test-time proxies. This consistency benefits Proxy-tuning and enhances model performance. Note that our method focuses solely on logit-level computation, which makes it model-agnostic and applicable to any task involving logit classification. Extensive experimental results demonstrate the superiority of our CPT in both black-box tuning of Large Language Models (LLMs) and Vision-Language Models (VLMs) across various datasets. The code is available at https://github.com/chunmeifeng/CPT.

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

↑