arXiv:2410.14690cs.LGcs.AI2024-10被引 19

不用大模型也能更好分类图像,小模型路由更省成本

Rethinking VLMs and LLMs for Image Classification

  • 用轻量级小模型自动判断该用哪个模型处理图像任务
  • 在物体场景分类上,不依赖大模型的视觉模型表现更优
  • 适合需要高效低成本推理的落地应用

视觉语言模型(VLMs)正越来越多地与大型语言模型(LLMs)融合,以提升交互性和开放响应能力。然而,这种融合对图像分类这一长期核心问题的改进仍不明确。通过涵盖七种模型、十个视觉理解数据集及每数据集多种提示方式的大量实验,我们发现:在物体和场景识别任务中,不使用LLM的VLM表现优于使用LLM的VLM。但引入LLM可提升需推理与外部知识的任务性能。为此,我们提出一种务实方案:一个轻量级的LLM路由机制,通过超过250万对视觉任务与模型准确率的数据训练,智能分配任务给最合适的模型。结果表明,该方法在精度上超越或媲美当前最优方案(如GPT-4V、HuggingGPT),同时显著提升成本效益。

原文摘要 · Abstract (English)

Visual Language Models (VLMs) are now increasingly being merged with Large Language Models (LLMs) to enable new capabilities, particularly in terms of improved interactivity and open-ended responsiveness. While these are remarkable capabilities, the contribution of LLMs to enhancing the longstanding key problem of classifying an image among a set of choices remains unclear. Through extensive experiments involving seven models, ten visual understanding datasets, and multiple prompt variations per dataset, we find that, for object and scene recognition, VLMs that do not leverage LLMs can achieve better performance than VLMs that do. Yet at the same time, leveraging LLMs can improve performance on tasks requiring reasoning and outside knowledge. In response to these challenges, we propose a pragmatic solution: a lightweight fix involving a relatively small LLM that efficiently routes visual tasks to the most suitable model for the task. The LLM router undergoes training using a dataset constructed from more than 2.5 million examples of pairs of visual task and model accuracy. Our results reveal that this lightweight fix surpasses or matches the accuracy of state-of-the-art alternatives, including GPT-4V and HuggingGPT, while improving cost-effectiveness.

视觉模型模型路由成本优化

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