arXiv:2411.18363cs.CV2024-11被引 38

让多模态大模型更准地识别物体,结合感知与理解能力。

ChatRex: Taming Multimodal LLM for Joint Perception and Understanding

  • 将检测任务转为检索,让大模型更擅长定位物体
  • 在COCO上达43.9%召回率,提升感知精度
  • 适合需要精准视觉理解的AI应用开发

感知与理解是计算机视觉的两大支柱。尽管多模态大语言模型(MLLM)展现出强大的视觉理解能力,但其感知能力仍不足,例如当前最先进模型Qwen2-VL在COCO数据集上的召回率仅为43.9%,限制了需融合感知与理解的任务。本文从模型设计与数据构建两方面弥补这一差距。提出ChatRex,采用解耦感知架构:不直接由大模型预测框坐标,而是将通用提议网络生成的候选框输入大模型,由其输出对应框索引,将回归任务转化为更适配大模型的检索任务。同时,构建全自动数据引擎,生成包含多粒度标注的Rexverse-2M数据集,支持感知与理解联合训练。经三阶段训练后,ChatRex在感知与理解任务中均表现优异,二者协同能力也催生多项创新应用,验证了其互补性。代码已开源。

原文摘要 · Abstract (English)

Perception and understanding are two pillars of computer vision. While multimodal large language models (MLLM) have demonstrated remarkable visual understanding capabilities, they arguably lack accurate perception abilities, e.g. the stage-of-the-art model Qwen2-VL only achieves a 43.9 recall rate on the COCO dataset, limiting many tasks requiring the combination of perception and understanding. In this work, we aim to bridge this perception gap from both model designing and data development perspectives. We first introduce ChatRex, an MLLM with a decoupled perception design. Instead of having the LLM directly predict box coordinates, we feed the output boxes from a universal proposal network into the LLM, allowing it to output the corresponding box indices to represent its detection results, turning the regression task into a retrieval-based task that LLM handles more proficiently. From the data perspective, we build a fully automated data engine and construct the Rexverse-2M dataset which possesses multiple granularities to support the joint training of perception and understanding. After a three-stage training approach, ChatRex demonstrates strong perception and understanding performance, and the combination of these two capabilities also unlocks many attractive applications, demonstrating their complementary roles in MLLM. Code is available at https://github.com/IDEA-Research/ChatRex.

多模态模型目标检测视觉理解

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