通过对齐图像与描述的细粒度和整体表示,提升零样本视觉识别能力。
Grounding Descriptions in Images informs Zero-Shot Visual Recognition
- 联合学习图像区域与文本描述、全局图像与整体描述的对齐。
- 在11个数据集上超越当前最佳方法,新产品数据集上识别新概念能力显著提升。
- 适合需要高精度零样本识别的研究者与应用开发者。
视觉语言模型(如CLIP)因其在开放词汇概念上的零样本视觉识别能力而备受青睐。该方法通过选择与查询图像文本表示相似度最高的类别来实现分类。尽管在某些领域表现良好,但在细粒度实体识别及未见概念泛化方面仍存在挑战。现有工作尝试在测试时引入类别描述,但效果有限。我们归因于图像与描述表示间的根本性错位,源于CLIP的预训练结构。本文提出GRAIN,一种新的预训练策略,旨在同时对齐细粒度与粗粒度表示。该方法学习将文本描述联合定位到图像区域,并将整体描述与全局图像表示对齐。为驱动预训练,我们利用冻结的多模态大语言模型(MLLMs)生成大规模合成标注。我们在11个多样化的图像分类数据集上验证了模型的零样本性能优于当前最优方法。此外,我们引入新产品-2023(Products-2023),一个手动标注的新概念数据集,展示模型对新概念的识别能力。在检索等下游任务中,模型也表现出更优的表征质量。代码已公开于https://github.com/shaunak27/grain-clip。
原文摘要 · Abstract (English)
Vision-language models (VLMs) like CLIP have been cherished for their ability to perform zero-shot visual recognition on open-vocabulary concepts. This is achieved by selecting the object category whose textual representation bears the highest similarity with the query image. While successful in some domains, this method struggles with identifying fine-grained entities as well as generalizing to unseen concepts that are not captured by the training distribution. Recent works attempt to mitigate these challenges by integrating category descriptions at test time, albeit yielding modest improvements. We attribute these limited gains to a fundamental misalignment between image and description representations, which is rooted in the pretraining structure of CLIP. In this paper, we propose GRAIN, a new pretraining strategy aimed at aligning representations at both fine and coarse levels simultaneously. Our approach learns to jointly ground textual descriptions in image regions along with aligning overarching captions with global image representations. To drive this pre-training, we leverage frozen Multimodal Large Language Models (MLLMs) to derive large-scale synthetic annotations. We demonstrate the enhanced zero-shot performance of our model compared to current state-of-the art methods across 11 diverse image classification datasets. Additionally, we introduce Products-2023, a newly curated, manually labeled dataset featuring novel concepts, and showcase our model's ability to recognize these concepts by benchmarking on it. Significant improvements achieved by our model on other downstream tasks like retrieval further highlight the superior quality of representations learned by our approach. Code available at https://github.com/shaunak27/grain-clip .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。