用自然语言解释任务,让视觉模型零样本泛化到新任务
Explanatory Instructions: Towards Unified Vision Tasks Understanding and Zero-shot Generalization
- 用语言描述任务目标替代术语定义,增强模型理解力
- 在1200万组数据上训练,实现指令级零样本能力
- 适合希望提升模型泛化能力的研究者和工程师
计算机视觉(CV)虽借鉴了自然语言处理(NLP)的大规模变换模型、预训练和自回归范式等里程碑,仍未实现类似NLP的零样本任务泛化。本文认为,CV采用离散且术语化的任务定义(如“图像分割”)可能是阻碍零样本泛化的关键原因。我们假设:由于缺乏对已有任务的真正理解,深度模型难以推广到新任务。为此,提出“解释性指令”(Explanatory Instructions),通过将输入图像映射到输出的详细语言描述来定义任务目标。构建了一个包含1200万条“图像输入→解释性指令→输出”三元组的大型数据集,并训练一个基于自回归的视觉语言模型(AR-based VLM),该模型同时接收图像和解释性指令作为输入。通过学习遵循这些指令,该模型不仅实现了已知任务的指令级零样本能力,还在未见视觉任务上展现出强零样本泛化性能。代码与数据集将在GitHub公开。
原文摘要 · Abstract (English)
Computer Vision (CV) has yet to fully achieve the zero-shot task generalization observed in Natural Language Processing (NLP), despite following many of the milestones established in NLP, such as large transformer models, extensive pre-training, and the auto-regression paradigm, among others. In this paper, we explore the idea that CV adopts discrete and terminological task definitions (\eg, ``image segmentation''), which may be a key barrier to zero-shot task generalization. Our hypothesis is that without truly understanding previously-seen tasks--due to these terminological definitions--deep models struggle to generalize to novel tasks. To verify this, we introduce Explanatory Instructions, which provide an intuitive way to define CV task objectives through detailed linguistic transformations from input images to outputs. We create a large-scale dataset comprising 12 million ``image input $\to$ explanatory instruction $\to$ output'' triplets, and train an auto-regressive-based vision-language model (AR-based VLM) that takes both images and explanatory instructions as input. By learning to follow these instructions, the AR-based VLM achieves instruction-level zero-shot capabilities for previously-seen tasks and demonstrates strong zero-shot generalization for unseen CV tasks. Code and dataset will be openly available on our GitHub repository.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。