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

一种有效的视觉-语言模型选择路由器

An Effective Router for Vision-Language Model Selection

Harbin Institute of Technology(哈尔滨工业大学);Shandong Key Laboratory of Digital Service Computing Technology and Systems(山东省数字服务计算技术与系统重点实验室)

Can Wang, Shengwei Wang, Bolin Zhang, Zhiying Tu, Dianhui Chu

AI总结:

针对视觉-语言模型(VLM)选择中数据缺乏、特征表示无效和模型空间僵化的问题,提出ARMS路由器,通过增强输入信号和扩展训练策略,在分布内和分布外测试集上表现优异,仅800M参数即可超越GPT-4o。

摘要:

具有不同性能和资源需求的视觉-语言模型(VLM)被广泛部署,使得用户难以从众多VLM候选中选择最合适的。现有工作揭示了语言模型中的性能悖论现象,并专注于路由方法来解决它。然而,开发用于VLM选择的路由器仍然是一个关键且具有挑战性的问题,主要面临:1)缺乏专门数据,2)特征表示无效,以及3)模型空间僵化和适应成本高。在本文中,我们构建了一个用于VLM选择的多模态数据集,包含七个主流VLM在32,626个独特图像-文本查询上的输出。然后,我们提出了ARMS,一个用于VLM选择的路由器。ARMS通过VLM配置文件增强输入信号,采用简单但有效的架构来改进查询和VLM能力的表示。为了提高ARMS对新VLM的适应性,我们提出了两种扩展训练策略:增量训练和独立训练。在分布内和分布外测试集上的实验结果表明了ARMS的有效性。特别是,使用我们的训练策略,ARMS(仅800M参数)可以适应更广泛的VLM空间,并击败规模大数百倍的商业模型如GPT-4o。我们的代码、模型和数据集可在匿名仓库中获取。

英文摘要:

Vision-language models (VLMs) with varying performance and resource requirements are widely deployed, making it difficult for users to select the most appropriate one among numerous VLM candidates. Existing work reveals the performance paradox phenomenon in language models and focuses on routing methods to solve it. However, developing a router for VLM selection is still a critical yet challenging problem, which primarily faces: 1) lack of specialized data, 2) ineffective feature representation, and 3) rigid model space and costly adaptation. In this paper, we construct a multimodal dataset for VLM selection, containing the outputs of seven mainstream VLMs on 32,626 unique image-text queries. We then propose ARMS, a router for VLM selection. ARMS enhances input signals with VLM profiles, employs a simple but effective architecture to improve representations of queries and VLM capabilities. To improve ARMS' adaptation to new VLMs, we propose two extension training strategies: incremental training and independent training. Experimental results on both in-distribution and out-of-distribution test sets demonstrate the effectiveness of ARMS. In particular, using our training strategy, ARMs (only 800M in size) can adapt to a broader VLM space and defeat commercial models like GPT-4o that are hundreds of times larger in scale. Our code, models, and datasets are available in the anonymous repository.

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

↑