arXiv:2604.12012cs.CV2026-04被引 12

提升视觉语言模型的像素级图文对齐能力,显著增强下游任务表现

TIPSv2: Advancing Vision-Language Pretraining with Enhanced Patch-Text Alignment

  • 通过令牌蒸馏和改进的iBOT++目标函数,强化图像块与文本嵌入对齐
  • 在9个任务、20个数据集上性能超越或媲美主流视觉编码器
  • 适合需要精细图文理解的多模态应用,如图像分割与视觉问答

视觉语言预训练近年在分类、检索、分割和深度预测等下游任务中取得显著进展,但模型在密集图像块与对应概念文本嵌入的对齐能力仍不足。本文揭示,块级蒸馏可显著提升这种对齐效果——学生模型的块-文本对齐性能远超教师模型。受此启发,提出iBOT++,改进常用iBOT掩码图像目标,使未掩码令牌也直接参与损失计算,大幅增强预训练模型的块-文本对齐能力。此外,优化学习过程中的指数移动平均设置,并引入不同粒度的合成标题采样策略,以提升训练效率与效果。综合上述组件,构建TIPSv2系列图像-文本编码器,适用于广泛下游任务。在9项任务、20个数据集上的全面实验表明,其性能普遍优于或媲美近期先进视觉编码器。代码与模型已在https://gdm-tipsv2.github.io/ 公开。

原文摘要 · Abstract (English)

Recent progress in vision-language pretraining has enabled significant improvements to many downstream computer vision applications, such as classification, retrieval, segmentation and depth prediction. However, a fundamental capability that these models still struggle with is aligning dense patch representations with text embeddings of corresponding concepts. In this work, we investigate this critical issue and propose novel techniques to enhance this capability in foundational vision-language models. First, we reveal that a patch-level distillation procedure significantly boosts dense patch-text alignment -- surprisingly, the patch-text alignment of the distilled student model strongly surpasses that of the teacher model. This observation inspires us to consider modifications to pretraining recipes, leading us to propose iBOT++, an upgrade to the commonly-used iBOT masked image objective, where unmasked tokens also contribute directly to the loss. This dramatically enhances patch-text alignment of pretrained models. Additionally, to improve vision-language pretraining efficiency and effectiveness, we modify the exponential moving average setup in the learning recipe, and introduce a caption sampling strategy to benefit from synthetic captions at different granularities. Combining these components, we develop TIPSv2, a new family of image-text encoder models suitable for a wide range of downstream applications. Through comprehensive experiments on 9 tasks and 20 datasets, we demonstrate strong performance, generally on par with or better than recent vision encoder models. Code and models are released via our project page at https://gdm-tipsv2.github.io/ .

视觉语言模型图文对齐预训练iBOT

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。