用单一Transformer实现像素级图文理解,无需额外视觉模块。
Pixel-SAIL: Single Transformer For Pixel-Grounded Understanding
- 仅用一个Transformer处理图像和文本,省去视觉编码器等组件。
- 在多个分割与问答任务上表现接近或优于复杂模型。
- 适合追求轻量级、可扩展多模态系统的研究者与开发者。
多模态大语言模型在细粒度像素级理解任务中表现优异,但现有方法高度依赖额外组件(如CLIP视觉编码器、分割专家),导致系统复杂且难以扩展。本文提出Pixel-SAIL,一种无需引入额外组件的单一致构视觉-语言模型。受近期统一视觉-语言建模(SAIL)启发,我们对基础模型进行三项改进:首先设计可学习上采样模块以优化视觉令牌特征;其次提出新型视觉提示注入策略,实现视觉提示输入的理解并促进早期融合;第三采用视觉专家蒸馏策略,高效增强模型细粒度特征提取能力。此外,我们构建了综合性像素理解基准PerBench,包含三个任务:详细物体描述、基于视觉提示的问题回答、视觉-文本指代分割。在四个指代分割基准、一个视觉提示基准及PerBench上的大量实验表明,Pixel-SAIL以更简洁的流水线达到相当或更优性能。代码与模型将发布于https://github.com/magic-research/Sa2VA。
原文摘要 · Abstract (English)
Multimodal Large Language Models (MLLMs) achieve remarkable performance for fine-grained pixel-level understanding tasks. However, all the works rely heavily on extra components, such as vision encoder (CLIP), segmentation experts, leading to high system complexity and limiting model scaling. In this work, our goal is to explore a highly simplified MLLM without introducing extra components. Our work is motivated by the recent works on Single trAnsformer as a unified vIsion-Language Model (SAIL) design, where these works jointly learn vision tokens and text tokens in transformers. We present Pixel-SAIL, a single transformer for pixel-wise MLLM tasks. In particular, we present three technical improvements on the plain baseline. First, we design a learnable upsampling module to refine visual token features. Secondly, we propose a novel visual prompt injection strategy to enable the single transformer to understand visual prompt inputs and benefit from the early fusion of visual prompt embeddings and vision tokens. Thirdly, we introduce a vision expert distillation strategy to efficiently enhance the single transformer's fine-grained feature extraction capability. In addition, we have collected a comprehensive pixel understanding benchmark (PerBench), using a manual check. It includes three tasks: detailed object description, visual prompt-based question answering, and visual-text referring segmentation. Extensive experiments on four referring segmentation benchmarks, one visual prompt benchmark, and our PerBench show that our Pixel-SAIL achieves comparable or even better results with a much simpler pipeline. Code and model will be released at https://github.com/magic-research/Sa2VA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。