用简单分类监督提升CLIP的图文细粒度对齐能力
SuperCLIP: CLIP with Simple Classification Supervision
- 在视觉编码器加轻量线性层,引入词元级文本监督
- 零样本分类准确率提升1.8%,图像-文本检索效果更好
- 无需额外数据,小批量训练也更稳定,适合实用部署
对比学习型图文预训练模型(CLIP)通过将图像与文本映射到共享嵌入空间,实现了强大的跨模态泛化能力。然而,现有研究发现,这类模型仍未能充分利用文本中的细粒度语义信息,尤其在处理长而详细的标题时问题更明显。这源于其训练目标仅优化全局图像-文本相似度,忽视了词元级别的监督信号,限制了细粒度对齐能力。为此,我们提出SuperCLIP,一种简单有效的框架,通过在视觉编码器中添加一个轻量级线性层,引入基于分类的监督,利用词元级线索增强视觉-文本对齐。该方法仅增加0.077%的总计算量,无需额外标注数据。实验表明,SuperCLIP在零样本分类、图像-文本检索及纯视觉任务上均实现持续提升,无论模型训练数据是原始网络数据还是经丰富重标注的数据,都能有效恢复文本监督。此外,分类监督机制缓解了CLIP在小批量情况下的性能下降问题。代码与模型将开源。
原文摘要 · Abstract (English)
Contrastive Language-Image Pretraining (CLIP) achieves strong generalization in vision-language tasks by aligning images and texts in a shared embedding space. However, recent findings show that CLIP-like models still underutilize fine-grained semantic signals in text, and this issue becomes even more pronounced when dealing with long and detailed captions. This stems from CLIP's training objective, which optimizes only global image-text similarity and overlooks token-level supervision - limiting its ability to achieve fine-grained visual-text alignment. To address this, we propose SuperCLIP, a simple yet effective framework that augments contrastive learning with classification-based supervision. By adding only a lightweight linear layer to the vision encoder, SuperCLIP leverages token-level cues to enhance visual-textual alignment - with just a 0.077% increase in total FLOPs, and no need for additional annotated data. Experiments show that SuperCLIP consistently improves zero-shot classification, image-text retrieval, and purely visual tasks. These gains hold regardless of whether the model is trained on original web data or rich re-captioned data, demonstrating SuperCLIP's ability to recover textual supervision in both cases. Furthermore, SuperCLIP alleviates CLIP's small-batch performance drop through classification-based supervision that avoids reliance on large batch sizes. Code and models will be made open source.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。