用开放语言接口统一细粒度视觉感知,让模型一次搞定检测、分割和图文任务。
UFO: A Unified Approach to Fine-grained Visual Perception via Open-ended Language Interface
- 将检测、分割等任务全部转为语言空间处理,用同一套框架实现
- 在COCO上比之前最优模型高12.3 mAP,ADE20K上高3.3 mIoU
- 适配现有多模态大模型,支持推理分割等复杂任务
通用模型在语言与视觉语言任务中取得显著进展,展现出统一建模的潜力。然而,如何有效整合检测、分割等细粒度感知任务仍是挑战,主要因这些任务依赖特定设计与架构。为此,我们提出 extbf{U}nified extbf{F}ine-grained visual perception via extbf{O}pen-ended language interface(UFO),通过将所有感知目标转换到语言空间,统一对象级检测、像素级分割与图像级视觉语言任务于单一模型。我们还引入一种仅依赖语言接口的新型嵌入检索方法,支持分割任务。该框架简化了模型结构与训练策略,性能达到或超过复杂专用设计的方法。在五个标准视觉感知数据集上进行多任务训练后,UFO 在 COCO 实例分割上超越先前最优模型 12.3 mAP,ADE20K 语义分割上提升 3.3 mIoU。此外,该方法可无缝集成至现有 MLLMs,结合其强大的语言能力,实现如推理分割等更复杂任务。代码与模型已开源。
原文摘要 · Abstract (English)
Generalist models have achieved remarkable success in both language and vision-language tasks, showcasing the potential of unified modeling. However, effectively integrating fine-grained perception tasks like detection and segmentation into these models remains a significant challenge. This is primarily because these tasks often rely heavily on task-specific designs and architectures that can complicate the modeling process. To address this challenge, we present \ours, a framework that \textbf{U}nifies \textbf{F}ine-grained visual perception tasks through an \textbf{O}pen-ended language interface. By transforming all perception targets into the language space, \ours unifies object-level detection, pixel-level segmentation, and image-level vision-language tasks into a single model. Additionally, we introduce a novel embedding retrieval approach that relies solely on the language interface to support segmentation tasks. Our framework bridges the gap between fine-grained perception and vision-language tasks, significantly simplifying architectural design and training strategies while achieving comparable or superior performance to methods with intricate task-specific designs. After multi-task training on five standard visual perception datasets, \ours outperforms the previous state-of-the-art generalist models by 12.3 mAP on COCO instance segmentation and 3.3 mIoU on ADE20K semantic segmentation. Furthermore, our method seamlessly integrates with existing MLLMs, effectively combining fine-grained perception capabilities with their advanced language abilities, thereby enabling more challenging tasks such as reasoning segmentation. Code and models are available at https://github.com/nnnth/UFO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。