arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.04429cs.CVcs.LG
收藏

在图像中落地描述可增强零样本视觉识别

Grounding Descriptions in Images informs Zero-Shot Visual Recognition

Georgia Institute of Technology(佐治亚理工学院);Adobe Research(奥多比研究院);Indian Institute of Technology, Hyderabad(印度理工学院海得拉巴分校)

Shaunak Halbe, Junjiao Tian, K J Joseph, James Seale Smith, Katherine Stevo, Vineeth N Balasubramanian, Zsolt Kira

AI总结:

针对 CLIP 零样本识别中图像与类别描述表示错位的问题,本文提出 GRAIN 预训练策略,结合区域描述落地与全局字幕对齐,并在 11 个数据集及 Products-2023 上验证其优势。

摘要:

像 CLIP 这样的视觉-语言模型(VLM)因其能够对开放词汇概念执行零样本视觉识别而受到重视。这是通过选择其文本表示与查询图像具有最高相似性的对象类别来实现的。尽管该方法在某些领域取得了成功,但在识别细粒度实体以及泛化到训练分布未涵盖的未见概念方面存在困难。近期工作尝试在测试时整合类别描述以缓解这些挑战,但仅取得有限提升。我们将这些有限收益归因于图像与描述表示之间的根本性错位,其根源在于 CLIP 的预训练结构。本文提出 GRAIN,这是一种旨在同时对齐细粒度和粗粒度层级表示的新型预训练策略。我们的方法学习将文本描述共同落地到图像区域中,同时将总体字幕与全局图像表示对齐。为推动该预训练,我们利用冻结的多模态大语言模型(MLLM)获取大规模合成标注。我们在 11 个多样化图像分类数据集上证明,与当前最先进方法相比,我们的模型具有更强的零样本性能。此外,我们引入 Products-2023,这是一个新近整理、人工标注且包含新颖概念的数据集,并通过在其上进行基准测试展示模型识别这些概念的能力。我们的模型在检索等其他下游任务上取得的显著提升,进一步凸显了本方法所学表示的卓越质量。代码可在 https://github.com/shaunak27/grain-clip 获取。

英文摘要:

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 .

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑