首个像素级图文描述数据集,提升模型细粒度视觉理解能力
Pix2Cap-COCO: Advancing Visual Comprehension via Pixel-Level Captioning
- 用GPT-4V自动生成像素对齐的物体专属描述,实现细粒度标注
- 构建16.7万条平均22.9词的详细描述,挑战模型多任务理解能力
- 适用于提升大模型视觉-语言能力,尤其适合图像区域理解研究者
我们提出Pix2Cap-COCO,首个面向全景像素级描述的标注数据集,旨在推动细粒度视觉理解。通过设计自动化标注流程,利用GPT-4V为图像中每个物体生成像素对齐、实例特定的描述,使模型能学习物体与其上下文间的精细关系。该方法生成167,254条详细描述,平均每条22.94词。基于此数据集,我们引入全新任务——全景分割-描述,要求模型同时识别图像中的实例并提供详细描述。为评估该任务,我们基于X-Decoder设计稳健基线。实验表明,Pix2Cap-COCO极具挑战性,要求模型在细粒度视觉理解与语言生成两方面均表现优异。此外,我们将Pix2Cap-COCO用于大模态模型(LMMs)的监督微调(SFT),显著提升性能:例如,在GPT4RoI上训练后,Visual Genome数据集上CIDEr提升+1.4%,ROUGE提升+0.4%,SPICE提升+0.5%;在ViP-BENCH上整体提升+5.1%,其中识别准确率提升+11.2%,语言生成质量提升+22.2%。
原文摘要 · Abstract (English)
We present Pix2Cap-COCO, the first panoptic pixel-level caption dataset designed to advance fine-grained visual understanding. To achieve this, we carefully design an automated annotation pipeline that prompts GPT-4V to generate pixel-aligned, instance-specific captions for individual objects within images, enabling models to learn more granular relationships between objects and their contexts. This approach results in 167,254 detailed captions, with an average of 22.94 words per caption. Building on Pix2Cap-COCO, we introduce a novel task, panoptic segmentation-captioning, which challenges models to recognize instances in an image and provide detailed descriptions for each simultaneously. To benchmark this task, we design a robust baseline based on X-Decoder. The experimental results demonstrate that Pix2Cap-COCO is a particularly challenging dataset, as it requires models to excel in both fine-grained visual understanding and detailed language generation. Furthermore, we leverage Pix2Cap-COCO for Supervised Fine-Tuning (SFT) on large multimodal models (LMMs) to enhance their performance. For example, training with Pix2Cap-COCO significantly improves the performance of GPT4RoI, yielding gains in CIDEr +1.4%, ROUGE +0.4%, and SPICE +0.5% on Visual Genome dataset, and strengthens its region understanding ability on the ViP-BENCH, with an overall improvement of +5.1%, including notable increases in recognition accuracy +11.2% and language generation quality +22.2%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。